本文主要参考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"