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"