From b6a2ad9caa8d7b5b87d3808650376e7751d4e4da Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Wed, 30 Dec 2015 12:45:24 -0800 Subject: BrokerConnection: wrap socket errors in retriable ConnectionError --- kafka/conn.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/kafka/conn.py b/kafka/conn.py index 8ce4a6f..3e49841 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -165,7 +165,7 @@ class BrokerConnection(object): self._sock.setblocking(False) except (AssertionError, socket.error) as e: log.exception("Error sending %s to %s", request, self) - self.close(error=e) + self.close(error=Errors.ConnectionError(e)) return future.failure(e) log.debug('%s Request %d: %s', self, correlation_id, request) @@ -230,7 +230,7 @@ class BrokerConnection(object): return None log.exception('%s: Error receiving 4-byte payload header -' ' closing socket', self) - self.close(error=e) + self.close(error=Errors.ConnectionError(e)) return None if self._rbuffer.tell() == 4: @@ -253,7 +253,7 @@ class BrokerConnection(object): if e.errno == errno.EWOULDBLOCK: return None log.exception('%s: Error in recv', self) - self.close(error=e) + self.close(error=Errors.ConnectionError(e)) return None staged_bytes = self._rbuffer.tell() -- cgit v1.2.1