Tooz 使用实例

本文主要参考Tooz中官网中的例子: http://docs.openstack.org/developer/tooz/，为了更好的展示功能，对部分代码进行了修改。

本文采用Zookeeper作为Tooz的driver， 如果希望了解Tooz，请查看我翻译的一篇博文： 通过Tooz在python中实现分布式群组和锁管理

安装和启动Zookeeper:

1.  在ubuntu中直接采用apt的方式安装Zookeeper: # sudo apt-get install zookeeper

2. 启动Zookeeper的方式： # /usr/share/zookeeper/bin/zkServer.sh start

接下来就可以在Tooz中使用Zookeeper driver。

Tooz中的群组管理功能：

1.  建立群组管理的watcher:

#! /usr/bin/env python

import sys
import uuid
import six
import time

from tooz import coordination

coordinator = coordination.get_coordinator('kazoo://localhost:2181',b'host-1')
coordinator.start()

# Create a group
group = six.binary_type(six.text_type(uuid.uuid4()).encode('ascii'))
print group
coordinator.create_group(group).get()

# Watch join group
def group_joined(event):
    print(event.member_id + ' joined group ' + event.group_id)

# Watch leave group
def group_leaved(event):
    print(event.member_id + ' leaved group ' + event.group_id)

coordinator.watch_join_group(group, group_joined)
coordinator.watch_leave_group(group, group_leaved)

try:
    while True:
        coordinator.heartbeat()
        coordinator.run_watchers()
        time.sleep(1)
except KeyboardInterrupt:
    coordinator.stop()

这段代码中创建了一个group，并把group的id输出，同时监听了该group成员的加入和离开事件并输出。 

需要不断的调用coordinator.heartbeat() 来保持和coordination 服务器的连接。

需要不断的调用coordinator.run_watchers()来执行成员加入和离开监听操作。

2. 新建一个成员加入这个群组。

#! /usr/bin/env python
import sys
import uuid
import six
import time

from tooz import coordination

coordinator = coordination.get_coordinator('kazoo://localhost:2181',b'host-2')
coordinator.start()

# Create a group
group = sys.argv[1:]
if group is None:
   print 'Please specify the group'
   sys.exit(1)

coordinator.join_group(group[0]).get()
coordinator.stop()

第一段代码的输入(group_id) 将作为第二段代码的输入以保证它们处在同一个group中。

其中coordination.get_coordinator()中的第二个参数标识了成员的id.

执行结果：

1.启动watcher

# python group_change_watcher.py 
16551711-b1bb-4337-b300-45a0540dc0fe

2. 添加成员

# python join_group.py 16551711-b1bb-4337-b300-45a0540dc0fe

3. watcher的输出

# python group_change_watcher.py 
16551711-b1bb-4337-b300-45a0540dc0fe
host-2 joined group 16551711-b1bb-4337-b300-45a0540dc0fe
host-2 leaved group 16551711-b1bb-4337-b300-45a0540dc0fe

Tooz中的领导选取功能

这个功能文档中写的比较清楚，请参考这个文档：http://docs.openstack.org/developer/tooz/tutorial/leader_election.html

Tooz中的锁机制

Tooz中提供了分布式的锁机制。一个锁可以由一个名字代表，并且一个锁只能同时被一个节点拥有。

1. 建立锁

#! /usr/bin/env python

import time

from tooz import coordination

coordinator = coordination.get_coordinator('kazoo://localhost:2181', b'host-1')
coordinator.start()

lock = coordinator.get_lock("test")
try:
    with lock:
        while True:
            print "Locked in test"
            time.sleep(1)
except KeyboardInterrupt:
    coordinator.stop()

其中由“test” 来代表这个锁。

2. 申请锁

#! /usr/bin/env python

from tooz import coordination

coordinator = coordination.get_coordinator('kazoo://localhost:2181', b'host-2')
coordinator.start()

lock = coordinator.get_lock("test")
if lock.acquire(1000):
    print "Lock released"
    coordinator.stop()

其中lock.acquire(1000) 表示申请锁，并在1000s之后超时。

执行结果

1. 建立锁

# python lock_server.py 
Locked in test
Locked in test
Locked in test
....

2. 申请锁

# python lock_client.py

3.  ctrl+c 第一个进程，lock_client会输出"Lock released"