def wrapper(self, *args, **kwargs): if self.zk is None: self.zk = zk.ZooKeeper(config=self.config) try: self.zk.connect() func(self, *args, **kwargs) finally: self.zk.disconnect()
def get_zk_cli(): global GLOBAL_ZK if not GLOBAL_ZK: cfg = configparser.ConfigParser() cfg.read(ZK_CLI_CONF) zk_cli = zk.ZooKeeper(cfg) zk_cli.connect() GLOBAL_ZK = zk_cli return GLOBAL_ZK
def __init__(self, config_file): zk_cfg = configparser.ConfigParser() zk_cfg.read(config_file) self.zk_client = zk.ZooKeeper(zk_cfg) self.cluster_config = None self.github = None