summaryrefslogtreecommitdiff
path: root/sleekxmpp/xmlstream/xmlstream.py
diff options
context:
space:
mode:
authorNathan Fritz <nathan@andyet.net>2010-10-20 16:32:50 -0700
committerNathan Fritz <nathan@andyet.net>2010-10-20 16:32:50 -0700
commit6e34b2cfdd2cd4605bac6e0d8695b87b5a184ae2 (patch)
tree287243b47ec58f91d0be1e402275d0c4d281223e /sleekxmpp/xmlstream/xmlstream.py
parente18354ae0e068d6e95ea13567e75da4f0c6c03eb (diff)
downloadslixmpp-6e34b2cfdd2cd4605bac6e0d8695b87b5a184ae2.tar.gz
slixmpp-6e34b2cfdd2cd4605bac6e0d8695b87b5a184ae2.tar.bz2
slixmpp-6e34b2cfdd2cd4605bac6e0d8695b87b5a184ae2.tar.xz
slixmpp-6e34b2cfdd2cd4605bac6e0d8695b87b5a184ae2.zip
fixed disconnect
Diffstat (limited to 'sleekxmpp/xmlstream/xmlstream.py')
-rw-r--r--sleekxmpp/xmlstream/xmlstream.py12
1 files changed, 10 insertions, 2 deletions
diff --git a/sleekxmpp/xmlstream/xmlstream.py b/sleekxmpp/xmlstream/xmlstream.py
index 5fd7be9d..3459f507 100644
--- a/sleekxmpp/xmlstream/xmlstream.py
+++ b/sleekxmpp/xmlstream/xmlstream.py
@@ -175,6 +175,8 @@ class XMLStream(object):
self.stream_footer = "</stream>"
self.stop = threading.Event()
+ self.stream_end_event = threading.Event()
+ self.stream_end_event.set()
self.event_queue = queue.Queue()
self.send_queue = queue.Queue()
self.scheduler = Scheduler(self.event_queue, self.stop)
@@ -295,8 +297,9 @@ class XMLStream(object):
# closed in the other direction.
if not reconnect:
self.auto_reconnect = False
+ self.stream_end_event.wait(4)
+ if not self.auto_reconnect:
self.stop.set()
- time.sleep(1)
try:
self.socket.close()
self.filesocket.close()
@@ -699,6 +702,7 @@ class XMLStream(object):
root = xml
# Perform any stream initialization actions, such
# as handshakes.
+ self.stream_end_event.clear()
self.start_stream_handler(root)
depth += 1
if event == b'end':
@@ -707,6 +711,7 @@ class XMLStream(object):
# The stream's root element has closed,
# terminating the stream.
logging.debug("Ending read XML loop")
+ self.stream_end_event.set()
return False
elif depth == 1:
# We only raise events for stanzas that are direct
@@ -831,7 +836,10 @@ class XMLStream(object):
"""
try:
while not self.stop.isSet():
- data = self.send_queue.get(True)
+ try:
+ data = self.send_queue.get(True, 1)
+ except queue.Empty:
+ continue
logging.debug("SEND: %s" % data)
try:
self.socket.send(data.encode('utf-8'))