Catch websocket disconnect (#184)

reflex-dev · Dec 27, 2022 · a93b142 · a93b142
1 parent af97339
commit a93b142
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pynecone/app.py b/pynecone/app.py
@@ -4,6 +4,7 @@
 
 from fastapi import FastAPI, WebSocket
 from fastapi.middleware import cors
+from starlette.websockets import WebSocketDisconnect
 
 from pynecone import constants, utils
 from pynecone.base import Base
@@ -320,7 +321,11 @@ async def ws(websocket: WebSocket):
         # Process events until the connection is closed.
         while True:
             # Get the event.
-            event = Event.parse_raw(await websocket.receive_text())
+            try:
+                event = Event.parse_raw(await websocket.receive_text())
+            except WebSocketDisconnect:
+                # Close the connection.
+                return
 
             # Process the event.
             update = await process(app, event)