GitHub/PyLink
3
0
Fork 0
mirror of https://github.com/jlu5/PyLink.git synced 2024-11-27 13:09:23 +01:00
Code Issues Projects Releases Wiki Activity

Retry when socket.send() fails with BlockingIOError / EAGAIN

Browse Source
This commit is contained in:
James Lu 2020-09-29 17:43:38 +00:00
parent 2aa00d6efc
commit d50de12834
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
classes.py
View File

@ -2091,11 +2091,22 @@ class IRCNetwork(PyLinkNetworkCoreWithUtils):
log.debug("(%s) -> %s", self.name, data)
try:
self._socket.send(encoded_data)
except:
log.exception("(%s) Failed to send message %r; aborting!", self.name, data)
self.disconnect()
while True:
try:
self._socket.send(encoded_data)
except (BlockingIOError, ssl.SSLWantReadError, ssl.SSLWantWriteError):
# The send attempt failed, wait a little bit.
# I would prefer using a blocking socket and MSG_DONTWAIT in recv()'s flags
# but SSLSocket doesn't support that...
throttle_time = self.serverdata.get('throttle_time', 0)
if self._aborted.wait(throttle_time):
break
continue
except:
log.exception("(%s) Failed to send message %r; aborting!", self.name, data)
self.disconnect()
else:
break
def send(self, data, queue=True):
"""send() wrapper with optional queueing support."""