Commit cc1cffe7 by Sheng

Update handler events

parent c2ce7d10
...@@ -84,6 +84,9 @@ class Worker(object): ...@@ -84,6 +84,9 @@ class Worker(object):
data = data[sent:] data = data[sent:]
if data: if data:
self.data_to_dst.append(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): def close(self):
logging.debug('Closing worker {}'.format(self.id)) logging.debug('Closing worker {}'.format(self.id))
...@@ -196,7 +199,7 @@ class WsockHandler(tornado.websocket.WebSocketHandler): ...@@ -196,7 +199,7 @@ class WsockHandler(tornado.websocket.WebSocketHandler):
self.set_nodelay(True) self.set_nodelay(True)
worker.set_handler(self) worker.set_handler(self)
self.worker_ref = weakref.ref(worker) 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): def on_message(self, message):
logging.debug('"{}" from {}'.format(message, self.src_addr)) logging.debug('"{}" from {}'.format(message, self.src_addr))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment