diff options
Diffstat (limited to 'kafka/conn.py')
| -rw-r--r-- | kafka/conn.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/kafka/conn.py b/kafka/conn.py index 5b199f8..94cf584 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -354,7 +354,7 @@ class BrokerConnection(object): next_lookup = self._next_afi_sockaddr() if not next_lookup: self.close(Errors.KafkaConnectionError('DNS failure')) - return + return self.state else: log.debug('%s: creating new socket', self) self._sock_afi, self._sock_addr = next_lookup @@ -409,6 +409,7 @@ class BrokerConnection(object): ' Disconnecting.', self, ret) errstr = errno.errorcode.get(ret, 'UNKNOWN') self.close(Errors.KafkaConnectionError('{} {}'.format(ret, errstr))) + return self.state # Needs retry else: @@ -443,6 +444,7 @@ class BrokerConnection(object): if time.time() > request_timeout + self.last_attempt: log.error('Connection attempt to %s timed out', self) self.close(Errors.KafkaConnectionError('timeout')) + return self.state return self.state |