diff options
Diffstat (limited to 'tooz/drivers/zookeeper.py')
-rw-r--r-- | tooz/drivers/zookeeper.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tooz/drivers/zookeeper.py b/tooz/drivers/zookeeper.py index 7ea8e58..9b110e1 100644 --- a/tooz/drivers/zookeeper.py +++ b/tooz/drivers/zookeeper.py @@ -207,9 +207,13 @@ class KazooDriver(BaseZooKeeperDriver): def __init__(self, member_id, parsed_url, options): super(KazooDriver, self).__init__(member_id, parsed_url, options) - self._coord = client.KazooClient(hosts=parsed_url.netloc) + self._coord = self._make_client(parsed_url, options) self._member_id = member_id + @classmethod + def _make_client(cls, parsed_url, options): + return client.KazooClient(hosts=parsed_url.netloc) + def _watch_group(self, group_id): get_members_req = self.get_members(group_id) |