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)
