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()
