0x2620/openmedialibrary
1
0
Fork 0
Code Issues 131 Pull requests Releases Wiki Activity

watch websocket errors

Browse source
This commit is contained in:
j 2019-02-02 12:57:46 +05:30
parent c95e22869c
commit ad15b8a25a
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
oml/websocket.py
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):