Zookeeper的增删改查Java代码详解编程语言

package zookeeper; 
  
import java.io.IOException; 
import java.security.NoSuchAlgorithmException; 
import java.util.ArrayList; 
import java.util.List; 
  
import org.apache.zookeeper.CreateMode; 
import org.apache.zookeeper.WatchedEvent; 
import org.apache.zookeeper.Watcher; 
import org.apache.zookeeper.ZooDefs.Ids; 
import org.apache.zookeeper.ZooDefs.Perms; 
import org.apache.zookeeper.ZooKeeper; 
import org.apache.zookeeper.data.ACL; 
import org.apache.zookeeper.data.Id; 
import org.apache.zookeeper.server.auth.DigestAuthenticationProvider; 
  
public class ZookeeperClient implements Watcher { 
  
    private static class PropertiesDynLoading { 
  
        public static final String connectString = "192.168.57.128:9181"; 
        public static final int sessionTimeout = 3000; 
        public static final String authScheme = "digest"; 
        public static final String accessKey = "cache:svcctlg"; 
        public static final boolean authentication = false; 
    } 
  
    private ZooKeeper zk; 
  
    /** 
     * 创建zookeeper客户端 
     * 
     * @return 
     */ 
    private boolean createZkClient() { 
        try { 
            zk = new ZooKeeper(PropertiesDynLoading.connectString, PropertiesDynLoading.sessionTimeout, this); 
        } catch (IOException e) { 
            this.log("{}", e); 
            e.printStackTrace(); 
            return false; 
        } 
        if (PropertiesDynLoading.authentication) { 
            zk.addAuthInfo(PropertiesDynLoading.authScheme, PropertiesDynLoading.accessKey.getBytes()); 
        } 
        if (!isConnected()) { 
            log(" ZooKeeper client state [{}]", zk.getState().toString()); 
        } 
        try { 
            if (zk.exists("/zookeeper", false) != null) { 
                log("create ZooKeeper Client Success! connectString", PropertiesDynLoading.connectString); 
                log(" ZooKeeper client state [{}]", zk.getState()); 
                return true; 
            } 
        } catch (Exception e) { 
            this.log("create ZooKeeper Client Fail! connectString", PropertiesDynLoading.connectString); 
            e.printStackTrace(); 
        } 
        return false; 
    } 
  
    /** 
     * 新增持久化节点 
     * 
     * @param path 
     *            节点路径 
     * @param data 
     *            节点数据 
     * @return 
     */ 
    private boolean createPersistentNode(String path, String data) { 
        if (isConnected()) { 
  
            try { 
                if (PropertiesDynLoading.authentication) { 
                    zk.create(path, data.getBytes(), getAdminAcls(), CreateMode.PERSISTENT); 
                } else { 
                    zk.create(path, data.getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); 
  
                } 
                return true; 
            } catch (Exception e) { 
                e.printStackTrace(); 
                log("{}", e); 
                return false; 
            } 
  
        } 
        this.log("zookeeper state", zk.getState()); 
        return false; 
    } 
  
    /** 
     * 创建瞬时节点 
     * 
     * @param path 
     * @param data 
     * @return 
     */ 
    private boolean creatEphemeralNode(String path, String data) { 
        if (isConnected()) { 
  
            try { 
                if (PropertiesDynLoading.authentication) { 
                    zk.create(path, data.getBytes(), getAdminAcls(), CreateMode.PERSISTENT); 
                } else { 
                    zk.create(path, data.getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL); 
                } 
                return true; 
            } catch (Exception e) { 
                e.printStackTrace(); 
                log("{}", e); 
                return false; 
            } 
  
        } 
        this.log("zookeeper state", zk.getState()); 
        return false; 
    } 
  
    /** 
     * 修改数据 
     * 
     * @param path 
     * @param data 
     * @return 
     */ 
    private boolean setNodeData(String path, String data) { 
        if (isConnected()) { 
            try { 
                zk.setData(path, data.getBytes(), -1); 
                return true; 
            } catch (Exception e) { 
                e.printStackTrace(); 
                this.log("{}", e); 
                return false; 
            } 
        } 
        this.log("zookeeper state = [{}]", zk.getState()); 
        return false; 
    } 
  
    /** 
     * 删除节点 
     * 
     * @param path 
     * @return 
     */ 
    private boolean deleteNode(String path) { 
        if (isConnected()) { 
            try { 
                zk.delete(path, -1); 
                return true; 
            } catch (Exception e) { 
                e.printStackTrace(); 
                this.log("{}", e); 
                return false; 
            } 
        } 
        this.log("zookeeper state = [{}]", zk.getState()); 
        return false; 
    } 
  
    /** 
     * 获取节点值 
     * 
     * @param path 
     * @return 
     */ 
    public String getNodeData(String path) { 
        if (isConnected()) { 
            String data = null; 
            try { 
                byte[] byteData = zk.getData(path, true, null); 
                data = new String(byteData, "utf-8"); 
                return data; 
            } catch (Exception e) { 
  
                e.printStackTrace(); 
                this.log("{}", e); 
                return null; 
            } 
        } 
        this.log("zookeeper state = [{}]", zk.getState()); 
        return null; 
    } 
  
    /** 
     * 获取path子节点名列表 
     * 
     * @param path 
     * @return 
     */ 
    public List<String> getChildren(String path) { 
        if (isConnected()) { 
            String data = null; 
            try { 
                return zk.getChildren(path, false); 
            } catch (Exception e) { 
                e.printStackTrace(); 
                this.log("{}", e); 
                return null; 
            } 
        } 
        this.log("zookeeper state = [{}]", zk.getState()); 
        return null; 
    } 
  
    public boolean startZkClient() { 
        return createZkClient(); 
    } 
  
    /** 
     * zookeeper是否连接服务器 
     * 
     * @return 
     */ 
    public boolean isConnected() { 
        return zk.getState().isConnected(); 
    } 
  
    /** 
     * 是否存在path路径节点 
     * 
     * @param path 
     * @return 
     */ 
    public boolean exists(String path) { 
        try { 
            return zk.exists(path, false) != null; 
        } catch (Exception e) { 
  
            this.log("{}", e); 
        } 
        return false; 
    } 
  
    /** 
     * 关闭zookeeper 
     */ 
    public void closeZk() { 
        if (isConnected()) { 
            try { 
                zk.close(); 
                this.log("close zookeeper [{}]", "success"); 
            } catch (InterruptedException e) { 
                this.log("zookeeper state = [{}]", e); 
                e.printStackTrace(); 
            } 
        } else { 
            this.log("zookeeper state = [{}]", zk.getState()); 
        } 
  
    } 
  
    /** 
     * 
     * @return 
     */ 
    public List<ACL> getCreateNodeAcls() { 
        List<ACL> listAcls = new ArrayList<ACL>(3); 
        try { 
            Id id = new Id(PropertiesDynLoading.authScheme, 
                    DigestAuthenticationProvider.generateDigest(PropertiesDynLoading.accessKey)); 
            ACL acl = new ACL(Perms.CREATE, id); 
            listAcls.add(acl); 
  
        } catch (NoSuchAlgorithmException e) { 
  
            e.printStackTrace(); 
            return Ids.OPEN_ACL_UNSAFE; 
        } 
        return listAcls; 
    } 
  
    public List<ACL> getAdminAcls() { 
        List<ACL> listAcls = new ArrayList<ACL>(3); 
        try { 
            Id id = new Id(PropertiesDynLoading.authScheme, 
                    DigestAuthenticationProvider.generateDigest(PropertiesDynLoading.accessKey)); 
            ACL acl = new ACL(Perms.ALL, id); 
            listAcls.add(acl); 
  
        } catch (NoSuchAlgorithmException e) { 
  
            e.printStackTrace(); 
            return Ids.OPEN_ACL_UNSAFE; 
        } 
        return listAcls; 
    } 
  
    public void log(String format, Object args) { 
        int index = format.indexOf("{"); 
        StringBuilder sb = new StringBuilder(format); 
        sb.insert(index + 1, "%s"); 
        System.out.println(String.format(sb.toString(), args)); 
    } 
  
    @Override 
    public void process(WatchedEvent event) { 
    } 
  
    public static void main(String[] args) { 
  
        ZookeeperClient zkc = new ZookeeperClient(); 
        zkc.createZkClient(); 
  
        if (!zkc.exists("/windowcreate")) { 
  
            zkc.createPersistentNode("/windowcreate", "windowcreate"); 
        } 
        if (!zkc.exists("/windowcreate/value")) { 
            System.out.println("not exists /windowcreate/value"); 
  
            zkc.createPersistentNode("/windowcreate/value", "A0431P001"); 
        } 
        if (!zkc.exists("/windowcreate/valuetmp")) { 
            System.out.println("not exists /windowcreate/valuetmp"); 
            zkc.creatEphemeralNode("/windowcreate/valuetmp", "A0431P002"); 
        } 
        System.out.println(zkc.getNodeData("/zookeeper")); 
        System.out.println(zkc.getChildren("/windowcreate")); 
        System.out.println(zkc.getNodeData("/windowcreate/value")); 
        System.out.println(zkc.getNodeData("/windowcreate/valuetmp")); 
        zkc.setNodeData("/windowcreate/value", "A0431P003"); 
        System.out.println(zkc.getNodeData("/windowcreate/value")); 
        zkc.deleteNode("/windowcreate/value"); 
        System.out.println(zkc.exists("/windowcreate/value")); 
        zkc.closeZk(); 
    } 
  
}