Close websocket if there is no corresponding ssh connection

pull/296/merge
Sheng 2022-11-21 14:52:38 +08:00
parent 9a7cfe767c
commit 2a46b52eac
1 changed files with 6 additions and 0 deletions

View File

@ -566,7 +566,13 @@ class WsockHandler(MixinHandler, tornado.websocket.WebSocketHandler):
*self.src_addr *self.src_addr
) )
) )
self.close(reason='No worker found')
return return
if worker.closed:
self.close(reason='Worker closed')
return
try: try:
msg = json.loads(message) msg = json.loads(message)
except JSONDecodeError: except JSONDecodeError: