jinql
/
webssh
mirror of https://github.com/huashengdun/webssh
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Keep tracking worker mode, update handler when necessary

pull/4/head
Sheng 2017-11-12 10:33:58 +08:00
parent 08af39dcb8
commit 04105c11c9
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
main.py
View File

@ -41,6 +41,7 @@ class Worker(object):
self.id = str(id(self))
self.data_to_dst = []
self.handler = None
self.mode = IOLoop.READ
def __call__(self, fd, events):
if events & IOLoop.READ:
@ -51,9 +52,13 @@ class Worker(object):
self.close()
def set_handler(self, handler):
if self.handler:
return
self.handler = handler
if not self.handler:
self.handler = handler
def update_handler(self, mode):
if self.mode != mode:
self.loop.update_handler(self.fd, mode)
self.mode = mode
def on_read(self):
logging.debug('worker {} on read'.format(self.id))
@ -90,15 +95,15 @@ class Worker(object):
if errno_from_exception(e) in _ERRNO_CONNRESET:
self.close()
else:
self.loop.update_handler(self.fd, IOLoop.WRITE)
self.update_handler(IOLoop.WRITE)
else:
self.data_to_dst = []
data = data[sent:]
if data:
self.data_to_dst.append(data)
self.loop.update_handler(self.fd, IOLoop.WRITE)
self.update_handler(IOLoop.WRITE)
else:
self.loop.update_handler(self.fd, IOLoop.READ)
self.update_handler(IOLoop.READ)
def close(self):
logging.debug('Closing worker {}'.format(self.id))