gavindiaz/PolyWeather
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Direct resync stale SSE clients

Browse Source
This commit is contained in:
2569718930@qq.com
2026-05-31 21:48:09 +08:00
parent a3df2e00dc
commit 38c6cefb27
2 changed files with 111 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
web/routers/sse_router.py
+39 -10
View File
@@ -22,6 +22,7 @@ _live_subscription_started = False
SSE_REPLAY_BASE_LIMIT = 60
SSE_REPLAY_EVENTS_PER_CITY = 24
SSE_REPLAY_MAX_LIMIT = 240
SSE_REPLAY_DIRECT_RESYNC_FACTOR = 3
def _parse_cities_param(cities: str) -> Set[str]:
@@ -46,6 +47,21 @@ def _bounded_replay_limit(value: int, *, city_count: int = 0) -> int:
return max(1, min(MAX_REPLAY_LIMIT, route_limit, limit))
def _should_direct_resync(
*,
since_revision: int,
latest_revision: int,
limit: int,
) -> bool:
since = max(0, int(since_revision or 0))
latest = max(0, int(latest_revision or 0))
if since <= 0 or latest <= since:
return False
gap = latest - since
threshold = max(SSE_REPLAY_MAX_LIMIT, max(1, int(limit or 1)) * SSE_REPLAY_DIRECT_RESYNC_FACTOR)
return gap > threshold
def _ensure_live_subscription() -> None:
starter = getattr(event_store, "start_live_subscription", None)
if not callable(starter):
@@ -82,23 +98,36 @@ async def sse_events(
if since_revision is not None:
try:
replay_events = event_store.replay_events(
cities=city_set,
since_revision=max(0, int(since_revision)),
limit=limit,
)
if event_store.replay_requires_resync(
cities=city_set,
since_revision=max(0, int(since_revision)),
replay_count=len(replay_events),
since = max(0, int(since_revision))
if _should_direct_resync(
since_revision=since,
latest_revision=latest_revision,
limit=limit,
):
resync_event = {
"type": "resync_required",
"reason": "replay_window_exceeded",
"reason": "replay_gap_too_large",
"latest_revision": latest_revision,
"ts": int(time.time() * 1000),
}
else:
replay_events = event_store.replay_events(
cities=city_set,
since_revision=since,
limit=limit,
)
if event_store.replay_requires_resync(
cities=city_set,
since_revision=since,
replay_count=len(replay_events),
limit=limit,
):
resync_event = {
"type": "resync_required",
"reason": "replay_window_exceeded",
"latest_revision": latest_revision,
"ts": int(time.time() * 1000),
}
except Exception:
resync_event = {
"type": "resync_required",