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