Zookeeper——服务器动态上下线、客户端动态监听_服务器动态上下线监听逻辑

文章目录：

1.前言

2.实操步骤

2.1 服务端代码

2.2 客户端代码

2.3 测试

---

1.前言

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

---

2.实操步骤

首先，我这里还是像前几篇zookeeper的文章一样，考虑到电脑的8G内存以及性能，我这里就不显示zk集群了，还是以zk单机版来演示。

先到zk客户端的根节点下，创建一个 servers 节点。

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

2.1 服务端代码

我们先来完成服务端向zookeeper注册、动态上下线的代码。

package com.szh.case1;
 
import org.apache.zookeeper.*;
 
import java.io.IOException;
 
/**
 *
 */
public class DistributeServer {
 
    private ZooKeeper zk;
    //获取zk连接时，要连接的zk单机版信息
    private String connectString = "192.168.40.130:2181";
    //获取zk连接时，要连接的zk集群信息
    //private String connectString = "zk101:2181,zk102:2181,zk103:2181";
    //zk服务端与客户端连接的最大超时时间
    private int sessionTimeout = 30000;
 
    public static void main(String[] args) throws IOException, KeeperException, InterruptedException {
        DistributeServer server = new DistributeServer();
        //1.获取zk连接
        server.getConnection();
        //2.将服务器注册到zk集群, args参数通过启动main方法时传入即可
        server.register(args[0]);
        //3.启动业务逻辑（这里采用线程睡眠模拟）
        server.business();
    }
 
    private void getConnection() throws IOException {
        zk = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
            @Override
            public void process(WatchedEvent watchedEvent) {
 
            }
        });
    }
 
    /**
     * hostname: 将服务器注册到zk集群时，所需的服务器名称
     * ZooDefs.Ids.OPEN_ACL_UNSAFE: 此权限表示允许所有人访问该节点（服务器）
     * CreateMode.EPHEMERAL_SEQUENTIAL: 由于服务器是动态上下线的，上线后存在，下线后不存在，所以是临时节点
     *                                  而服务器一般都是有序号的，所以是临时、有序的节点
     */
    private void register(String hostname) throws KeeperException, InterruptedException {
        String node = zk.create("/servers/" + hostname, hostname.getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);
        System.out.println("已成功创建 " + node + "节点");
        System.out.println(hostname + " 已经上线");
    }
 
    private void business() throws InterruptedException {
        Thread.sleep(Long.MAX_VALUE);
    }
}

2.2 客户端代码

服务端代码写好之后，再来完成客户端动态监听zk服务端各个节点的代码。

package com.szh.case1;
 
import org.apache.zookeeper.KeeperException;
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.Arrays;
import java.util.List;
 
/**
 *
 */
public class DistributeClient {
 
    private ZooKeeper zk;
    //获取zk连接时，要连接的zk单机版信息
    private String connectString = "192.168.40.130:2181";
    //获取zk连接时，要连接的zk集群信息
    //private String connectString = "zk101:2181,zk102:2181,zk103:2181";
    //zk服务端与客户端连接的最大超时时间
    private int sessionTimeout = 30000;
 
    public static void main(String[] args) throws IOException, KeeperException, InterruptedException {
        DistributeClient client = new DistributeClient();
        //1.获取zk连接
        client.getConnection();
        //2.监听/servers下面所有的子节点的变化
        client.getServerList();
        //3.业务逻辑（这里采用线程睡眠模拟）
        client.business();
    }
 
    private void getConnection() throws IOException {
        zk = new ZooKeeper(connectString, sessionTimeout, new Watcher() {
            @Override
            public void process(WatchedEvent watchedEvent) {
                try {
                    getServerList();
                } catch (KeeperException e) {
                    e.printStackTrace();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        });
    }
 
    private void getServerList() throws KeeperException, InterruptedException {
        //监听 /servers 路徑下的所有子节点的变化，true表示启动监听器，回调getConnection()方法中new的那个Watcher对象中的process方法
        List<String> zkChildren = zk.getChildren("/servers", true);
        List<String> servers = new ArrayList<>();
        zkChildren.forEach(z -> {
            try {
                byte[] zkData = zk.getData("/servers/" + z, false, null);
                servers.add(new String(zkData));
            } catch (KeeperException e) {
                e.printStackTrace();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        });
        System.out.println(servers);
        System.out.println("-------------");
    }
 
    private void business() throws InterruptedException {
        Thread.sleep(Long.MAX_VALUE);
    }
}

2.3 测试

首先，我先不启动服务端的代码，先启动客户端代码，然后通过zookeeper命令行来完成测试。

先在客户端命令行中执行如下操作，在servers节点下，创建子结点 zk101，然后看看客户端代码监听到了什么？

上面的执行结果可以看到，在servers下面依次创建子结点，客户端代码都可以成功监听到。

下面我们删除节点，看看客户端能不能做到动态监听功能（也即删除的节点不会再被监听到）。 

---

下面，我们换一种测试方法，先启动客户端代码，然后再启动服务端代码，完全采用Java代码来测试。

只是在服务端代码中，我们的 register 方法中传参用到了 args ，所以启动之前要传入这个参数。

传入192.168.40.130之后，可以看到服务端代码已经能够实现动态上线了。 

这里我们的192.168.40.130动态上线之后，可以看到客户端也正常的监听到它了。 

转到zk命令行中，也可以看到这台服务器的节点信息。

由于我们之前的服务端代码还启动着，此时我们再传入新的参数 192.168.40.131，那么之前的130服务器肯定会被挤掉（这里模拟就是main方法，同一个类肯定只能同时启动一次main方法了），那么我们看看客户端能不能动态监听到130下线、131上线。 

服务端自然可以正常实现131这台服务器的动态上线。在客户端代码中可以看到List集合中已经没有130了（即130已经下线了），而131正常上线。