基于库zkclient 的leader选举代码实现

利用了zookeeper临时节点，在当连接或session断掉时被删除这一特性来做选举。（简单简单互斥锁）

查了下网上的做法。

大致流程：

<1>判定是否存在/wzgtest路径

<2>如果不存在，那么试图创建一个节点znode（Ephemeral Path）（path = /wzgtest，data=client id）

<2.1>创建成功,标识自己是leader。

<2.2>创建不成功（包括异常）标识自己是slave。

<3>监听/wzgtest节点的数据变化。节点移除（zookeeper连接断开，session超时）时，重走<1>重新进行选举。

具体代码详见附件。

该代码只是实现，并未进行包装优化。

import java.net.InetAddress;
import java.net.UnknownHostException;

import org.I0Itec.zkclient.IZkDataListener;
import org.I0Itec.zkclient.ZkClient;

/**
 * @author 作者 王志刚
 * @version 创建时间：2016年8月18日 下午7:13:49
 * @Description:  zkleader选举
 */
public class ZKLeaderElect {

    public boolean isLeader = false;
    private ZkClient zkClient = null;
    private String path = "XXXXX";
    private String hostInfo = "initString";

    public void initZKLeaderElect(String zkServers, String path, String s) {
        zkClient = new ZkClient(zkServers, 10000, 10000);
        this.path = path;
        try {
            hostInfo = InetAddress.getLocalHost().toString();
        } catch (UnknownHostException e) {
        }
        hostInfo = s;
    }

    /**
    * 选举leader
    *
    */
    public void tryLeader() {
        if (!zkClient.exists(path)) {
            try {
                zkClient.createEphemeral(path, hostInfo);
                isLeader = true;
                System.out.println(hostInfo + "  成为leader了");
            } catch (Exception e) {
                System.out.println(hostInfo+"成为leader失败");
                e.printStackTrace();
                isLeader = false;
            }
        }
    }

    /**
    * 监听指定节点的数据变化
    *
    */
    public void testListener() throws InterruptedException {
        // 监听指定节点的数据变化
        zkClient.subscribeDataChanges(path, new IZkDataListener() {
            public void handleDataChange(String s, Object o) throws Exception {
//                System.out.println(hostInfo+"说：");
//                System.out.println("node data changed!");
//                System.out.println("node=>" + s);
//                System.out.println("data=>" + o);
//                System.out.println("--------------");
//                tryLeader();
            }

            public void handleDataDeleted(String s) throws Exception {
                System.out.println(hostInfo+"说：");
                System.out.println("node data deleted!");
                System.out.println("s=>" + s);
                System.out.println("--------------");
                tryLeader();
            }
        });

        System.out.println(hostInfo + " ----- ready!");

    }

    public static void main(String[] args) throws InterruptedException {
        ZKLeaderElect zk1 = new ZKLeaderElect();
        zk1.initZKLeaderElect("10.1.51.221:7605", "/wzgtest","1");

        ZKLeaderElect zk2 = new ZKLeaderElect();
        zk2.initZKLeaderElect("10.1.51.221:7605", "/wzgtest","2");

        ZKLeaderElect zk3 = new ZKLeaderElect();
        zk3.initZKLeaderElect("10.1.51.221:7605", "/wzgtest","3");

        zk1.tryLeader();
        zk2.tryLeader();
        zk3.tryLeader();
        zk1.testListener();
        zk2.testListener();
        zk3.testListener();
        int i = 0;
        // junit测试时，防止线程退出
        while (true) {
            Thread.sleep(1000);
            i++;
            if(i % 5 == 0){
                if(zk1.isLeader){
                    zk1.zkClient.close();
                    System.out.println("1 关闭了");
                    zk1.isLeader = false;
                }
                if(zk2.isLeader){
                    zk2.zkClient.close();
                    System.out.println("2 关闭了");
                    zk2.isLeader = false;
                }
                if(zk3.isLeader){
                    zk3.zkClient.close();
                    System.out.println("3 关闭了");
                    zk3.isLeader = false;
                }
            }
            System.out.println("1"+zk1.isLeader+"       2:"+zk2.isLeader+"   3:"+zk3.isLeader);
            
        }
    }
}