Make _wake_r socket non-blocking; drop select from _clear_wake_fd

1 files changed, 4 insertions, 3 deletions

diff --git a/kafka/client_async.py b/kafka/client_async.py
index e921fa4..ca51987 100644
--- a/kafka/client_async.py
+++ b/kafka/client_async.py
@@ -98,6 +98,7 @@ class KafkaClient(object):
         self._bootstrap_fails = 0
         self._bootstrap(collect_hosts(self.config['bootstrap_servers']))
         self._wake_r, self._wake_w = socket.socketpair()
+        self._wake_r.setblocking(False)
 
     def __del__(self):
         self._wake_r.close()
@@ -682,10 +683,10 @@ class KafkaClient(object):
 
     def _clear_wake_fd(self):
         while True:
-            fds, _, _ = select.select([self._wake_r], [], [], 0)
-            if not fds:
+            try:
+                self._wake_r.recv(1)
+            except:
                 break
-            self._wake_r.recv(1)
 
 
 class DelayedTaskQueue(object):