1580158
/
lora-scripts
mirror of https://github.com/Akegarasu/lora-scripts
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update

pull/363/head
Akegarasu 2024-01-27 14:03:26 +08:00
parent d0aa118d0a
commit ac44b7688d
No known key found for this signature in database
GPG Key ID: 9D600258808ACBCD
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mikazuki/app/proxy.py
View File

@ -60,6 +60,8 @@ async def proxy_ws_forward(ws_a: WebSocket, ws_b: websockets.WebSocketClientProt
try:
data = await ws_a.receive_text()
await ws_b.send(data)
except starlette.websockets.WebSocketDisconnect as e:
break
except Exception as e:
log.error(f"Error when proxy data client -> backend: {e}")
break
@ -70,6 +72,8 @@ async def proxy_ws_reverse(ws_a: WebSocket, ws_b: websockets.WebSocketClientProt
try:
data = await ws_b.recv()
await ws_a.send_text(data)
except websockets.exceptions.ConnectionClosedOK as e:
break
except Exception as e:
log.error(f"Error when proxy data backend -> client: {e}")
break