ZooKeeper学习笔记三:使用ZooKeeper实现一个简单的配置中心

作者:Grey

原文地址:ZooKeeper学习笔记三:使用ZooKeeper实现一个简单的配置中心

前置知识

完成ZooKeeper集群搭建以及熟悉ZooKeeperAPI基本使用

需求

很多程序往往是通过心跳检测来判断配置的变更,通过zk的回调机制,我们可以实现比心跳更为快速的配置检测机制,包括配置的新增,修改和删除

流程也比较简单:
image

环境准备

一个zk集权,ip和端口分别为:

  • 192.168.205.145:2181
  • 192.168.205.146:2181
  • 192.168.205.147:2181
  • 192.168.205.148:2181

定义主方法

App.java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
public class App {
public static void main(String[] args) {
// 需要监控的路径是/AppConf
String path = "/AppConf";
while (true) {
ConfigCenter configCenter = new ConfigCenter(path);
String conf = configCenter.getConf();
// 配置不为空则拿到最新的配置
if (null != conf && !conf.trim().isEmpty()) {
System.out.println(conf);
}
// 睡眠一段时间
pending(1000);
}
}
}

zk初始化工具类

参考ZooKeeperAPI基本使用中的ZookeeperConfig类,主要的方法为:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
public class ZookeeperConfig {
private static final String ADDRESS = "192.168.205.145:2181,192.168.205.146:2181,192.168.205.147:2181,192.168.205.148:2181";
private static ZooKeeper zk;
static CountDownLatch latch;
// 获取一个zk客户端
public static ZooKeeper create() {
latch = new CountDownLatch(1);
try {
zk = new ZooKeeper(ADDRESS, 3000, new DefaultWatch());
latch.await();
} catch (IOException | InterruptedException e) {
e.printStackTrace();
}
return zk;
}
...
}

实现配置中心逻辑

配置中心的入口方法为:getConf()

1
2
3
4
public String getConf() {
aWait();
return value;
}

其中aWait()方法用于监听配置信息的变动(比如修改,删除,增加),并且通过CountDownLatch阻塞运行,zk一旦监听到配置信息的变动,即会触发回调,并执行countDown(),这样前面的CountDownLatch即可往下运行:

1
2
3
4
5
6
7
8
public void aWait() {
zk.exists(conf, this, this, "dasdfa");
try {
latch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
}

ConfigCenter类实现了Watcher, AsyncCallback.StatCallback, AsyncCallback.DataCallback三个接口,所以监听方法和回调方法都在ConfigCenter中实现,其中,监听方法:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
@Override
public void process(WatchedEvent event) {

Event.EventType type = event.getType();
switch (type) {
case None:
break;
case NodeCreated:
System.out.println("node created");
zk.getData(conf, this, this, "node created");
latch.countDown();
break;
case NodeDeleted:
try {
System.out.println("config deleted");
this.value = "";
latch = new CountDownLatch(1);
} catch (Exception e) {
e.printStackTrace();
}
break;
case NodeDataChanged:
System.out.println("node changed");
zk.getData(conf, this, this, "node changed");
latch.countDown();
break;
case NodeChildrenChanged:
break;
case DataWatchRemoved:
break;
case ChildWatchRemoved:
break;
case PersistentWatchRemoved:
break;
}
}

如上,当监听到节点创建和修改的时候,触发getData,当监听到节点删除的时候,重新初始化CountDownLatch,让线程阻塞。

回调方法如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
@Override
public void processResult(int rc, String path, Object ctx, Stat stat) {
if (stat != null) {
zk.getData(conf, this, this, "getData");
}
}

@Override
public void processResult(int rc, String path, Object ctx, byte[] data, Stat stat) {
if (data != null) {
this.value = new String(data);
latch.countDown();
}
}

方法1表示,当节点存在(stat!=null)的时候,获取一次数据(重新注册监听)。
方法2表示,当节点数据存在的时候,将节点数据取出,并且countDown(),解除阻塞。

运行效果

先把zk中的/AppConf节点删除,执行App.java,程序阻塞

通过zkCli.sh连接任意zk服务端,创建一个/AppConf节点,并赋值

1
2
3
[zk: localhost:2181(CONNECTED) 58] create /AppConf "hello"
Created /AppConf

控制台同时打印出:

1
2
3
4
5
6
7
8
9
node created
hello
hello
hello
hello
hello
hello
hello
hello

继续通过zkCli重新设置/AppConf节点的值

1
2
3
[zk: localhost:2181(CONNECTED) 59] set /AppConf "world"
[zk: localhost:2181(CONNECTED) 60]

配置信息立马生效,控制台打印

1
2
3
4
5
6
7
8
...
hello
hello
node changed
world
world
world
...

通过zkCli.sh删除/AppConf节点,控制台阻塞运行,并打印

1
config deleted

通过zkCli.sh重新创建/AppConf节点,

1
2
3
[zk: localhost:2181(CONNECTED) 61] create /AppConf "hello2"
Created /AppConf

控制台立即解除阻塞,并打印

1
2
3
4
5
config deleted
node created
hello2
hello2
..

完整代码

Github