diff options
Diffstat (limited to 'kafka/conn.py')
| -rw-r--r-- | kafka/conn.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/kafka/conn.py b/kafka/conn.py index c5d3be1..005dd7e 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -561,9 +561,9 @@ class BrokerConnection(object): class ConnFilter(Filter): def filter(self, record): - if record.funcName in ('recv', 'send'): - return False - return True + if record.funcName == 'check_version': + return True + return False log_filter = ConnFilter() log.addFilter(log_filter) @@ -598,11 +598,11 @@ class BrokerConnection(object): # the attempt to write to a disconnected socket should # immediately fail and allow us to infer that the prior # request was unrecognized - self.send(MetadataRequest[0]([])) + mr = self.send(MetadataRequest[0]([])) if self._sock: self._sock.setblocking(True) - while not f.is_done: + while not (f.is_done and mr.is_done): self.recv() if self._sock: self._sock.setblocking(False) |