mirror of
https://github.com/Monadical-SAS/reflector.git
synced 2026-03-21 22:56:47 +00:00
Fix websocket disconnect errors
This commit is contained in:
@@ -1,6 +1,6 @@
|
|||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
|
||||||
from fastapi import APIRouter, WebSocket
|
from fastapi import APIRouter, WebSocket, WebSocketDisconnect
|
||||||
|
|
||||||
from reflector.auth.auth_jwt import JWTAuth # type: ignore
|
from reflector.auth.auth_jwt import JWTAuth # type: ignore
|
||||||
from reflector.db.users import user_controller
|
from reflector.db.users import user_controller
|
||||||
@@ -60,6 +60,8 @@ async def user_events_websocket(websocket: WebSocket):
|
|||||||
try:
|
try:
|
||||||
while True:
|
while True:
|
||||||
await websocket.receive()
|
await websocket.receive()
|
||||||
|
except (RuntimeError, WebSocketDisconnect):
|
||||||
|
pass
|
||||||
finally:
|
finally:
|
||||||
if room_id:
|
if room_id:
|
||||||
await ws_manager.remove_user_from_room(room_id, websocket)
|
await ws_manager.remove_user_from_room(room_id, websocket)
|
||||||
|
|||||||
Reference in New Issue
Block a user