Catch socket.errors when sending / recving bytes on wake socketpair (#1069)
This commit is contained in:
parent
d40d106c73
commit
bb9642f04c
@ -829,7 +829,9 @@ class KafkaClient(object):
|
|||||||
|
|
||||||
def wakeup(self):
|
def wakeup(self):
|
||||||
with self._wake_lock:
|
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!')
|
log.warning('Unable to send to wakeup socket!')
|
||||||
|
|
||||||
def _clear_wake_fd(self):
|
def _clear_wake_fd(self):
|
||||||
@ -837,7 +839,7 @@ class KafkaClient(object):
|
|||||||
while True:
|
while True:
|
||||||
try:
|
try:
|
||||||
self._wake_r.recv(1024)
|
self._wake_r.recv(1024)
|
||||||
except:
|
except socket.error:
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user