From 3085faa5484123b5dd18aa96d0f331593228603d Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Sat, 6 Jun 2015 18:13:13 -0700 Subject: [PATCH] KafkaClient should try/except ConnnectionError when calling _get_conn --- kafka/client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kafka/client.py b/kafka/client.py index 4302ce5..63b33b3 100644 --- a/kafka/client.py +++ b/kafka/client.py @@ -169,19 +169,19 @@ class KafkaClient(object): responses_by_broker = collections.defaultdict(list) broker_failures = [] for broker, payloads in payloads_by_broker.items(): - conn = self._get_conn(broker.host.decode('utf-8'), broker.port) requestId = self._next_id() request = encoder_fn(client_id=self.client_id, correlation_id=requestId, payloads=payloads) # Send the request, recv the response try: + conn = self._get_conn(broker.host.decode('utf-8'), broker.port) conn.send(requestId, request) except ConnectionError as e: broker_failures.append(broker) log.warning("Could not send request [%s] to server %s: %s", - binascii.b2a_hex(request), conn, e) + binascii.b2a_hex(request), broker, e) for payload in payloads: responses_by_broker[broker].append(FailedPayloadsError(payload))