Commit 2e9595fd authored by Jim Fulton's avatar Jim Fulton

Added a should_close method that allows connection closes to be

queued.
parent 3600059c
......@@ -68,6 +68,8 @@ SEND_SIZE = 60000
MAC_BIT = 0x80000000L
_close_marker = object()
class SizedMessageAsyncConnection(asyncore.dispatcher):
__super_init = asyncore.dispatcher.__init__
__super_close = asyncore.dispatcher.close
......@@ -235,6 +237,9 @@ class SizedMessageAsyncConnection(asyncore.dispatcher):
else:
return True
def should_close(self):
self.__output.append(_close_marker)
def handle_write(self):
self.__output_lock.acquire()
try:
......@@ -250,7 +255,13 @@ class SizedMessageAsyncConnection(asyncore.dispatcher):
l = 0
for i in range(len(output)):
try:
l += len(output[i])
except TypeError:
# We had an output marker, close the connection
assert output[i] is _close_marker
return self.close()
if l > SEND_SIZE:
break
......
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