From 58375955a9b832356ba68cc1f9ecb3bc274b6922 Mon Sep 17 00:00:00 2001 From: Nathan Fritz Date: Mon, 15 Feb 2010 02:13:44 -0800 Subject: [PATCH] added send queueing to avoid mixed sending --- setup.py | 8 ++++---- sleekxmpp/xmlstream/xmlstream.py | 34 +++++++++++++++++++++----------- 2 files changed, 26 insertions(+), 16 deletions(-) diff --git a/setup.py b/setup.py index 163cb8c..7b1c087 100644 --- a/setup.py +++ b/setup.py @@ -44,10 +44,10 @@ packages = [ 'sleekxmpp', 'sleekxmpp/xmlstream/matcher', 'sleekxmpp/xmlstream/handler' ] -if sys.version_info < (3, 0): - packages.append('sleekxmpp/xmlstream/tostring26') -else: - packages.append('sleekxmpp/xmlstream/tostring') +#if sys.version_info < (3, 0): +# packages.append('sleekxmpp/xmlstream/tostring26') +#else: +# packages.append('sleekxmpp/xmlstream/tostring') setup( name = "sleekxmpp", diff --git a/sleekxmpp/xmlstream/xmlstream.py b/sleekxmpp/xmlstream/xmlstream.py index 0cdfdf9..44f8384 100644 --- a/sleekxmpp/xmlstream/xmlstream.py +++ b/sleekxmpp/xmlstream/xmlstream.py @@ -66,6 +66,7 @@ class XMLStream(object): self.stream_footer = "" self.eventqueue = queue.Queue() + self.sendqueue = queue.Queue() self.namespace_map = {} @@ -139,12 +140,17 @@ class XMLStream(object): for t in range(0, HANDLER_THREADS): self.__thread['eventhandle%s' % t] = threading.Thread(name='eventhandle%s' % t, target=self._eventRunner) self.__thread['eventhandle%s' % t].start() + self.__thread['sendthread'] = threading.Thread(name='sendthread', target=self._sendThread) + self.__thread['sendthread'].start() if threaded: self.__thread['process'] = threading.Thread(name='process', target=self._process) self.__thread['process'].start() else: self._process() + def schedule(self, seconds, handler, args=None): + threading.Timer(seconds, handler, args).start() + def _process(self): "Start processing the socket." firstrun = True @@ -222,19 +228,23 @@ class XMLStream(object): if event == b'start': edepth += 1 + def _sendThread(self): + while True: + data = self.sendqueue.get(True) + logging.debug("SEND: %s" % data) + try: + self.socket.send(data.encode('utf-8')) + #self.socket.send(bytes(data, "utf-8")) + #except socket.error,(errno, strerror): + except: + self.state.set('connected', False) + if self.state.reconnect: + logging.error("Disconnected. Socket Error.") + traceback.print_exc() + self.disconnect(reconnect=True) + def sendRaw(self, data): - logging.debug("SEND: %s" % data) - try: - self.socket.send(data.encode('utf-8')) - #self.socket.send(bytes(data, "utf-8")) - #except socket.error,(errno, strerror): - except: - self.state.set('connected', False) - if self.state.reconnect: - logging.error("Disconnected. Socket Error.") - traceback.print_exc() - self.disconnect(reconnect=True) - return False + self.sendqueue.put(data) return True def disconnect(self, reconnect=False):