def get_data_from_zk(zk_path, mode): kazoo_client = KazooClientManager(ZK_HOSTS).get_client() if not kazoo_client.exists(zk_path): log.error("Path %s does not exist" % (zk_path)) exit(_ZK_PATH_NOT_EXIST_CODE) try: if mode == "CONFIG": value, stat = kazoo_client.get(zk_path) return value, stat.version elif mode == 'SERVERSET': value = _get_hosts_from_zk(kazoo_client, zk_path) return value, None else: log.error("Unrecognized mode: %s" % mode) exit(_UNRECOGNIZED_MODE_CODE) except: log.error("Failed to get value from zookeeper path %s with mode %s" % (zk_path, mode)) exit(_FAILED_TO_GET_DATA_FROM_ZK_CODE)
def get_data_from_zk(zk_path, mode): kazoo_client = KazooClientManager(ZK_HOSTS).get_client() if not kazoo_client.exists(zk_path): log.error("Path %s does not exist" % (zk_path)) exit(_ZK_PATH_NOT_EXIST_CODE) try: if mode == "CONFIG": value, stat = kazoo_client.get(zk_path) return value, stat.version elif mode == 'SERVERSET': value = _get_hosts_from_zk(kazoo_client, zk_path) return value, None else: log.error("Unrecognized mode: %s" % mode) exit(_UNRECOGNIZED_MODE_CODE) except: log.error("Failed to get value from zookeeper path %s with mode %s" % (zk_path, mode)) exit(_FAILED_TO_GET_DATA_FROM_ZK_CODE)