kennethnym/infinifi
1
0
Fork 0
Code Issues 2 Pull Requests 2 Actions Packages Projects Releases Wiki Activity

fix: websocket disconnection not handled properly

Browse Source
This commit is contained in:
Kenneth
2024-07-28 12:01:47 +01:00
parent 78ea0244de
commit 0179f03b9a
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
server.py
View File

@@ -111,7 +111,7 @@ async def ws_endpoint(ws: WebSocket):
await ws.close() await ws.close()
ws_connection_manager.disconnect(ws) ws_connection_manager.disconnect(ws)
await ws.send_text(f"{len(active_listeners)}") await ws_connection_manager.broadcast(f"{len(active_listeners)}")
try: try:
while True: while True:
@@ -121,15 +121,12 @@ async def ws_endpoint(ws: WebSocket):
active_listeners.add(addr) active_listeners.add(addr)
await ws_connection_manager.broadcast(f"{len(active_listeners)}") await ws_connection_manager.broadcast(f"{len(active_listeners)}")
elif msg == "paused": elif msg == "paused":
active_listeners.remove(addr) active_listeners.discard(addr)
await ws_connection_manager.broadcast(f"{len(active_listeners)}") await ws_connection_manager.broadcast(f"{len(active_listeners)}")
except WebSocketDisconnect: except WebSocketDisconnect:
if ws.client: active_listeners.discard(addr)
addr, _ = ws.client
active_listeners.discard(addr)
ws_connection_manager.disconnect(ws) ws_connection_manager.disconnect(ws)
await ws_connection_manager.broadcast(f"{len(active_listeners)}") await ws_connection_manager.broadcast(f"{len(active_listeners)}")