openstack/deb-python-kafka
Code Issues Proposed changes

Add next_correlation_id_recv + send helper methods to BrokerConnection

Browse Source
This commit is contained in:
Dana Powers
2015-12-09 15:58:18 -08:00
parent 06cc91c64c
commit e37049fb69
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
kafka/conn.py
View File

@@ -105,6 +105,14 @@ class BrokerConnection(local):
return None return None
return response return response
def next_correlation_id_recv(self):
if len(self.in_flight_requests) == 0:
return None
return self.in_flight_requests[0][0]
def next_correlation_id_send(self):
return self.correlation_id + 1
def __getnewargs__(self): def __getnewargs__(self):
return (self.host, self.port, self.timeout) return (self.host, self.port, self.timeout)