openstack/deb-python-kafka
Code Issues Proposed changes

Also/continue to check for timeouts in BrokerConnection.recv()

Browse Source
This commit is contained in:
Dana Powers
2016-11-17 20:24:14 -08:00
parent 0bf7ea0f67
commit dc45935ec8
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
kafka/conn.py
View File

@@ -575,7 +575,15 @@ class BrokerConnection(object):
log.warning('%s: No in-flight-requests to recv', self) log.warning('%s: No in-flight-requests to recv', self)
return None return None
return self._recv() response = self._recv()
if not response and self.requests_timed_out():
log.warning('%s timed out after %s ms. Closing connection.',
self, self.config['request_timeout_ms'])
self.close(error=Errors.RequestTimedOutError(
'Request timed out after %s ms' %
self.config['request_timeout_ms']))
return None
return response
def _recv(self): def _recv(self):
# Not receiving is the state of reading the payload header # Not receiving is the state of reading the payload header