laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Catch websocket disconnect (#184)

Browse Source
This commit is contained in:
Nikhil Rao 2022-12-26 23:14:35 -08:00 committed by GitHub
parent af9733996a
commit a93b1425c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pynecone/app.py
View File

@ -4,6 +4,7 @@ from typing import Any, Callable, Coroutine, Dict, List, Optional, Tuple, Type,
from fastapi import FastAPI, WebSocket from fastapi import FastAPI, WebSocket
from fastapi.middleware import cors from fastapi.middleware import cors
from starlette.websockets import WebSocketDisconnect
from pynecone import constants, utils from pynecone import constants, utils
from pynecone.base import Base from pynecone.base import Base
@ -320,7 +321,11 @@ def _event(app: App):
# Process events until the connection is closed. # Process events until the connection is closed.
while True: while True:
# Get the event. # 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. # Process the event.
update = await process(app, event) update = await process(app, event)