Merge pull request dpkp#45 from quixey/allow-client-id

mumrah · mumrah · commit 9af7b817c17a · 2013-09-24T16:44:28.000-07:00
Allow a client id to be passed to the client +1 thanks, @jimjh
diff --git 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 @@ def _load_metadata_for_topics(self, *topics):
         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 @@ def _send_broker_aware_request(self, payloads, encoder_fn, decoder_fn):
         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
