github/webssh
1
0
Fork 0
mirror of https://github.com/huashengdun/webssh.git synced 2026-08-20 04:36:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update handler events

Browse Source
This commit is contained in:
Sheng
2017-11-10 17:49:26 +08:00
parent c2ce7d10c4
commit cc1cffe7da
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+4 -1
View File
@@ -84,6 +84,9 @@ class Worker(object):
data = data[sent:]
if data:
self.data_to_dst.append(data)
self.loop.update_handler(self.fd, IOLoop.WRITE)
else:
self.loop.update_handler(self.fd, IOLoop.READ)
def close(self):
logging.debug('Closing worker {}'.format(self.id))
@@ -196,7 +199,7 @@ class WsockHandler(tornado.websocket.WebSocketHandler):
self.set_nodelay(True)
worker.set_handler(self)
self.worker_ref = weakref.ref(worker)
self.loop.add_handler(worker.fd, worker, IOLoop.READ | IOLoop.WRITE)
self.loop.add_handler(worker.fd, worker, IOLoop.READ)
def on_message(self, message):
logging.debug('"{}" from {}'.format(message, self.src_addr))