diff options
| -rw-r--r-- | kafka/client.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/kafka/client.py b/kafka/client.py index 2ec1f1f..be61e91 100644 --- a/kafka/client.py +++ b/kafka/client.py @@ -20,9 +20,10 @@ class KafkaClient(object): CLIENT_ID = "kafka-python" ID_GEN = count() - def __init__(self, host, port, bufsize=4096): + def __init__(self, host, port, bufsize=4096, client_id=CLIENT_ID): # We need one connection to bootstrap - self.bufsize = bufsize + self.bufsize = bufsize + self.client_id = client_id self.conns = { # (host, port) -> KafkaConnection (host, port): KafkaConnection(host, port, bufsize) } @@ -59,7 +60,7 @@ class KafkaClient(object): recurse in the event of a retry. """ requestId = self._next_id() - request = KafkaProtocol.encode_metadata_request(KafkaClient.CLIENT_ID, + request = KafkaProtocol.encode_metadata_request(self.client_id, requestId, topics) response = self._send_broker_unaware_request(requestId, request) @@ -156,7 +157,7 @@ class KafkaClient(object): for broker, payloads in payloads_by_broker.items(): conn = self._get_conn_for_broker(broker) requestId = self._next_id() - request = encoder_fn(client_id=KafkaClient.CLIENT_ID, + request = encoder_fn(client_id=self.client_id, correlation_id=requestId, payloads=payloads) # Send the request, recv the response |