diff options
| author | Omar Ghishan <omar.ghishan@rd.io> | 2014-01-07 17:23:30 -0800 |
|---|---|---|
| committer | Omar Ghishan <omar.ghishan@rd.io> | 2014-01-07 17:35:01 -0800 |
| commit | e5a5477d346a140abd8bba9e5da2a8f9967ac911 (patch) | |
| tree | 605ce9737ff41db307faba3e12599bfcc8dafb96 | |
| parent | e0c45ff898aa4cb8fb4534be7d585451bafecd35 (diff) | |
| download | kafka-python-e5a5477d346a140abd8bba9e5da2a8f9967ac911.tar.gz | |
Use the same timeout when reinitializing a connection
| -rw-r--r-- | kafka/conn.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/kafka/conn.py b/kafka/conn.py index ff823d2..a8d6b1f 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -25,7 +25,8 @@ class KafkaConnection(local): self.port = port self._sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self._sock.connect((host, port)) - self._sock.settimeout(timeout) + self.timeout = timeout + self._sock.settimeout(self.timeout) self._dirty = False def __str__(self): @@ -123,5 +124,5 @@ class KafkaConnection(local): self.close() self._sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self._sock.connect((self.host, self.port)) - self._sock.settimeout(10) + self._sock.settimeout(self.timeout) self._dirty = False |