Catch socket.errors when sending / recving bytes on wake socketpairwake_socketpair_exceptions

1 files changed, 4 insertions, 2 deletions

diff --git a/kafka/client_async.py b/kafka/client_async.py
index cf62a8a..fbeb775 100644
--- a/kafka/client_async.py
+++ b/kafka/client_async.py
@@ -829,7 +829,9 @@ class KafkaClient(object):
 
     def wakeup(self):
         with self._wake_lock:
-            if self._wake_w.send(b'x') != 1:
+            try:
+                assert self._wake_w.send(b'x') == 1
+            except (AssertionError, socket.error):
                 log.warning('Unable to send to wakeup socket!')
 
     def _clear_wake_fd(self):
@@ -837,7 +839,7 @@ class KafkaClient(object):
         while True:
             try:
                 self._wake_r.recv(1024)
-            except:
+            except socket.error:
                 break