时间:2021-07-01 10:21:17 帮助过:17人阅读
需求: 检查一个zookeeper服务中的某些节点是否存在,如果存在返回OK,不存在则给出不存的节点信息。要求返回的信息是和pyunit的结果信息一致。
实现环境:
1. python 2.7 以及自带的pyunit
2. bottle 作为一个python的简易服务器
pip install bottle
3. kazoo 一个python的zookeeper客户端
pip install kazoo
1. 创建一个python的测试类 zk_check.py
-*- coding: utf-8 -*-
from kazoo.client import KazooClient
import unittest
class zktest(unittest.TestCase):
def runTest(self):
zknamespace = “/app/zktest_performance_1”
zkhosts = “127.0.0.1:2181,127.0.0.1:2182,127.0.0.1:2183”
ZKTEST_DRIVERS = [“ip1”, “ip2”]
ZKTEST_NODES = [“ip3”, “ip4”, “ip5”, “ip6”]
driverChildren = []
nodeChildren = []
badDrivers = []
badNodes = []
# checking
zk = KazooClient(hosts=zkhosts, read_only=True)
zk.start()
driverFatherPath = zknamespace + “/status/drivers”
nodeFatherPath = zknamespace + “/status/nodes”
if zk.exists(driverFatherPath):
driverChildren = zk.get_children(driverFatherPath)
if len(driverChildren) >
for driver in zktest_DRIVERS:
if driver not in driverChildren:
badDrivers.append(driver)
if zk.exists(nodeFatherPath):
nodeChildren = zk.get_children(nodeFatherPath)
if len(nodeChildren) >
for node in zktest_NODES:
if node not in nodeChildren:
badNodes.append(node)
zk.stop()
if (len(badNodes)==0) and (len(badDrivers)==0):
self.assertEquals(1,1,”pass”)
else:
if len(badDrivers) > 0:
self.assertEquals(1,2,'len : %d , error : %s' % (len(badDrivers),badDrivers))
if len(badNodes) > 0:
self.assertEquals(1,2,'len : %d , error : %s' % (len(badNodes),badNodes))
if __name__ == ‘__main__':
unittest.main()
2. 写一个bottle服务,将结果输出
import commands
from bottle import route, run, template
@route(‘/alisa')
def index():
command = “python /Users/metaboy/script/zk_check.py”
#output = os.popen(command)
return template(‘{{text}}', text=commands.getoutput(command))
run(host='localhost', port=8888)
3. 后台启动bottle服务,提供外部访问ip
现在可以直接通过 http://localhost:8888/alisa 进行访问。