From 4356830fd278ca286c2d229ef2a95975500acbef Mon Sep 17 00:00:00 2001 From: bohat <32742355+Bohatman@users.noreply.github.com> Date: Fri, 2 Aug 2024 02:32:01 +0700 Subject: [PATCH] At stop_consumer function pass additional config to AdminClient (#50) --- src/ConfluentKafkaLibrary/consumer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ConfluentKafkaLibrary/consumer.py b/src/ConfluentKafkaLibrary/consumer.py index 4196029..847626c 100644 --- a/src/ConfluentKafkaLibrary/consumer.py +++ b/src/ConfluentKafkaLibrary/consumer.py @@ -34,7 +34,7 @@ def __init__( server=server, port=port, **kwargs) - + self.kwargs = kwargs if not isinstance(topics, list): topics = [topics] self.consumer.subscribe_topic(self.group_id, topics=topics) @@ -65,7 +65,7 @@ def stop_consumer(self): self.join() self.consumer.unsubscribe(self.group_id) self.consumer.close_consumer(self.group_id) - admin_client = AdminClient({'bootstrap.servers': f'{self.server}:{self.port}'}) + admin_client = AdminClient({'bootstrap.servers': f'{self.server}:{self.port}', **self.kwargs}) response = admin_client.delete_consumer_groups([self.group_id], request_timeout=10) try: response[self.group_id].result()