chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(plugin): a try catch for webscoket send

Browse Source
This commit is contained in:
BennyKok 2023-12-18 13:23:35 +08:00
parent 5b2cb407c7
commit 0dec0015a5
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
custom_routes.py
View File

@ -132,13 +132,18 @@ async def websocket_handler(request):
return ws return ws
async def send(event, data, sid=None): async def send(event, data, sid=None):
if sid: try:
ws = sockets.get(sid) if sid:
if ws: ws = sockets.get(sid)
await ws.send_json({ 'event': event, 'data': data }) if ws and not ws.closed: # Check if the WebSocket connection is open and not closing
else: await ws.send_json({ 'event': event, 'data': data })
for ws in sockets.values(): else:
await ws.send_json({ 'event': event, 'data': data }) for ws in sockets.values():
if not ws.closed: # Check if the WebSocket connection is open and not closing
await ws.send_json({ 'event': event, 'data': data })
except Exception as e:
print(f"Exception: {e}")
traceback.print_exc()
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)