Zookeeper(3)-使用ZooKeeper作为配置中心

ZooKeeper作为配置中心

现在我们大多数应用都是采用的是分布式开发的应用，搭建到不同的服务器上，我们的配置文件，同一个应用程序的配置文件一样，还有就是多个程序存在相同的配置。当我们配置文件中有个配置属性需要改变，我们需要改变每个程序的配置属性，这样会很麻烦的去修改配置。而现在可以使用SpringCloud提供的配置中心，或者使用zookeeper来实现配置中心。

如何使用Zookeeper作为配置中心？

我们可以清除的了解ZooKeeper的Znode节点提供了一个Watcher，我们可以通过该方法来实现配置改变，来通知所有客户端配置的变化，并赋值。

如何存储配置信息呢

其实我们可以把key作为znode的节点名称，value则可以使用znode的数据域来储存，从而达成一个key-value的形式来存储配置信息。

下面是Zookeeper作为配置中心的代码

1. ConfigServer主要是对配置的读取，配置信息的写入，获取全部配置信息

import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooDefs;
import org.apache.zookeeper.data.Stat;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class ConfigServer extends ConnectWatcher{
    private String rootPath="/config";
    public ConfigServer(){
        this.connect();
    }
    //写入配置信息
    public void write(String key,String value){
        try {
            Stat stat=this.zooKeeper.exists(rootPath+"/"+key,false);
            if(stat!=null){
                this.zooKeeper.setData(rootPath+"/"+key,value.getBytes(),-1);
            }else{
                this.zooKeeper.create(rootPath+"/"+key,value.getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
            }
        } catch (KeeperException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
    public Map<String,String> getAllConfig(){
        Map<String,String> resultMap=new HashMap<String, String>();
        try {
            List<String> paths=this.zooKeeper.getChildren(rootPath,false);
            for (String path:paths
                 ) {
                System.out.println(path);
                String value=new String(this.zooKeeper.getData(rootPath+"/"+path,false,null));
                resultMap.put(path,value);
            }
        } catch (KeeperException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return resultMap;
    }
    public void createRoot(){
        try {
            this.zooKeeper.create("/config",null, ZooDefs.Ids.OPEN_ACL_UNSAFE,CreateMode.PERSISTENT);
        } catch (KeeperException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
    public String readConfig(String key, Watcher watcher){
        byte[] bytes=null;
        try {
            bytes=this.zooKeeper.getData(rootPath+"/"+key,watcher,null);
        } catch (KeeperException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        return new String(bytes);
    }

}

1. configClient主要是相当于我们每台服务器上对于配置中心的配置属性的监控

import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;

import java.util.Iterator;
import java.util.Map;
import java.util.concurrent.CountDownLatch;

public class ConfigClient implements Watcher {
    private ConfigServer configServer=new ConfigServer();
    private CountDownLatch countDownLatch=new CountDownLatch(1);
    public void process(WatchedEvent watchedEvent) {
        String path=watchedEvent.getPath();
        if(watchedEvent.getType()== Event.EventType.NodeDataChanged){
            this.updateConfig(path);
        }
    }
    public void updateConfig(String path){

        System.out.println(configServer.readConfig(path.substring(path.lastIndexOf("/")+1),this));
    }
    public void getAllConfig(){
        Map<String,String> map=configServer.getAllConfig();
        Iterator<String> keys=map.keySet().iterator();
        while (keys.hasNext()){
            String key=keys.next();
            String value=map.get(key);
            System.out.println("key:"+key+"  value:"+value);
        }
    }

}

1. connectWatcher主要是用来创建ZK的连接及关闭

import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooDefs;
import org.apache.zookeeper.ZooKeeper;

import java.io.IOException;
import java.util.concurrent.CountDownLatch;

public class ConnectWatcher  implements Watcher{
    private final static String ZK_HOST="47.106.132.60:2181,47.106.132.60:2182,47.106.132.60:2183";
    private final static int SESSION_TIME_OUT=3000;
    private CountDownLatch countDownLatch=new CountDownLatch(1);
    ZooKeeper zooKeeper;
    public void process(WatchedEvent watchedEvent) {
        if(watchedEvent.getState()== Event.KeeperState.SyncConnected){
            countDownLatch.countDown();
        }
    }
    public void connect(){
        try {
            zooKeeper=new ZooKeeper(ZK_HOST,SESSION_TIME_OUT,this);
            countDownLatch.await();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    public void close(){
        try {
            this.zooKeeper.close();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

1. 测试类

import org.junit.Test;

import java.util.concurrent.CountDownLatch;

public class Main {
    public static void main(String[] args) {
        CountDownLatch countDownLatch=new CountDownLatch(1);
        ConfigClient configClient=new ConfigClient();
        configClient.updateConfig("jdbc.url");
        try {
            countDownLatch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
    @Test
    public void addConfig(){
        ConfigServer configServer=new ConfigServer();
        configServer.createRoot();
        configServer.write("jdbc.url","com.mysql.jdbc.Driver");
        configServer.write("jdbc.password","1234");
        configServer.close();
    }
    @Test
    public void updateConfig(){
        ConfigServer configServer=new ConfigServer();
        configServer.write("jdbc.url","com.mysql.jdbc.update.test2");
        configServer.close();
    }
}