From 14b48489d49edd59d847f4f793130a3c17f3cb11 Mon Sep 17 00:00:00 2001 From: Mayuri Nehate <33225191+mayurinehate@users.noreply.github.com> Date: Wed, 28 Dec 2022 02:15:11 +0530 Subject: [PATCH] feat(ingest): pass timeout config in kafka admin client api calls (#6863) --- metadata-ingestion/src/datahub/ingestion/source/kafka.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/metadata-ingestion/src/datahub/ingestion/source/kafka.py b/metadata-ingestion/src/datahub/ingestion/source/kafka.py index 61f75fd3a4..54dee81fb9 100644 --- a/metadata-ingestion/src/datahub/ingestion/source/kafka.py +++ b/metadata-ingestion/src/datahub/ingestion/source/kafka.py @@ -417,7 +417,8 @@ class KafkaSource(StatefulIngestionSourceBase): configs: Dict[ ConfigResource, concurrent.futures.Future ] = self.admin_client.describe_configs( - resources=[ConfigResource(ResourceType.TOPIC, t) for t in topics] + resources=[ConfigResource(ResourceType.TOPIC, t) for t in topics], + request_timeout=self.source_config.connection.client_timeout_seconds, ) logger.debug("Waiting for config details futures to complete") concurrent.futures.wait(configs.values())