class WebSocketConnHandler(tornado.websocket.WebSocketHandler): def open(self): pass def on_message(self, msg): global backend_worker if hasattr(self, 'conntype'): self.stream.recv_msg(msg) else: info = json.loads(msg) self.conntype = info['conntype'] self.stream = WebSocketStream(self) if self.conntype == 'main': self._handle_mainconn(self.stream, info) elif self.conntype == 'file': self._handle_fileconn(self.stream, info) else: self.stream.close() def on_close(self): if hasattr(self, 'conntype'): self.stream.close() def _handle_mainconn(self, main_stream, info): global backend_worker try: backend_worker.add_client(info['client_link'], main_stream) except Exception: main_stream.close() def _handle_fileconn(self, file_stream, info): global backend_worker try: backend_worker.add_filestream('websocket', info['filekey'], file_stream) print('test') except Exception as err: file_stream.close()
class WebSocketConnHandler(tornado.websocket.WebSocketHandler): def open(self): pass def on_message(self,msg): global backend_worker if hasattr(self,'conntype'): self.stream.recv_msg(msg) else: info = json.loads(msg) self.conntype = info['conntype'] self.stream = WebSocketStream(self) if self.conntype == 'main': self._handle_mainconn(self.stream,info) elif self.conntype == 'file': self._handle_fileconn(self.stream,info) else: self.stream.close() def on_close(self): if hasattr(self,'conntype'): self.stream.close() def _handle_mainconn(self,main_stream,info): global backend_worker try: backend_worker.add_client(info['client_link'],main_stream) except Exception: main_stream.close() def _handle_fileconn(self,file_stream,info): global backend_worker try: backend_worker.add_filestream('websocket',info['filekey'], file_stream) print('test') except Exception as err: file_stream.close()
def on_message(self, msg): global backend_worker if hasattr(self, 'conntype'): self.stream.recv_msg(msg) else: info = json.loads(msg) self.conntype = info['conntype'] self.stream = WebSocketStream(self) if self.conntype == 'main': self._handle_mainconn(self.stream, info) elif self.conntype == 'file': self._handle_fileconn(self.stream, info) else: self.stream.close()
def on_message(self,msg): global backend_worker if hasattr(self,'conntype'): self.stream.recv_msg(msg) else: info = json.loads(msg) self.conntype = info['conntype'] self.stream = WebSocketStream(self) if self.conntype == 'main': self._handle_mainconn(self.stream,info) elif self.conntype == 'file': self._handle_fileconn(self.stream,info) else: self.stream.close()