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

5
kafka/conn.py
View File

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