Create Kafka client for confluent lib

DataDog · Mar 1, 2023 · 0d556d7 · 0d556d7
1 parent 0a0dfb5
commit 0d556d7
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/kafka_consumer/datadog_checks/kafka_consumer/client/confluent_kafka_client.py b/kafka_consumer/datadog_checks/kafka_consumer/client/confluent_kafka_client.py
@@ -2,9 +2,22 @@
 # All rights reserved
 # Licensed under a 3-clause BSD style license (see LICENSE)
 from datadog_checks.kafka_consumer.client.kafka_client import KafkaClient
-
+from confluent_kafka.admin import AdminClient
 
 class ConfluentKafkaClient(KafkaClient):
+    @property
+    def kafka_client(self):
+        if self._kafka_client is None:
+            # self.conf is just the config options from librdkafka
+            self._kafka_client = AdminClient(
+                {
+                    "bootstrap.servers": self.config._kafka_connect_str,
+                    "socket.timeout.ms": self.config._request_timeout_ms,
+                    "client.id": "dd-agent",
+                }
+            )
+        return self._kafka_client
+
     def create_kafka_admin_client(self):
         raise NotImplementedError