openstack/deb-python-kafka
Code Issues Proposed changes

Support requests that do not expect a response in BrokerConnection

Browse Source
This commit is contained in:
Dana Powers
2015-12-09 15:53:07 -08:00
parent 98f393ba6f
commit 06cc91c64c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
kafka/conn.py
View File

@@ -61,7 +61,7 @@ class BrokerConnection(local):
self._write_fd = None self._write_fd = None
self.in_flight_requests.clear() self.in_flight_requests.clear()
def send(self, request): def send(self, request, expect_response=True):
if not self.connected() and not self.connect(): if not self.connected() and not self.connect():
return None return None
self.correlation_id += 1 self.correlation_id += 1
@@ -78,6 +78,7 @@ class BrokerConnection(local):
log.exception("Error in BrokerConnection.send()") log.exception("Error in BrokerConnection.send()")
self.close() self.close()
return None return None
if expect_response:
self.in_flight_requests.append((self.correlation_id, request.RESPONSE_TYPE)) self.in_flight_requests.append((self.correlation_id, request.RESPONSE_TYPE))
return self.correlation_id return self.correlation_id