From ef4648a0840bf8a212de5c7129d38e34cb335acb Mon Sep 17 00:00:00 2001 From: maciejkula Date: Wed, 16 Apr 2014 17:05:45 +0100 Subject: [PATCH] Fix lack of timeout support in KafkaClient and KafkaConnection --- kafka/client.py | 2 +- kafka/conn.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/kafka/client.py b/kafka/client.py index 39c89ba43..5dee7b752 100644 --- a/kafka/client.py +++ b/kafka/client.py @@ -48,7 +48,7 @@ def _get_conn(self, host, port): host_key = (host, port) if host_key not in self.conns: - self.conns[host_key] = KafkaConnection(host, port) + self.conns[host_key] = KafkaConnection(host, port, timeout=self.timeout) return self.conns[host_key] diff --git a/kafka/conn.py b/kafka/conn.py index 4fdeb17c7..5dc3d5acb 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -150,6 +150,6 @@ def reinit(self): """ self.close() self._sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) - self._sock.connect((self.host, self.port)) self._sock.settimeout(self.timeout) + self._sock.connect((self.host, self.port)) self._dirty = False