openstack/deb-python-kafka
Code Issues Proposed changes

Use the same timeout when reinitializing a connection

Browse Source
This commit is contained in:
Omar Ghishan
2014-01-07 17:23:30 -08:00
parent e0c45ff898
commit e5a5477d34
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
kafka/conn.py
View File

@@ -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