Minor: Fix kafkaconnect test connection (#18965)

This commit is contained in:
Suman Maharana 2024-12-09 16:52:35 +05:30 committed by GitHub
parent fa9ee99458
commit 85c7f6d22e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 13 additions and 14 deletions

View File

@ -61,15 +61,18 @@ class KafkaConnectClient:
"""
Get the version and other details of the Kafka Connect cluster.
"""
try:
result = self.client.get_cluster_info()
return result
return self.client.get_cluster_info()
except Exception as exc:
logger.debug(traceback.format_exc())
logger.error(f"Unable to get cluster info :{exc}")
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 @@ class KafkaConnectClient:
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}")

View File

@ -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,
}