def __init__(self, verbose=True):
self.VERBOSE = verbose
self.masters = []
self.is_master = False
self.path = None
self.zk = ZKClient(self.ZK_HOST, timeout=self.TIMEOUT)
self.say("login ok!")
# init
self.__init_zk()
# register
self.register()
def __init__(self,
verbose=True,
database='test',
table='targets',
zkhost='127.0.0.1:32181'):
self.VERBOSE = verbose
self.path = None
self.ZK_HOST = zkhost
self.database = "/" + database
self.table = os.path.join(self.database, table)
self.zk = ZKClient(self.ZK_HOST, timeout=self.TIMEOUT)
self.say("login ok")
self._init_zk()
self.value = value
def __str__(self):
return repr(self.value)
if __name__ == '__main__':
servers = options.servers.split(",")
# create all the sessions first to ensure that all servers are
# at least available & quorum has been formed. otw this will
# fail right away (before we start creating nodes)
sessions = []
# create one session to each of the servers in the ensemble
for i, server in enumerate(servers):
sessions.append(ZKClient(server, options.timeout))
# create on first server
zk = ZKClient(servers[0], options.timeout)
rootpath = "/zk-smoketest"
if zk.exists(rootpath):
if not options.quiet:
print("Node %s already exists, attempting reuse" % (rootpath))
else:
zk.create(
rootpath, "smoketest root, delete after test done, created %s" %
(datetime.datetime.now().ctime()))
# make sure previous cleaned up properly
pool.join()
print("Duration: ", str(time.time() - startTime))
print("Total writes: ", total_writes)
if __name__ == '__main__':
data = options.znode_size * "x"
servers = options.servers.split(",")
# create all the sessions first to ensure that all servers are
# at least available & quorum has been formed. otw this will
# fail right away (before we start creating nodes)
sessions = []
# create sessions to each of the servers in the ensemble
for server in servers:
for i in range(SESSIONS_NUM):
sessions.append(ZKClient(server, options.timeout))
# ensure root_znode doesn't exist
if sessions[0].exists(options.root_znode) and options.type == "create":
# unless forced by user
if not options.force:
raise SmokeError("Node %s already exists!" % (options.root_znode))
children = sessions[0].get_children(options.root_znode)
for child in children:
sessions[0].delete("%s/%s" % (options.root_znode, child))
elif options.type == "create":
sessions[0].create(
options.root_znode,
"smoketest root, delete after test done, created %s" %
(datetime.datetime.now().ctime()))
def help(err_info=""):
info="""
usage: sh rm_rf.sh -s 127.0.0.1:2181 -p /node_path"
"""
print err_info+info
def rm_rf(path):
if(path==None):
help("路径为空\n")
sys.exit(1)
if(not zk.exists(path)):
print "path(%s) is not exists!"%(path)
sys.exit(0)
the_childs=zk.get_children(path)
for a_child in the_childs:
rm_rf(path+"/"+a_child)
zk.delete(path)
pass
if __name__ == '__main__':
help()
if(options.servers is not None):
zk=ZKClient(options.servers, options.timeout)
rm_rf(options.path)
print "rm_rf the node (%s)done!"%(options.path)
else:
help("zookeeper服务为空!\n")
sys.exit(1)
return sum(summary[summary_key]) / len(summary[summary_key])
if __name__ == '__main__':
data = options.znode_size * "x"
servers = get_zk_servers(options.configfile)
# create all the sessions first to ensure that all servers are
# at least available & quorum has been formed. otw this will
# fail right away (before we start creating nodes)
sessions = []
server_id = 0
# create one session to each of the servers in the ensemble
for server in servers:
sessions.append(
ZKClient(server, options.timeout, ssl_server_cert_id=server_id))
server_id += 1
# ensure root_znode doesn't exist
if sessions[0].exists(options.root_znode):
# unless forced by user
if not options.force:
raise SmokeError("Node %s already exists!" % (options.root_znode))
children = sessions[0].get_children(options.root_znode)
for child in children:
sessions[0].delete("%s/%s" % (options.root_znode, child))
else:
sessions[0].create(
options.root_znode,
"smoketest root, delete after test done, created %s" %
import zkclient
from zkclient import ZKClient
zk = ZKClient("10.187.176.194:2181")
newKeyList = [
"goods_homef~~~brand", "qrqm_hp_cfb~~~brand", "goods_homef~~~c3brand",
"qrqm_hp_cfb~~~c3brand_0", "goods_homef~~~c3", "qrqm_hp_cfb~~~c3",
"qrqm_hp_cfb~~~channelBrand_0", "qrqm_hp_cfb~~~channelC3Brand_0",
"qrqm_hp_cfb~~~channelC3_0", "qrqm_hp_cfb~~~channel", "cd3_fzxp_fs",
"cd3_jdjx_fs", "cd3_jdmarket_fs", "cd3_newsh_fs", "cd3_oversea_fs",
"cd3_7toreturn_fs", "seckill_cfb~~~brand_code",
"seckill_cfb~~~cid3_discount_ratio_cat", "seckill_cfb~~~cid3",
"seckill_cfb~~~cid3_pw3_ids", "seckill_cfb~~~discount_amount_cat",
"seckill_cfb~~~discount_ratio_cat"
]
created = zk.exists("/root/redis")
if created:
print 'deleting /root/redis ...'
oldKeyList = zk.get_children("/root/redis")
for k in oldKeyList:
zk.delete("/root/redis/" + k)
else:
zk.create("/root/redis", "")
for k in newKeyList:
zk.create("/root/redis/" + k, k)
print 'add keys on zk success!'
cb.waitForSuccess()
timer2(func, "deleted %7d permanent znodes " % (options.znode_count))
def get_zk_servers():
return options.cluster
if __name__ == '__main__':
data = options.znode_size * "x"
servers = get_zk_servers()
session_gets = []
for x in xrange(options.watch_session):
session_gets.append(ZKClient(servers, options.timeout))
session_set = ZKClient(servers, options.timeout)
session_set.create(
options.root_znode,
"smoketest root, delete after test done, created %s" %
(datetime.datetime.now().ctime()))
create_nodes(session_set)
for s in session_gets:
add_wathches(s)
set_nodes(session_set, data)
delete_nodes(session_set)
session_set.delete(options.root_znode)
time.sleep(10)
