Chia-Network/chia-blockchain
1
0
Fork 0
mirror of https://github.com/Chia-Network/chia-blockchain.git synced 2024-09-11 18:58:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

more ws message type awareness in the deamon (#15797)

Browse Source
This commit is contained in:
Kyle Altendorf 2023-09-07 15:05:57 -04:00 committed by GitHub
parent d3f2dfaa1e
commit 1c5b469c5b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
chia/daemon/server.py
View File

@ -288,12 +288,23 @@ class WebSocketServer:
if len(service_names) == 0:
service_names = ["Unknown"]
if msg.type == WSMsgType.CLOSE:
self.log.info(f"ConnectionClosed. Closing websocket with {service_names}")
closing_message = f"Closing websocket with {service_names}."
level = logging.INFO
if msg.type == WSMsgType.CLOSED:
message = f"Connection closed. {closing_message}"
elif msg.type == WSMsgType.CLOSING:
message = f"Connection closing. {closing_message}"
elif msg.type == WSMsgType.CLOSE:
message = f"Connection close requested. {closing_message}"
elif msg.type == WSMsgType.ERROR:
self.log.info(f"Websocket exception. Closing websocket with {service_names}. {ws.exception()}")
level = logging.ERROR
message = f"Websocket exception. {closing_message} {ws.exception()}"
else:
self.log.info(f"Unexpected message type. Closing websocket with {service_names}. {msg.type}")
level = logging.ERROR
message = f"Unexpected message type. {closing_message} {msg.type}"
self.log.log(level=level, msg=message)
await ws.close()
break