首页 > 代码库 > Zookeeper实现分布式选举算法
Zookeeper实现分布式选举算法
分布式系统中经常采用Master/Slave架构。(截止到目前为止我还没有碰到过其他架构。。。)这种架构中如果Master发生故障就会导致整个集群停止服务,为了提高系统的高可用性通常采用选举算法来选出Master。这样Master如果出现故障,Slave经过选举算法重新选择Master。通过Zookeeper可以非常容易实现这个功能,关键代码如下:(完整代码见文章最后的GitHub地址)
//申请做 leader
String prefix = "/ticket-";
String myVote = zooKeeper.create(root + prefix, new byte[]{}, ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
//获取所有参选人
List<String> allVote = zooKeeper.getChildren(root, new Watcher() {…………});
//寻找最小的id,谁最小谁是leader
String minVotePath = allVote.get(0);
String minVote = fixForSorting(prefix, minVotePath);
for (String vote : allVote) {
String thisVoteId = fixForSorting(prefix, vote);
if (thisVoteId.compareTo(minVote) < 0) {
minVotePath = vote;
minVote = thisVoteId;
}
}
LOGGER.debug("当前leader {}", minVotePath);
选举过程非常简单,分为三步
在zookeeper上新增一个节点作为自己的选票(相当于自己指定自己做leader)。比如例子中我的root节点选择的是/ha,所有的服务器启动后都会在这个节点下新增一个ticket为前缀的新节点。这是一个比较特殊的节点,通过指定
EPHEMERAL_SEQUENTIAL
可以让Zookeeper帮我们给节点新增一串数字。(比如第一台启动的服务器得到的是ticket-0000000002、第二台得到的是ticket-0000000003)获取所有选票,(比如例子中通过getChildren方法获取/ha下面所有的节点)
比较选票,如果自己的选票是最小的,说明自己被选中做leader,(判断是否是leader的规则是id最小)否则就认为自己没有当选,等待节点变化迎接下次选举(通过Watcher对象)。
分析一下这个算法不难发现,如果有3台服务器启动,第一个向zookeeper“报告”的人会被当选为leader;如果它出现故障,第二个向zookeeper“报告”的人会被当选为leader,以此类推。这是一种非常原始的民主选举制度,有一个象征最高权力的“神器”,得到“神器”的就是大部落的酋长;很多人想要参选大酋长,那么谁跑得快最先抢到“神器”谁就是大酋长;如果在后面的“执政”期间酋长因为“太堕落”被干掉了那么第二名自动接管“神器”变成大酋长。把上面的代码执行两次,最先执行的程序会被选择为leader;杀死第一个进程,第二个进程的控制台会输出自己当选为leader的信息。(第二个进程不是立即输出信息,需要等待几秒钟)
Zookeeper Watcher
Watcher是一个接口,它的定义很简单
public interface Watcher {
abstract public void process(WatchedEvent event);
}
一个典型的Callback,当发生事件的时候(WatchedEvent
)由系统会调用process
方法。以Zookeeper Java Client为例,我们会在两个地方用到Watcher
对象
new ZooKeeper
的时候需要传一个Watcher
对象,在客户端连接到Zookeeper服务器或者断开连接或者Session过期的时候都会调用Watcher对象。这种情况下我们关注WatchedEvent
中的keeperState
成员变量,它是一个枚举类型,可以是:Disconnected
、SyncConnected
、AuthFailed
、ConnectedReadOnly
、Expired
、SaslAuthenticated
等。调用
getChildren
的时候需要传一个Watcher
对象,某个数据节点发送变化的时候服务器会推送消息给客户端,此时Watcher对象就会被调用。这种情况下我们关注eventType
成员变量,它是一个枚举类型,可以是:NodeCreated
、NodeDeleted
、NodeDataChanged
、NodeChildrenChanged
等。
总结下来,一种Watcher
用来监控Zookeeper的连接;一种Watcher
用来监控数据的变化。
注意
Leader和Zookeeper之间是有心跳数据的,时间间隔是sessionTimeout决定的。
EPHEMERAL_SEQUENTIAL
的含义有两个:EPHEMERAL
,表示节点是临时的,当zookeeper和客户端断开连接的时候节点会被阐述;SEQUENTIAL
,zookeeper会在节点最后加上一串数字后缀。所以重新选举的时间是实际上是由sessionTimeout决定的,zookeeper服务器探测到客户端断开后才会删除临时节点,推送变化,此时选举才会进行。
完整代码 https://gist.github.com/fireflyc/51d5467ef48b7f8c4a7747e5ecbd3fd0 (或者点击“原文链接”)
Zookeeper实现分布式选举算法