watch websocket errors

This commit is contained in:
j 2019-02-02 12:57:46 +05:30
parent c95e22869c
commit ad15b8a25a

View file

@ -49,7 +49,11 @@ class Handler(WebSocketHandler):
#websocket calls #websocket calls
def on_message(self, message): def on_message(self, message):
try:
action, data = json.loads(message) action, data = json.loads(message)
except json.decoder.JSONDecodeError:
logger.debug('invalid websocket message: %s', message)
return
if state.tasks: if state.tasks:
state.tasks.queue(action, data) state.tasks.queue(action, data)
@ -62,7 +66,17 @@ class Handler(WebSocketHandler):
if self.ws_connection is None: if self.ws_connection is None:
self.on_close() self.on_close()
else: else:
state.main.add_callback(lambda: self.write_message(message)) state.main.add_callback(lambda: self._write_message(message))
async def _write_message(self, message):
try:
task = self.write_message(message)
await task
except tornado.iostream.StreamClosedError as e:
self.on_close()
except tornado.websocket.WebSocketClosedError as e:
self.on_close()
def trigger_event(event, data): def trigger_event(event, data):
#if len(state.websockets): #if len(state.websockets):