diff options
author | Himanshu Mishra <sigmablazer@gmail.com> | 2020-08-23 03:26:21 +0530 |
---|---|---|
committer | Jeff Widman <jeff@jeffwidman.com> | 2020-09-16 18:02:37 -0700 |
commit | b32f369e5a4b9914e92b4818b4d1bc6152264ec5 (patch) | |
tree | c8c67b9e5db4fae0edd63e7c7bfc6fbb1de985f6 | |
parent | d09ff02bbc19a8b762be67a294e0da1bfdff8029 (diff) | |
download | kafka-python-b32f369e5a4b9914e92b4818b4d1bc6152264ec5.tar.gz |
Allow configurable timeouts in admin client check version
Currently there's no way to pass timeout to check_version if called from admin.
-rw-r--r-- | kafka/admin/client.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/kafka/admin/client.py b/kafka/admin/client.py index e820587..1b91e1b 100644 --- a/kafka/admin/client.py +++ b/kafka/admin/client.py @@ -206,7 +206,7 @@ class KafkaAdminClient(object): self._client = KafkaClient(metrics=self._metrics, metric_group_prefix='admin', **self.config) - self._client.check_version() + self._client.check_version(timeout=(self.config['api_version_auto_timeout_ms'] / 1000)) # Get auto-discovered version from client if necessary if self.config['api_version'] is None: @@ -273,7 +273,7 @@ class KafkaAdminClient(object): response = future.value controller_id = response.controller_id # verify the controller is new enough to support our requests - controller_version = self._client.check_version(controller_id) + controller_version = self._client.check_version(controller_id, timeout=(self.config['api_version_auto_timeout_ms'] / 1000)) if controller_version < (0, 10, 0): raise IncompatibleBrokerVersion( "The controller appears to be running Kafka {}. KafkaAdminClient requires brokers >= 0.10.0.0." |