Retry `select` calls on `InterruptedError`

Ref #738

1 files changed, 11 insertions, 0 deletions

diff --git a/redis/_compat.py b/redis/_compat.py
index c4a72d9..4713d01 100644
--- a/redis/_compat.py
+++ b/redis/_compat.py
@@ -9,6 +9,15 @@ if sys.version_info[0] < 3 or (sys.version_info[0] == 3 and
     import time
     import errno
 
+    from select import select as _select
+
+    def select(rlist, wlist, xlist, timeout):
+        while True:
+            try:
+                return _select(rlist, wlist, xlist, timeout)
+            except InterruptedError:
+                continue
+
     # Wrapper for handling interruptable system calls.
     def _retryable_call(s, func, *args, **kwargs):
         # Some modules (SSL) use the _fileobject wrapper directly and
@@ -54,6 +63,8 @@ if sys.version_info[0] < 3 or (sys.version_info[0] == 3 and
         return _retryable_call(sock, sock.recv_into, *args, **kwargs)
 
 else:  # Python 3.5 and above automatically retry EINTR
+    from select import select
+
     def recv(sock, *args, **kwargs):
         return sock.recv(*args, **kwargs)