feat(devices): skip realtime requests when device offline

This commit is contained in:
stu2not
2026-05-25 17:45:16 +08:00
parent 3449224cd2
commit 6c77672ca7
6 changed files with 135 additions and 6 deletions

View File

@@ -12,15 +12,23 @@ class ConnectionManager:
async with self.lock:
self.connections[device_id] = websocket
async def remove_connection(self, device_id: str):
async def remove_connection(self, device_id: str, websocket: Optional[WebSocket] = None):
async with self.lock:
if device_id in self.connections:
if device_id in self.connections and (websocket is None or self.connections[device_id] is websocket):
del self.connections[device_id]
async def get_connection(self, device_id: str) -> Optional[WebSocket]:
async with self.lock:
return self.connections.get(device_id)
async def is_connected(self, device_id: str) -> bool:
websocket = await self.get_connection(device_id)
return (
websocket is not None
and getattr(getattr(websocket, "client_state", None), "name", None) == "CONNECTED"
and not getattr(websocket, "_closed", False)
)
async def get_all_connections(self) -> Dict[str, WebSocket]:
async with self.lock:
return dict(self.connections)