github/SleekXMPP
1
0
Fork 0
mirror of https://github.com/correl/SleekXMPP.git synced 2024-12-24 11:50:53 +00:00
Code Issues Projects Releases Packages Wiki Activity

Handle sending stanzas in chunks if the socket has poor performance.

Browse source
This commit is contained in:
Lance Stout 2011-08-25 15:08:45 -07:00
parent d929e0deb2
commit b8a4ffece9
3 changed files with 26 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
sleekxmpp/test/livesocket.py
View file

@ -138,7 +138,7 @@ class TestLiveSocket(object):
"""
with self.send_queue_lock:
self.send_queue.put(data)
self.socket.send(data)
return self.socket.send(data)
# ------------------------------------------------------------------
# File Socket

1
sleekxmpp/test/mocksocket.py
View file

@ -121,6 +121,7 @@ class TestSocket(object):
if self.disconnected:
raise socket.error
self.send_queue.put(data)
return len(data)
# ------------------------------------------------------------------
# File Socket

36
sleekxmpp/xmlstream/xmlstream.py
View file

@ -915,7 +915,15 @@ class XMLStream(object):
if now:
log.debug("SEND (IMMED): %s" % data)
try:
self.socket.send(data.encode('utf-8'))
data = data.encode('utf-8')
total = len(data)
sent = 0
count = 0
while sent < total and not self.stop.is_set():
sent += self.socket.send(data[sent:])
count += 1
if count > 1:
log.debug('SENT: %d chunks' % count)
except Socket.error as serr:
self.event('socket_error', serr)
log.warning("Failed to send %s" % data)
@ -1222,7 +1230,7 @@ class XMLStream(object):
Extract stanzas from the send queue and send them on the stream.
"""
try:
while not self.stop.isSet():
while not self.stop.is_set():
self.session_started_event.wait()
if self.__failed_send_stanza is not None:
data = self.__failed_send_stanza
@ -1234,22 +1242,26 @@ class XMLStream(object):
continue
log.debug("SEND: %s" % data)
try:
self.socket.send(data.encode('utf-8'))
enc_data = data.encode('utf-8')
total = len(enc_data)
sent = 0
count = 0
while sent < total and not self.stop.is_set():
sent += self.socket.send(enc_data[sent:])
count += 1
if count > 1:
log.debug('SENT: %d chunks' % count)
self.send_queue.task_done()
except Socket.error as serr:
self.event('socket_error', serr)
log.warning("Failed to send %s" % data)
self.__failed_send_stanza = data
self.disconnect(self.auto_reconnect)
except KeyboardInterrupt:
log.debug("Keyboard Escape Detected in _send_thread")
self.event('killed', direct=True)
self.disconnect()
return
except SystemExit:
self.disconnect()
self.event_queue.put(('quit', None, None))
return
except Exception as ex:
log.exception('Unexpected error in send thread: %s' % ex)
self.exception(ex)
if not self.stop.is_set():
self.disconnect(self.auto_reconnect)
def exception(self, exception):
"""