Zookeeper——服务器动态上下线客户端动态监听

Posted 2022-08-01 张起灵-小哥

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Zookeeper——服务器动态上下线客户端动态监听相关的知识，希望对你有一定的参考价值。

文章目录：

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);