zookeeper LeaderElection(leader 选举)

package zookeeper;  
  
import org.apache.zookeeper.*;  
import org.apache.zookeeper.data.Stat;  
  
import java.net.InetAddress;  
import java.net.UnknownHostException;  
import java.util.List;  
import java.util.concurrent.CountDownLatch;  
  
/** 
 * Modify by bamboo.hey
 * User: EDENYIN 
 * Date: 1/4/11 
 * Time: 3:13 PM 
 * To change this template use File | Settings | File Templates. 
 */  
public class LeaderElection implements Watcher, Runnable {  
  
  
    private String zookeeperConnectionString;  
    private String rootPath;  
  
    private ZooKeeper zk;  
  
    private byte[] hostAddress;  
  
  
    public LeaderElection(String zookeeperConnectionString, String rootPath) {  
        this.zookeeperConnectionString = zookeeperConnectionString;  
        this.rootPath = rootPath;  
        try {  
            hostAddress = InetAddress.getLocalHost().getHostAddress().getBytes();  
        } catch (UnknownHostException e) {  
            e.printStackTrace();  
        }  
        buildZK();  
    }  
  
  
    private void buildZK() {  
        System.out.println("Build zk client");  
        try {  
            zk = new ZooKeeper(zookeeperConnectionString, 10000, this);  
            Stat s = zk.exists(rootPath, false);  
            if (s == null) {  
                zk.create(rootPath, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);  
                zk.create(rootPath + "/ELECTION", new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);  
            }  
            String value = zk.create(rootPath + "/ELECTION/n_", hostAddress, ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);  
        } catch (Exception e) {  
            e.printStackTrace();  
            System.err.println("Error connect to zoo keeper");  
        }  
    }  
  
  
    public void process(WatchedEvent event) {  
        System.out.println(event);  
        if (event.getState() == Event.KeeperState.Disconnected || event.getState() == Event.KeeperState.Expired) {  
            System.out.println("Zookeeper connection timeout.");  
            buildZK();  
        }  
  
    }  
  
    public void run() {  
            while (true) {  
                try {  
                    List<String> children = zk.getChildren(rootPath + "/ELECTION", false);  
                    String leaderPath = "Not found";  
                    int minValue = -1;  
                    for (int i=0;i<children.size();i++) {  
                        String child = children.get(i);  
                        int index = Integer.parseInt(child.substring(2));  
                        if (i == 0) {  
                            minValue = index;  
                            leaderPath = child;  
                        }else if (index < minValue) {  
                            minValue = index;  
                            leaderPath = child;  
                        }  
                    }  
                    LatchChildWatcher latchChildWatcher = new LatchChildWatcher();  
                    byte[] data = zk.getData(rootPath + "/ELECTION/" + leaderPath, latchChildWatcher, null);  
                    System.out.println("find the leader on the path:" + leaderPath + " whose host address is " + new String(data));  
                    latchChildWatcher.await();  
                    System.out.println("aaaaaaaaaaaaaaaa");
                } catch (Exception e) {  
                    e.printStackTrace();  
                    System.err.println("Error get the leader." + e.getMessage());  
                }  
        }  
    }  
  
    private class LatchChildWatcher implements Watcher {  
  
        CountDownLatch latch;  
  
        public LatchChildWatcher(){  
            latch = new CountDownLatch(1);  
        }  
  
        public void process(WatchedEvent event){  
            System.out.println("Watcher fired on path: " + event.getPath() + " state: " +  
                    event.getState() + " type " + event.getType());  
            latch.countDown();  
        }  
        public void await() throws InterruptedException {  
            latch.await();  
        }  
    }  
    public static void main(String[] args) 
    {
    String zookeeperConnectionString="inc-dw-hadoop-151-7:2181,inc-dw-hadoop-151-8:2181,inc-dw-hadoop-151-9:2181";
        String rootPath="/le";  
    LeaderElection le=new LeaderElection(zookeeperConnectionString,rootPath);
    le.run();
    }
}