大数据技术之Zookeeper服务器动态上下线监听（6）

目录

 服务器动态上下线监听案例

步骤 1: 创建 Zookeeper 客户端

步骤 2: 注册 Watcher 并获取服务器列表

步骤 3: 服务器上下线操作

示例代码

说明

注意事项

 服务器动态上下线监听案例测试

测试环境准备

测试步骤

示例代码

ServerListWatcherExample.java

运行测试

控制台输出示例

---

 服务器动态上下线监听案例

在 Zookeeper 中，你可以使用 Watcher 机制来实现对服务器列表的动态上下线监听。Watcher 机制允许客户端注册监听器到特定的 Zookeeper 节点上，当这些节点发生变化时（例如节点被创建、删除或数据改变），Zookeeper 会向客户端发送通知。

下面是一个使用原生 Zookeeper 客户端 API 实现服务器动态上下线监听的示例：

步骤 1: 创建 Zookeeper 客户端

首先，创建一个 Zookeeper 客户端，并定义一个 Watcher 来监听事件。

import org.apache.zookeeper.*;
import org.apache.zookeeper.data.Stat;

import java.io.IOException;
import java.util.List;

public class ServerListWatcherExample {

    private static final String ZK_SERVERS = "localhost:2181";
    private static final int SESSION_TIMEOUT_MS = 5000;
    private static final String SERVER_LIST_PATH = "/server-list";

    private ZooKeeper zooKeeper;

    public ServerListWatcherExample() throws IOException {
        zooKeeper = new ZooKeeper(ZK_SERVERS, SESSION_TIMEOUT_MS, this::watcherMethod);
    }

    public void close() throws InterruptedException {
        if (zooKeeper != null) {
            zooKeeper.close();
        }
    }

    private void watcherMethod(WatchedEvent event) {
        if (event.getType() == Event.EventType.NodeChildrenChanged) {
            // 当服务器列表发生变化时，重新获取服务器列表
            getServerList();
        }
    }

    public void getServerList() {
        try {
            List<String> serverList = zooKeeper.getChildren(SERVER_LIST_PATH, true);
            System.out.println("Current server list: " + serverList);
        } catch (KeeperException | InterruptedException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) throws IOException, InterruptedException {
        ServerListWatcherExample example = new ServerListWatcherExample();
        example.getServerList(); // 初始获取服务器列表
        // 让主线程保持运行状态
        Thread.sleep(Long.MAX_VALUE);
    }
}

步骤 2: 注册 Watcher 并获取服务器列表

在 watcherMethod 方法中，当接收到 NodeChildrenChanged 事件时，调用 getServerList 方法来重新获取服务器列表。

步骤 3: 服务器上下线操作

为了让示例更加完整，你可以模拟服务器的上下线操作。可以通过在 Zookeeper 中创建或删除节点来模拟服务器的上线和下线。

示例代码

这里提供一个简单的服务器上下线操作示例：

import org.apache.zookeeper.KeeperException;

public class ServerListWatcherExample {

    // ... 上面的代码 ...

    public void addServer(String serverName) {
        try {
            zooKeeper.create(SERVER_LIST_PATH + "/" + serverName, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);
            System.out.println(serverName + " added to server list.");
        } catch (KeeperException | InterruptedException e) {
            e.printStackTrace();
        }
    }

    public void removeServer(String serverName) {
        try {
            zooKeeper.delete(SERVER_LIST_PATH + "/" + serverName, -1);
            System.out.println(serverName + " removed from server list.");
        } catch (KeeperException | InterruptedException e) {
            e.printStackTrace();
        }
    }

    // ... 其他代码 ...
}

// 在 main 方法中添加服务器
example.addServer("server1");
example.addServer("server2");
example.removeServer("server1");

// 保持主线程运行以便接收通知
Thread.sleep(Long.MAX_VALUE);

说明

• addServer 方法：在 /server-list 下创建一个临时节点，表示服务器上线。
• removeServer 方法：删除 /server-list 下的某个临时节点，表示服务器下线。

注意事项

• 确保你的 Zookeeper 服务器正在运行。
• 如果你使用的是不同的 Zookeeper 配置（如连接字符串），请相应地修改 ZK_SERVERS 和其他配置参数。

 服务器动态上下线监听案例测试

为了测试服务器动态上下线监听案例的有效性，你可以按照以下步骤进行：

测试环境准备

1. 启动 Zookeeper 服务：确保有一个可用的 Zookeeper 服务器。
2. 编写客户端程序：根据上一个回答中提供的示例代码，你可以编写一个客户端程序来监听服务器列表的变化。

测试步骤

1. 启动客户端程序：运行客户端程序，让其开始监听服务器列表的变化。
2. 模拟服务器上下线：手动在 Zookeeper 中创建或删除服务器节点来模拟服务器的上下线。
3. 观察变化：观察客户端程序是否能够正确检测到服务器列表的变化，并打印出更新后的服务器列表。

示例代码

下面是基于上一个回答中的示例代码进行的测试：

ServerListWatcherExample.java

import org.apache.zookeeper.*;
import org.apache.zookeeper.data.Stat;

import java.io.IOException;
import java.util.List;

public class ServerListWatcherExample {

    private static final String ZK_SERVERS = "localhost:2181";
    private static final int SESSION_TIMEOUT_MS = 5000;
    private static final String SERVER_LIST_PATH = "/server-list";

    private ZooKeeper zooKeeper;

    public ServerListWatcherExample() throws IOException {
        zooKeeper = new ZooKeeper(ZK_SERVERS, SESSION_TIMEOUT_MS, this::watcherMethod);
    }

    public void close() throws InterruptedException {
        if (zooKeeper != null) {
            zooKeeper.close();
        }
    }

    private void watcherMethod(WatchedEvent event) {
        if (event.getType() == Event.EventType.NodeChildrenChanged) {
            // 当服务器列表发生变化时，重新获取服务器列表
            getServerList();
        }
    }

    public void getServerList() {
        try {
            List<String> serverList = zooKeeper.getChildren(SERVER_LIST_PATH, true);
            System.out.println("Current server list: " + serverList);
        } catch (KeeperException | InterruptedException e) {
            e.printStackTrace();
        }
    }

    public void addServer(String serverName) {
        try {
            zooKeeper.create(SERVER_LIST_PATH + "/" + serverName, new byte[0], ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL);
            System.out.println(serverName + " added to server list.");
        } catch (KeeperException | InterruptedException e) {
            e.printStackTrace();
        }
    }

    public void removeServer(String serverName) {
        try {
            zooKeeper.delete(SERVER_LIST_PATH + "/" + serverName, -1);
            System.out.println(serverName + " removed from server list.");
        } catch (KeeperException | InterruptedException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) throws IOException, InterruptedException {
        ServerListWatcherExample example = new ServerListWatcherExample();
        example.getServerList(); // 初始获取服务器列表

        // 模拟服务器上下线
        example.addServer("server1");
        example.addServer("server2");
        example.removeServer("server1");

        // 保持主线程运行以便接收通知
        Thread.sleep(Long.MAX_VALUE);
    }
}

运行测试

1. 启动 Zookeeper 服务器。
2. 运行 ServerListWatcherExample 程序。
3. 观察控制台输出，确认客户端程序能够正确检测到服务器列表的变化，并打印出更新后的服务器列表。

控制台输出示例

Current server list: []
server1 added to server list.
Current server list: [server1]
server2 added to server list.
Current server list: [server1, server2]
server1 removed from server list.
Current server list: [server2]