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

需求

某分布式系统中，主节点可以有多台，可以动态上下线，任意一台客户端都能实时感知到主节点服务器的上下线。

需求分析

具体实现

（1）先在集群上创建/servers 节点

[zk: localhost:2181(CONNECTED) 10] create /servers "servers"
Created /servers

（2）在 Idea 中创建包名：com.huan.zkcase1
（3）服务器端向 Zookeeper 注册代码

package com.huan.zkcase1;

import org.apache.zookeeper.*;

import java.io.IOException;

public class DistributeServer {

    private static String connectString = "Bigdata04:2181,Bigdata05:2181,Bigdata06:2181";
    private static int sessionTimeout = 2000;
    private ZooKeeper zk = null;
    private String parentNode = "/servers";


    // 创建到 zk 的客户端连接
    public void getConnect() throws IOException {
        zk = new ZooKeeper( connectString, sessionTimeout, new Watcher() { 
            @Override
            public void process(WatchedEvent event) {

            }
        } );
    }


    // 注册服务器
    public void registServer(String hostname) throws Exception {
        String create = zk.create( parentNode + "/server",
                hostname.getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE,
                CreateMode.EPHEMERAL_SEQUENTIAL );
        System.out.println( hostname + " is online " + create );
    }

    // 业务功能
    public void business(String hostname) throws Exception {
        System.out.println( hostname + " is working ..." );
        Thread.sleep( Long.MAX_VALUE );
    }


    //程序入口
    public static void main(String[] args) throws Exception {
        // 1 获取 zk 连接
        DistributeServer server = new DistributeServer();
        server.getConnect();
        // 2 利用 zk 连接注册服务器信息
        server.registServer( args[0] );
        // 3 启动业务功能
        server.business( args[0] );
    }
}

（4）客户端代码

package com.huan.zkcase1;

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

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

public class DistributeClient {

    private static String connectString = "Bigdata04:2181,Bigdata05:2181,Bigdata06:2181";
    private static int sessionTimeout = 2000;
    private ZooKeeper zk = null;
    private String parentNode = "/servers";


    // 创建到 zk 的客户端连接
    public void getConnect() throws IOException {
        zk = new ZooKeeper( connectString, sessionTimeout, new Watcher() {
            @Override
            public void process(WatchedEvent event) {
                // 再次启动监听
                try {
                    getServerList();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        } );
    }

    // 获取服务器列表信息
    public void getServerList() throws Exception {
        // 1 获取服务器子节点信息，并且对父节点进行监听
        List<String> children = zk.getChildren( parentNode, true );
        // 2 存储服务器信息列表
        ArrayList<String> servers = new ArrayList<>();
        // 3 遍历所有节点，获取节点中的主机名称信息
        for (String child : children) {
            byte[] data = zk.getData( parentNode + "/" + child, false, null );
            servers.add( new String( data ) );
        }
        // 4 打印服务器列表信息
        System.out.println( servers );
    }

    // 业务功能
    public void business() throws Exception {
        System.out.println( "client is working ..." );
        Thread.sleep( Long.MAX_VALUE );
    }

    public static void main(String[] args) throws Exception{
        DistributeClient client = new DistributeClient();

        //1.获取zk连接
        client.getConnect();

        //2.获取servers 的子节点信息，从中获取服务器信息列表
        client.getServerList();

        //3.业务进程启动
        client.business();
    }
}

测试

1）在 Linux 命令行上操作增加减少服务器
（1）启动 DistributeClient 客户端
（2）在 Bigdata04上 zk 的客户端/servers 目录上创建临时带序号节点

[zk: localhost:2181(CONNECTED) 1] create -e -s /servers/Bigdata04 "Bigdata04 "
[zk: localhost:2181(CONNECTED) 2] create -e -s /servers/Bigdata05 "Bigdata05 "

（3）观察 Idea 控制台变化

[Bigdata04,Bigdata05]

（4）执行删除操作

[zk: localhost:2181(CONNECTED) 8] delete /servers/Bigdata050000000000 

（5）观察 Idea 控制台变化

[Bigdata05]

2）在 Idea 上操作增加减少服务器
（1）启动 DistributeClient 客户端（如果已经启动过，不需要重启）
（2）启动 DistributeServer 服务
①点击 Edit Configurations…

②在弹出的窗口中（Program arguments）输入想启动的主机，例如，Bigdata04

③回到 DistributeServer 的 main 方 法 ， 右 键 ， 在 弹 出 的 窗 口 中 点 击 Run “DistributeServer.main()”

④观察 DistributeServer 控制台，提示 Bigdata04 is working
⑤观察 DistributeClient 控制台，提示 Bigdata04 已经上线