diff --git a/src/config/common/zkclient.py b/src/config/common/zkclient.py index 56fe706cd15..4965dfba6e4 100644 --- a/src/config/common/zkclient.py +++ b/src/config/common/zkclient.py @@ -220,7 +220,7 @@ def delete_all(cls, zookeeper_client, path): class ZookeeperClient(object): - def __init__(self, module, server_list, logging_fn=None): + def __init__(self, module, server_list, logging_fn=None, zk_timeout=400): # logging logger = logging.getLogger(module) logger.setLevel(logging.DEBUG) @@ -245,7 +245,7 @@ def __init__(self, module, server_list, logging_fn=None): sleep_func=gevent.sleep) self._zk_client = kazoo.client.KazooClient( server_list, - timeout=400, + timeout=zk_timeout, handler=kazoo.handlers.gevent.SequentialGeventHandler(), logger=logger, connection_retry=self._retry, diff --git a/src/config/schema-transformer/to_bgp.py b/src/config/schema-transformer/to_bgp.py index 6ce1b59f09f..16becd2be2b 100644 --- a/src/config/schema-transformer/to_bgp.py +++ b/src/config/schema-transformer/to_bgp.py @@ -363,6 +363,7 @@ def parse_args(args_str): --use_syslog --syslog_facility LOG_USER --cluster_id + --zk_timeout 400 [--reset_config] ''' @@ -409,6 +410,7 @@ def parse_args(args_str): 'kombu_ssl_keyfile': '', 'kombu_ssl_certfile': '', 'kombu_ssl_ca_certs': '', + 'zk_timeout': 400, } secopts = { 'use_certs': False, @@ -525,6 +527,8 @@ def parse_args(args_str): help="Start port for bgp-as-a-service proxy") parser.add_argument("--bgpaas_port_end", type=int, help="End port for bgp-as-a-service proxy") + parser.add_argument("--zk_timeout", + help="Timeout for ZookeeperClient") args = parser.parse_args(remaining_argv) if type(args.cassandra_server_list) is str: @@ -585,7 +589,8 @@ def main(args_str=None): else: client_pfx = '' zk_path_pfx = '' - _zookeeper_client = ZookeeperClient(client_pfx+"schema", args.zk_server_ip) + _zookeeper_client = ZookeeperClient(client_pfx+"schema", args.zk_server_ip, + zk_timeout =args.zk_timeout) _zookeeper_client.master_election(zk_path_pfx + "/schema-transformer", os.getpid(), run_schema_transformer, args)