ws

transport_ws.py

@@ -20,9 +20,12 @@ class WSTransport:
                 return
             
             await websocket.accept()
-            while True:
+            try:
-                data = await websocket.receive_bytes()
+                while True:
-                await self._recv_q.put(data)
+                    data = await websocket.receive_bytes()
+                    await self._recv_q.put(data)
+            except WebSocketDisconnect:
+                pass
 
     async def start(self):
         if self.thread != None: