Retry on EINTR error

bombela · bombela · commit dd30e2cb7107 · 2013-06-09T18:48:40.000-07:00
For whatever reason ZMQ now raises ZMQErrors with an errno.EINTR code. We simply retry when they occur. fixes 0rpc#62
diff --git a/zerorpc/gevent_zmq.py b/zerorpc/gevent_zmq.py
@@ -33,6 +33,7 @@
 import gevent.event
 import gevent.core
 import sys
+import errno
 
 class Context(_zmq.Context):
 
@@ -104,7 +105,7 @@ def send(self, data, flags=0, copy=True, track=False):
                 self._on_state_changed()
                 return msg
             except _zmq.ZMQError, e:
-                if e.errno != _zmq.EAGAIN:
+                if e.errno not in (_zmq.EAGAIN, errno.EINTR):
                     raise
             self._writable.clear()
             # The following sleep(0) force gevent to switch out to another
@@ -140,7 +141,7 @@ def recv(self, flags=0, copy=True, track=False):
                 self._on_state_changed()
                 return msg
             except _zmq.ZMQError, e:
-                if e.errno != _zmq.EAGAIN:
+                if e.errno not in (_zmq.EAGAIN, errno.EINTR):
                     raise
             self._readable.clear()
             # The following sleep(0) force gevent to switch out to another
