Catch socket.errors when sending / recving bytes on wake socketpair (#1069)

Author: dpkp

kafka/client_async.py

@@ -829,15 +829,17 @@ def check_version(self, node_id=None, timeout=2, strict=False):
 
     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):
         # reading from wake socket should only happen in a single thread
         while True:
             try:
                 self._wake_r.recv(1024)
-            except:
+            except socket.error:
                 break