tooz/examples/leader_election.py

29 lines
637 B
Python

from tooz import coordination
coordinator = coordination.get_coordinator('zookeeper', b'host-1')
coordinator.start()
# Create a group
request = coordinator.create_group(b"my group")
request.get()
# Join a group
request = coordinator.join_group(b"my group")
request.get()
def when_i_am_elected_leader(event):
# event is a LeaderElected event
print(event.group_id, event.member_id)
# Propose to be a leader for the group
coordinator.watch_elected_as_leader(b"my_group",
when_i_am_elected_leader)
while True:
coordinator.heartbeat()
coordinator.run_watchers()
coordinator.stop()