Zookeeper詳解（十）：Python鏈接和操做Zookeeper

Python對Zookeeper的基本操做

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
from kazoo.client import KazooClient


def main():
    try:

        nodePath = "/zktest"
        host = "172.16.48.171"
        port = "2181"
        timeout = 100
        zkc = KazooClient(hosts=host + ':' + port, timeout=timeout)
        zkc.start()

        # 判斷節點是否存在
        if zkc.exists(nodePath+"/test111"):
            print nodePath + "/test111", "存在"
        else:
            # 創建節點，成功後返回新節點路徑
            childrenPath = zkc.create(nodePath+"/test111", "test111")
            print "建立節點：", childrenPath, "成功。"
            # 建立臨時節點，鏈接斷開則節點自動刪除
            zkc.create(nodePath+"/test999", "test999", ephemeral=True)

        # 獲取節點數據和節點數據，返回2個值，一個是節點數據，一個是節點stat，這是個ZnodeStat對象，它實際上是節點屬性，一共有12個屬性
        dataAndStat = zkc.get(nodePath)
        data = dataAndStat[0]
        print "數據爲：", data
        stat = dataAndStat[1]
        print "數據版本號爲：", stat.version
        print "數據長度爲：", stat.data_length
        print "子節點數量：", stat.numChildren

        # 更新節點數據,數據最大爲1MB超過則報錯，成功後返回 ZnodeStat 對象
        stat = zkc.set(nodePath, value="test222")
        print "數據版本號爲：", stat.version

        # 刪除節點，後面的參數用於控制是否遞歸刪除，默認是False,可是這樣就會有一個問題，若是該節點下有子節點則本次刪除失敗，你須要先刪除
        # 它下面的全部子節點才行
        if zkc.exists(nodePath+"/test111"):
            result = zkc.delete(nodePath+"/test111", recursive=False)
            if result:
                print "刪除節點成功。"

        print nodePath + " 的子節點爲：", zkc.get_children(nodePath)

        zkc.close()
        zkc.stop()
    except Exception as err:
        print err.message


if __name__ == "__main__":
    try:
        main()
    finally:
        sys.exit()

Watcher事件操做

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import sys
import time
from kazoo.client import KazooClient
from kazoo.client import ChildrenWatch
from kazoo.client import DataWatch

"""
Watcher能夠經過兩種方式設置，一種是在調用ZK客戶端方法的時候傳遞進去，好比 zk.get_children("/node", watch=FUN),可是這種方法是一次性的
也就是觸發一次就沒了，若是你還想繼續監聽一個事件就須要再次註冊。
另一種方法是經過高級API實現，監控數據或者節點變化，它只須要咱們註冊一次。一次性事件關注是zookeeper默認的即使在JAVA客戶端裏也是，這種高級別
API在JAVA裏是zkclient，而在Python裏面就是kazoo。高級API實際上是對低級API的封裝，對用戶來說更加好用。
"""

__metaclass__ = type


class zkWatcherTest:

    def __init__(self, host, port, timeout=10):
        self._nodename = ''
        self._host = host
        self._port = port
        self._timeout = timeout
        self._zk = KazooClient(hosts=self._host + ':' + self._port, timeout=self._timeout)
        self._zk.start()
        self._lastNodeList = []

    def start(self, zkPath):
        self._lastNodeList = self._zk.get_children(zkPath)
        try:
            ChildrenWatch(client=self._zk, path=zkPath, func=self._NodeChange)

            DataWatch(client=self._zk, path=zkPath, func=self._DataChange)
            # 這裏的死循環就是爲了避免讓程序退出，你能夠把時間設置長一點觀察，其實即使沒有到60秒的睡眠時間，若是
            # 子節點或者節點數量發生變化也會收到通知。這裏的wathch底層就是在節點上設置監聽器，而後捕捉事件，若是有
            # 事件觸發就調用你傳遞的方法來處理。
            while True:
                time.sleep(60)
                print "OK"
        except Exception as err:
            print err.message

    def _NodeChange(self, children):
        """
        處理子節點變化
        :param children: 這個參數並不須要你傳遞進來，由於把這個方法傳遞進ChiledrenWatcher，會返回一個當前子節點列表
        :return:
        """
        # print children
        # 若是新節點列表長度大於上次獲取的節點列表長度，說明有增長
        if len(children) > len(self._lastNodeList):
            for node in children:
                if node not in self._lastNodeList:
                    print "新增長的節點爲：", str(node)
                    self._lastNodeList = children
        else:
            for node in self._lastNodeList:
                if node not in children:
                    print "刪除的節點爲：", str(node)
                    self._lastNodeList = children

    def _DataChange(self, data, stat):
        """
        處理節點的數據變化
        :param data:
        :param stat:
        :return:
        """
        print "數據發生變化"
        print "數據爲：", data
        print "數據長度：", stat.dataLength
        print "數據版本號version：", stat.version
        print "cversion：", stat.cversion
        print "子節點數量：", stat.numChildren


def main():
    try:
        zkwt =zkWatcherTest(host="172.16.48.171", port="2181")
        zkwt.start("/zktest")
    except Exception as err:
        print err.message


if __name__ == "__main__":
    try:
        main()
    finally:
        sys.exit()

關於Watcher，網上不少帖子都是經過裝飾器的方式實現的，其實我上面的方式和裝飾器是同樣的，只是形式不一樣罷了。功能都能實現，只是用裝飾器有時候會不方便。