diff --git a/ch_backup/config.py b/ch_backup/config.py index 84c6c8ac..3774e3ec 100644 --- a/ch_backup/config.py +++ b/ch_backup/config.py @@ -235,6 +235,7 @@ def _as_seconds(t: str) -> int: "cert": None, "key": None, "ca": None, + "timeout": 10, "connect_timeout": 10, "hosts": [], "root_path": "", diff --git a/ch_backup/zookeeper/zookeeper.py b/ch_backup/zookeeper/zookeeper.py index 054dd07e..3089ec70 100644 --- a/ch_backup/zookeeper/zookeeper.py +++ b/ch_backup/zookeeper/zookeeper.py @@ -23,6 +23,7 @@ class ZookeeperClient: def __init__(self, config: dict): self._client = KazooClient( config["hosts"], + timeout=config.get("timeout"), use_ssl=config.get("secure"), certfile=config.get("cert"), keyfile=config.get("key"),