基于库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);

         }
     }
 }