diff options
author | Dana Powers <dana.powers@gmail.com> | 2016-04-08 15:36:18 -0700 |
---|---|---|
committer | Dana Powers <dana.powers@gmail.com> | 2016-04-08 15:57:47 -0700 |
commit | 897ca399917baa178390af78870fe4be90c051d5 (patch) | |
tree | 88d8165f61e23344728aa5490395ad2e42076583 /kafka/client_async.py | |
parent | 85261e02e3b1dcaaa4816ef2cea90326352135f3 (diff) | |
download | kafka-python-conn_state_callback.tar.gz |
Add state_change_callback to bootstrap connectionconn_state_callback
Diffstat (limited to 'kafka/client_async.py')
-rw-r--r-- | kafka/client_async.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/kafka/client_async.py b/kafka/client_async.py index d0a3723..0c22f90 100644 --- a/kafka/client_async.py +++ b/kafka/client_async.py @@ -119,7 +119,10 @@ class KafkaClient(object): metadata_request = MetadataRequest[0]([]) for host, port, afi in hosts: log.debug("Attempting to bootstrap via node at %s:%s", host, port) - bootstrap = BrokerConnection(host, port, afi, **self.config) + cb = functools.partial(self._conn_state_change, 'bootstrap') + bootstrap = BrokerConnection(host, port, afi, + state_change_callback=cb, + **self.config) bootstrap.connect() while bootstrap.connecting(): bootstrap.connect() |