update: samples/3catchall/main.py to handle exception in case of webs…

…ocket.
itaru2622 · Sep 22, 2024 · 577cf58 · 577cf58
1 parent 0e1d8df
commit 577cf58
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/samples/3catchall/main.py b/samples/3catchall/main.py
@@ -27,5 +27,8 @@ async def catch_websocket(ws:WebSocket, full_path: str):
     input = { "full_path": full_path, "query": dict(ws._query_params), "headers": dict(ws._headers) }
     await ws.send_text(f'new websocket connection: {input}')
     while True:
-        msg = await ws.receive_text()
-        await ws.send_text(f'got: {msg}')
+       try:
+          msg = await ws.receive_text()
+          await ws.send_text(f'got: {msg}')
+       except:
+          break