diff --git a/ingestion/src/metadata/ingestion/source/pipeline/kafkaconnect/client.py b/ingestion/src/metadata/ingestion/source/pipeline/kafkaconnect/client.py index e3a414b0e491..3c2b266d004e 100644 --- a/ingestion/src/metadata/ingestion/source/pipeline/kafkaconnect/client.py +++ b/ingestion/src/metadata/ingestion/source/pipeline/kafkaconnect/client.py @@ -61,15 +61,18 @@ def get_cluster_info(self) -> Optional[dict]: """ Get the version and other details of the Kafka Connect cluster. """ - try: - result = self.client.get_cluster_info() - return result - - except Exception as exc: - logger.debug(traceback.format_exc()) - logger.error(f"Unable to get cluster info :{exc}") + return self.client.get_cluster_info() - return None + def get_connectors_list( + self, + expand: str = None, + pattern: str = None, + state: str = None, + ) -> dict: + """ + Get the list of connectors from Kafka Connect cluster. + """ + return self.client.list_connectors(expand=expand, pattern=pattern, state=state) def get_connectors( self, @@ -85,12 +88,8 @@ def get_connectors( pattern (str): Only list connectors that match the regex pattern. state (str): Only list connectors that match the state. """ - try: - result = self.client.list_connectors( - expand=expand, pattern=pattern, state=state - ) - return result + return self.get_connectors_list(expand=expand, pattern=pattern, state=state) except Exception as exc: logger.debug(traceback.format_exc()) logger.error(f"Unable to get connectors list {exc}") diff --git a/ingestion/src/metadata/ingestion/source/pipeline/kafkaconnect/connection.py b/ingestion/src/metadata/ingestion/source/pipeline/kafkaconnect/connection.py index 38ea63920f0b..268b69bbac0f 100644 --- a/ingestion/src/metadata/ingestion/source/pipeline/kafkaconnect/connection.py +++ b/ingestion/src/metadata/ingestion/source/pipeline/kafkaconnect/connection.py @@ -51,7 +51,7 @@ def test_connection( test_fn = { "GetClusterInfo": client.get_cluster_info, - "GetPipelines": client.get_connectors, + "GetPipelines": client.get_connectors_list, "GetPlugins": client.get_connector_plugins, }