Apache Dubbo 源码搭建与解读（九）—— ZooKeeper客户端

最新推荐文章于 2022-09-27 11:57:22 发布

DAIHAO丶

最新推荐文章于 2022-09-27 11:57:22 发布

阅读量121

点赞数

本文链接：https://blog.csdn.net/D_A_I_H_A_O/article/details/109099061

版权

Apache Dubbo 专栏收录该内容

11 篇文章 3 订阅

订阅专栏

ZooKeeper客户端

ZooKeeper 官方提供的客户端支持了一些基本操作，例如，创建会话、创建节点、读取节点、更新数据、删除节点和检查节点是否存在等，但在实际开发中只有这些简单功能是根本不够的。而且，ZooKeeper 本身的一些 API 也存在不足，例如：

ZooKeeper 的 Watcher 是一次性的，每次触发之后都需要重新进行注册。
会话超时之后，没有实现自动重连的机制。
ZooKeeper 提供了非常详细的异常，异常处理显得非常烦琐，对开发新手来说，非常不友好。
只提供了简单的 byte[] 数组的接口，没有提供基本类型以及对象级别的序列化。
创建节点时，如果节点存在抛出异常，需要自行检查节点是否存在。
删除节点就无法实现级联删除。

常见的第三方开源 ZooKeeper 客户端有 ZkClient 和 Apache Curator。

ZkClient 是在 ZooKeeper 原生 API 接口的基础上进行了包装，虽然 ZkClient 解决了 ZooKeeper 原生 API 接口的很多问题，提供了非常简洁的 API 接口，实现了会话超时自动重连的机制，解决了 Watcher 反复注册等问题，但其缺陷也非常明显。例如，文档不全、重试机制难用、异常全部转换成了 RuntimeException、没有足够的参考示例等。可见，一个简单易用、高效可靠的 ZooKeeper 客户端是多么重要

Apache Curator

Apache Curator 是 Apache 基金会提供的一款 ZooKeeper 客户端，它提供了一套易用性和可读性非常强的 Fluent 风格的客户端 API ，可以帮助我们快速搭建稳定可靠的 ZooKeeper 客户端程序。

Apache Curator 案例

基础使用

1、创建一个 Maven 项目，并添加 Apache Curator 的依赖：

<dependency> 
    <groupId>org.apache.curator</groupId> 
    <artifactId>curator-recipes</artifactId> 
    <version>4.0.1</version> 
</dependency>

2、 Curator 提供的基础 API 的使用：

public class Main { 

    public static void main(String[] args) throws Exception { 

        // Zookeeper集群地址，多个节点地址可以用逗号分隔 

        String zkAddress = "127.0.0.1:2181"; 

        // 重试策略，如果连接不上ZooKeeper集群，会重试三次，重试间隔会递增 

        RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 3); 

        // 创建Curator Client并启动，启动成功之后，就可以与Zookeeper进行交互了 

        CuratorFramework client =  CuratorFrameworkFactory.newClient(zkAddress, retryPolicy); 

        client.start(); 

        // 下面简单说明Curator中常用的API 

        // create()方法创建ZNode，可以调用额外方法来设置节点类型、添加Watcher 

        // 下面是创建一个名为"user"的持久节点，其中会存储一个test字符串 

        String path = client.create().withMode(CreateMode.PERSISTENT) 

            .forPath("/user", "test".getBytes()); 

        System.out.println(path); 

        // 输出:/user 

        // checkExists()方法可以检查一个节点是否存在 

        Stat stat = client.checkExists().forPath("/user"); 

        System.out.println(stat!=null); 

        // 输出:true，返回的Stat不为null，即表示节点存在 

        // getData()方法可以获取一个节点中的数据 

        byte[] data = client.getData().forPath("/user"); 

        System.out.println(new String(data)); 

        // 输出:test 

        // setData()方法可以设置一个节点中的数据 

        stat = client.setData().forPath("/user","data".getBytes()); 

        data = client.getData().forPath("/user"); 

        System.out.println(new String(data)); 

        // 输出:data 

        // 在/user节点下，创建多个临时顺序节点 

        for (int i = 0; i < 3; i++) { 

            client.create().withMode(CreateMode.EPHEMERAL_SEQUENTIAL) 

                .forPath("/user/child-"; 

        } 

        // 获取所有子节点 

        List<String> children = client.getChildren().forPath("/user"); 

        System.out.println(children); 

        // 输出：[child-0000000002, child-0000000001, child-0000000000] 

        // delete()方法可以删除指定节点，deletingChildrenIfNeeded()方法 

        // 会级联删除子节点 

        client.delete().deletingChildrenIfNeeded().forPath("/user"); 

    } 

}

Background

上面介绍的创建、删除、更新、读取等方法都是同步的，Curator 提供异步接口，引入了BackgroundCallback 这个回调接口以及 CuratorListener 这个监听器，用于处理 Background 调用之后服务端返回的结果信息。BackgroundCallback 接口和 CuratorListener 监听器中接收一个 CuratorEvent 的参数，里面包含事件类型、响应码、节点路径等详细信息。

下面我们通过一个示例说明 BackgroundCallback 接口以及 CuratorListener 监听器的基本使用

package org.apache.dubbo.registry;

import org.apache.curator.RetryPolicy;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.framework.api.BackgroundCallback;
import org.apache.curator.framework.api.CuratorEvent;
import org.apache.curator.framework.api.CuratorListener;
import org.apache.curator.retry.ExponentialBackoffRetry;
import org.apache.zookeeper.CreateMode;

public class Main2 {

    public static void main(String[] args) throws Exception {

        // Zookeeper集群地址，多个节点地址可以用逗号分隔 

        String zkAddress = "127.0.0.1:2181";

        // 重试策略，如果连接不上ZooKeeper集群，会重试三次，重试间隔会递增 

        RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 3);

        // 创建Curator Client并启动，启动成功之后，就可以与Zookeeper进行交互了 

        CuratorFramework client = CuratorFrameworkFactory

                .newClient(zkAddress, retryPolicy);

        client.start();

        // 添加CuratorListener监听器，针对不同的事件进行处理 

        client.getCuratorListenable().addListener(

                new CuratorListener() {

                    public void eventReceived(CuratorFramework client,

                                              CuratorEvent event) throws Exception {

                        switch (event.getType()) {

                            case CREATE:

                                System.out.println("CREATE:" +

                                        event.getPath());

                                break;

                            case DELETE:

                                System.out.println("DELETE:" +

                                        event.getPath());

                                break;

                            case EXISTS:

                                System.out.println("EXISTS:" +

                                        event.getPath());

                                break;

                            case GET_DATA:

                                System.out.println("GET_DATA:" +

                                        event.getPath() + ","

                                        + new String(event.getData()));

                                break;

                            case SET_DATA:

                                System.out.println("SET_DATA:" +

                                        new String(event.getData()));

                                break;

                            case CHILDREN:

                                System.out.println("CHILDREN:" +

                                        event.getPath());

                                break;

                            default:

                        }

                    }

                });

        // 注意:下面所有的操作都添加了inBackground()方法，转换为后台操作 

        client.create().withMode(CreateMode.PERSISTENT)

                .inBackground().forPath("/user", "test".getBytes());

        client.checkExists().inBackground().forPath("/user");

        client.setData().inBackground().forPath("/user",

                "setData-Test".getBytes());

        client.getData().inBackground().forPath("/user");

        for (int i = 0; i < 3; i++) {

            client.create().withMode(CreateMode.EPHEMERAL_SEQUENTIAL)

                    .inBackground().forPath("/user/child-");

        }

        client.getChildren().inBackground().forPath("/user");

        // 添加BackgroundCallback 

        client.getChildren().inBackground(new BackgroundCallback() {

            public void processResult(CuratorFramework client,

                                      CuratorEvent event) throws Exception {

                System.out.println("in background:"

                        + event.getType() + "," + event.getPath());

            }

        }).forPath("/user");

        client.delete().deletingChildrenIfNeeded().inBackground()

                .forPath("/user");

        System.in.read();

    }

}

// 输出： 

// CREATE:/user 

// EXISTS:/user 

// GET_DATA:/user,setData-Test 

// CREATE:/user/child- 

// CREATE:/user/child- 

// CREATE:/user/child- 

// CHILDREN:/user 

// DELETE:/user

连接状态监听

除了基础的数据操作，Curator 还提供了监听连接状态的监听器——ConnectionStateListener，它主要是处理 Curator 客户端和 ZooKeeper 服务器间连接的异常情况，例如，短暂或者长时间断开连接。

短暂断开连接时，ZooKeeper 客户端会检测到与服务端的连接已经断开，但是服务端维护的客户端 Session 尚未过期，之后客户端和服务端重新建立了连接；当客户端重新连接后，由于 Session 没有过期，ZooKeeper 能够保证连接恢复后保持正常服务。

而长时间断开连接时，Session 已过期，与先前 Session 相关的 Watcher 和临时节点都会丢失。当 Curator 重新创建了与 ZooKeeper 的连接时，会获取到 Session 过期的相关异常，Curator 会销毁老 Session，并且创建一个新的 Session。由于老 Session 关联的数据不存在了，在 ConnectionStateListener 监听到 LOST 事件时，就可以依靠本地存储的数据恢复 Session 了。

这里 Session 指的是 ZooKeeper 服务器与客户端的会话。客户端启动的时候会与服务器建立一个 TCP 连接，从第一次连接建立开始，客户端会话的生命周期也开始了。客户端能够通过心跳检测与服务器保持有效的会话，也能够向 ZooKeeper 服务器发送请求并接受响应，同时还能够通过该连接接收来自服务器的 Watch 事件通知。

我们可以设置客户端会话的超时时间（sessionTimeout），当服务器压力太大、网络故障或是客户端主动断开连接等原因导致连接断开时，只要客户端在 sessionTimeout 规定的时间内能够重新连接到 ZooKeeper 集群中任意一个实例，那么之前创建的会话仍然有效。ZooKeeper 通过 sessionID 唯一标识 Session，所以在 ZooKeeper 集群中，sessionID 需要保证全局唯一。由于 ZooKeeper 会将 Session 信息存放到硬盘中，即使节点重启，之前未过期的 Session 仍然会存在

public class Main3 { 

    public static void main(String[] args) throws Exception { 

        // Zookeeper集群地址，多个节点地址可以用逗号分隔 

        String zkAddress = "127.0.0.1:2181"; 

        // 重试策略，如果连接不上ZooKeeper集群，会重试三次，重试间隔会递增 

        RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000,3); 

        // 创建Curator Client并启动，启动成功之后，就可以与Zookeeper进行交互了 

        CuratorFramework client = CuratorFrameworkFactory 

            .newClient(zkAddress, retryPolicy); 

        client.start(); 

        // 添加ConnectionStateListener监听器 

        client.getConnectionStateListenable().addListener( 

          new ConnectionStateListener() { 

            public void stateChanged(CuratorFramework client,  

                    ConnectionState newState) { 

                // 这里我们可以针对不同的连接状态进行特殊的处理 

                switch (newState) {  

                    case CONNECTED: 

                        // 第一次成功连接到ZooKeeper之后会进入该状态。 

                        // 对于每个CuratorFramework对象，此状态仅出现一次 

                        break; 

                    case SUSPENDED: //   ZooKeeper的连接丢失 

                        break; 

                    case RECONNECTED: // 丢失的连接被重新建立 

                        break; 

                    case LOST:   

                        // 当Curator认为会话已经过期时，则进入此状态 

                        break; 

                    case READ_ONLY: // 连接进入只读模式 

                        break; 

                } 

            } 

        }); 

   } 

}

Watcher

Watcher 监听机制是 ZooKeeper 中非常重要的特性，可以监听某个节点上发生的特定事件，例如，监听节点数据变更、节点删除、子节点状态变更等事件。当相应事件发生时，ZooKeeper 会产生一个 Watcher 事件，并且发送到客户端。通过 Watcher 机制，就可以使用 ZooKeeper 实现分布式锁、集群管理等功能。

在 Curator 客户端中，我们可以使用 usingWatcher() 方法添加 Watcher，前面示例中，能够添加 Watcher 的有 checkExists()、getData()以及 getChildren() 三个方法，下面我们来看一个具体的示例：

public class Main4 { 

    public static void main(String[] args) throws Exception { 

        // Zookeeper集群地址，多个节点地址可以用逗号分隔 

        String zkAddress = "127.0.0.1:2181"; 

        // 重试策略，如果连接不上ZooKeeper集群，会重试三次，重试间隔会递增 

        RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000,3); 

        // 创建Curator Client并启动，启动成功之后，就可以与Zookeeper进行交互了 

        CuratorFramework client = CuratorFrameworkFactory 

              .newClient(zkAddress, retryPolicy); 

        client.start(); 

        try { 

           client.create().withMode(CreateMode.PERSISTENT) 

                 .forPath("/user", "test".getBytes()); 

        } catch (Exception e) { 

        } 

        // 这里通过usingWatcher()方法添加一个Watcher 

        List<String> children = client.getChildren().usingWatcher( 

          new CuratorWatcher() { 

            public void process(WatchedEvent event) throws Exception { 

                System.out.println(event.getType() + "," +  

                    event.getPath()); 

            } 

        }).forPath("/user"); 

        System.out.println(children); 

        System.in.read(); 

    } 

}

接下来，我们打开 ZooKeeper 的命令行客户端，在 /user 节点下先后添加两个子节点

命令：

create /user/test1

create /user/test2

此时我们只得到一行输出：

NodeChildrenChanged,/user

之所以这样，是因为通过 usingWatcher() 方法添加的 CuratorWatcher 只会触发一次，触发完毕后就会销毁。checkExists() 方法、getData() 方法通过 usingWatcher() 方法添加的 Watcher 也是一样的原理，只不过监听的事件不同，你若感兴趣的话，可以自行尝试一下。

相信你已经感受到，直接通过注册 Watcher 进行事件监听不是特别方便，需要我们自己反复注册 Watcher。Apache Curator 引入了 Cache 来实现对 ZooKeeper 服务端事件的监听。Cache 是 Curator 中对事件监听的包装，其对事件的监听其实可以近似看作是一个本地缓存视图和远程ZooKeeper 视图的对比过程。同时，Curator 能够自动为开发人员处理反复注册监听，从而大大简化了代码的复杂程度。

实践中常用的 Cache 有三大类：

NodeCache。对一个节点进行监听，监听事件包括指定节点的增删改操作。注意哦，NodeCache 不仅可以监听数据节点的内容变更，也能监听指定节点是否存在，如果原本节点不存在，那么 Cache 就会在节点被创建后触发 NodeCacheListener，删除操作亦然。
PathChildrenCache。对指定节点的一级子节点进行监听，监听事件包括子节点的增删改操作，但是不对该节点的操作监听。
TreeCache。综合 NodeCache 和 PathChildrenCache 的功能，是对指定节点以及其子节点进行监听，同时还可以设置监听的深度。

下面通过示例介绍上述三种 Cache 的基本使用：

public class Main5 { 

    public static void main(String[] args) throws Exception { 

        // Zookeeper集群地址，多个节点地址可以用逗号分隔 

        String zkAddress = "127.0.0.1:2181"; 

        // 重试策略，如果连接不上ZooKeeper集群，会重试三次，重试间隔会递增 

        RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000,3); 

        // 创建Curator Client并启动，启动成功之后，就可以与Zookeeper进行交互了 

        CuratorFramework client = CuratorFrameworkFactory 

           .newClient(zkAddress, retryPolicy); 

        client.start(); 

        // 创建NodeCache，监听的是"/user"这个节点 

        NodeCache nodeCache = new NodeCache(client, "/user"); 

        // start()方法有个boolean类型的参数，默认是false。如果设置为true， 

        // 那么NodeCache在第一次启动的时候就会立刻从ZooKeeper上读取对应节点的 

        // 数据内容，并保存在Cache中。 

        nodeCache.start(true); 

        if (nodeCache.getCurrentData() != null) { 

            System.out.println("NodeCache节点初始化数据为："  

                + new String(nodeCache.getCurrentData().getData())); 

        } else { 

            System.out.println("NodeCache节点数据为空"); 

        } 

        // 添加监听器 

        nodeCache.getListenable().addListener(() -> { 

            String data = new String(nodeCache.getCurrentData().getData()); 

            System.out.println("NodeCache节点路径：" + nodeCache.getCurrentData().getPath() 

                    + "，节点数据为：" + data); 

        }); 

        // 创建PathChildrenCache实例，监听的是"user"这个节点 

        PathChildrenCache childrenCache = new PathChildrenCache(client, "/user", true); 

        // StartMode指定的初始化的模式 

        // NORMAL:普通异步初始化 

        // BUILD_INITIAL_CACHE:同步初始化 

        // POST_INITIALIZED_EVENT:异步初始化，初始化之后会触发事件 

        childrenCache.start(PathChildrenCache.StartMode.BUILD_INITIAL_CACHE); 

        // childrenCache.start(PathChildrenCache.StartMode.POST_INITIALIZED_EVENT); 

        // childrenCache.start(PathChildrenCache.StartMode.NORMAL); 

        List<ChildData> children = childrenCache.getCurrentData(); 

        System.out.println("获取子节点列表："); 

        // 如果是BUILD_INITIAL_CACHE可以获取这个数据，如果不是就不行 

        children.forEach(childData -> { 

            System.out.println(new String(childData.getData())); 

        }); 

        childrenCache.getListenable().addListener(((client1, event) -> { 

            System.out.println(LocalDateTime.now() + "  " + event.getType()); 

            if (event.getType().equals(PathChildrenCacheEvent.Type.INITIALIZED)) { 

                System.out.println("PathChildrenCache:子节点初始化成功..."); 

            } else if (event.getType().equals(PathChildrenCacheEvent.Type.CHILD_ADDED)) { 

                String path = event.getData().getPath(); 

                System.out.println("PathChildrenCache添加子节点:" + event.getData().getPath()); 

                System.out.println("PathChildrenCache子节点数据:" + new String(event.getData().getData())); 

            } else if (event.getType().equals(PathChildrenCacheEvent.Type.CHILD_REMOVED)) { 

                System.out.println("PathChildrenCache删除子节点:" + event.getData().getPath()); 

            } else if (event.getType().equals(PathChildrenCacheEvent.Type.CHILD_UPDATED)) { 

                System.out.println("PathChildrenCache修改子节点路径:" + event.getData().getPath()); 

                System.out.println("PathChildrenCache修改子节点数据:" + new String(event.getData().getData())); 

            } 

        })); 

        // 创建TreeCache实例监听"user"节点 

        TreeCache cache = TreeCache.newBuilder(client, "/user").setCacheData(false).build(); 

        cache.getListenable().addListener((c, event) -> { 

            if (event.getData() != null) { 

                System.out.println("TreeCache,type=" + event.getType() + " path=" + event.getData().getPath()); 

            } else { 

                System.out.println("TreeCache,type=" + event.getType()); 

            } 

        }); 

        cache.start(); 

        System.in.read(); 

    } 

}

此时，ZooKeeper 集群中存在 /user/test1 和 /user/test2 两个节点，启动上述测试代码，得到的输出如下：

NodeCache节点初始化数据为：test //NodeCache的相关输出  

获取子节点列表：// PathChildrenCache的相关输出 

xxx 

xxx2 

// TreeCache监听到的事件 

TreeCache,type=NODE_ADDED path=/user 

TreeCache,type=NODE_ADDED path=/user/test1 

TreeCache,type=NODE_ADDED path=/user/test2 

TreeCache,type=INITIALIZED

DAIHAO丶

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
Apache Dubbo 源码搭建与解读（九）—— ZooKeeper客户端

ZooKeeper客户端ZooKeeper 官方提供的客户端支持了一些基本操作，例如，创建会话、创建节点、读取节点、更新数据、删除节点和检查节点是否存在等，但在实际开发中只有这些简单功能是根本不够的。而且，ZooKeeper 本身的一些 API 也存在不足，例如：ZooKeeper 的 Watcher 是一次性的，每次触发之后都需要重新进行注册。会话超时之后，没有实现自动重连的机制。ZooKeeper 提供了非常详细的异常，异常处理显得非常烦琐，对开发新手来说，非常不友好。只提供了
复制链接

扫一扫

专栏目录