zookeeper实现master选举

最新推荐文章于 2023-06-06 23:17:03 发布

mjlfto

最新推荐文章于 2023-06-06 23:17:03 发布

阅读量233

点赞数

分类专栏： zookeeper

本文链接：https://blog.csdn.net/mjlfto/article/details/104057392

版权

zookeeper 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

zookeeper的使用创建非常多，在大数据中经常用来实现master选举，在分布式系统中实现分布式锁，也可以用来实现符在均衡，分布式队列，分布式命名服务，下边就来说说使用zookeeper实现master选举

1.定义一个节点数据类：

/**
 * @ClassName NodeData
 * @Author mjlft
 * @Date 2020/1/20 21:58
 * @Version 1.0
 * @Description TODO
 */
public class NodeData {
    private String cid;
    private String nodeName;

    public NodeData() {
    }

    public NodeData(String cid, String nodeName) {
        this.cid = cid;
        this.nodeName = nodeName;
    }

    public String getCid() {
        return cid;
    }

    public void setCid(String cid) {
        this.cid = cid;
    }

    public String getNodeName() {
        return nodeName;
    }

    public void setNodeName(String nodeName) {
        this.nodeName = nodeName;
    }
}

2.定义节点服务类，这个也是进行master选举的核心类，这里主要利用TreeCache来监听master节点上的事件，如果遇见master节点被删除，则触发选举逻辑

import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.recipes.cache.TreeCache;
import org.apache.curator.framework.recipes.cache.TreeCacheEvent;
import org.apache.curator.framework.recipes.cache.TreeCacheListener;
import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.data.Stat;


/**
 * @ClassName WorkService
 * @Author mjlft
 * @Date 2020/1/20 21:58
 * @Version 1.0
 * @Description TODO
 */
public class WorkService {

    private final static String MASTER_PATH = "/master";
    private final static String WORK_PATH = "/service";
    private NodeData nodeData;
    private NodeData master;
    private CuratorFramework client;

    public WorkService() {
    }

    public WorkService(NodeData nodeData, CuratorFramework client) {
        this.nodeData = nodeData;
        this.client = client;
    }

    //节点初始化
    public void init() throws Exception {
        //首先判断当前节点是否已经存在
        Stat stat = client.checkExists().forPath(WORK_PATH + "/" + nodeData.getNodeName());
        if (stat == null) {
            //如果不存在就创建一个新的节点
            client.create().withMode(CreateMode.EPHEMERAL).forPath(WORK_PATH + "/" + nodeData.getNodeName(),
                    nodeData.getNodeName().getBytes());
        } else {
            //如果已经存在，则更新数据
            client.setData().forPath(WORK_PATH + "/" + nodeData.getNodeName(), nodeData.getNodeName().getBytes());
        }
    }

    /**
     * 注册选举事件
     *
     * @throws Exception
     */
    public void register() throws Exception {
        final NodeData node = this.nodeData;

        //判断master是否存在，如果在起始注册事件时就不存在，那么试着将自己注册为master
        if (client.checkExists().forPath(MASTER_PATH) == null) {
            try {
                //注意，如果在多线程，或者是分布式系统中，这里可能会存在已经被其他线程或机器抢先注册，那么直接跳过
                client.create().withMode(CreateMode.EPHEMERAL).forPath(MASTER_PATH, nodeData.getNodeName().getBytes());
            } catch (Exception e) {
                e.printStackTrace();
            }
        }

        //注册监听事件
        final TreeCache treeCache = TreeCache.newBuilder(this.client, MASTER_PATH).setCacheData(true).setMaxDepth(2).build();
        treeCache.getListenable().addListener(new TreeCacheListener() {
            @Override
            public void childEvent(CuratorFramework curatorFramework, TreeCacheEvent treeCacheEvent) throws Exception {
                //如果节点被删除
                if (treeCacheEvent.getType().equals(TreeCacheEvent.Type.NODE_REMOVED)) {
                    System.out.println(Thread.currentThread().getName() + ":我监听到了master删除事件");
                    //如果已经被其他注册，跳过什么也不处理
                    if (client.checkExists().forPath(MASTER_PATH) != null) {
                        return;
                    }

                    //这里也存在并发问题
                    try {
                        curatorFramework.create().withMode(CreateMode.EPHEMERAL).forPath(MASTER_PATH,
                                node.getNodeName().getBytes());
                    } catch (Exception e) {
                        System.out.println("可能已经被其他机器注册");
                    }
                    System.out.println(node.getNodeName() + "竞选master成功");
                }
            }
        });
        treeCache.start();
    }

    //开启客户端
    public void start() {
        if (this.client != null) {
            this.client.start();
        }
    }

    //关闭客户端
    public void stop() {
        if (this.client != null) {
            this.client.close();
        }
    }
}

3.定义一个测试类，通过控制台输入当前master节点id，手动关闭节点触发机器宕机，这样就可以进一步触发master选举事件

import org.apache.curator.RetryPolicy;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.retry.RetryNTimes;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

/**
 * @ClassName Test
 * @Author mjlft
 * @Date 2020/1/20 22:27
 * @Version 1.0
 * @Description TODO
 */
public class Test {

    public static void main(String[] args) throws Exception {
        RetryPolicy retryPolicy = new RetryNTimes(3, 100);
        List<WorkService> workServices = new ArrayList<>(16);
        for(int i = 0; i < 10; i ++){
            CuratorFramework client = CuratorFrameworkFactory.newClient("192.168.1.107:2181, 192.168.1.107:2182",
                    30*60*1000, 5*1000, retryPolicy);
            NodeData nodeData = new NodeData("cid_"+i, "nodeName_" + i);
            WorkService workService = new WorkService(nodeData, client);
            workServices.add(workService);
            workService.start();
            workService.init();
            workService.register();
        }

        while (true){
            Scanner sc = new Scanner( System.in );
            int index = sc.nextInt(10);
            WorkService workService = workServices.get(index);
            workService.stop();
        }
    }
}

4.最后看看maven 依赖

 <dependency>
     <groupId>org.apache.curator</groupId>
     <artifactId>curator-framework</artifactId>
     <version>4.2.0</version>
</dependency>

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

由于只有一台机器，这里我们使用手动删除master节点，下边是测试结果：

0
nodeName_0节点宕机
Curator-TreeCache-0:我监听到了master删除事件
Curator-TreeCache-9:我监听到了master删除事件
Curator-TreeCache-7:我监听到了master删除事件
Curator-TreeCache-3:我监听到了master删除事件
Curator-TreeCache-4:我监听到了master删除事件
Curator-TreeCache-1:我监听到了master删除事件
Curator-TreeCache-5:我监听到了master删除事件
Curator-TreeCache-6:我监听到了master删除事件
Curator-TreeCache-2:我监听到了master删除事件
Curator-TreeCache-8:我监听到了master删除事件
nodeName_9竞选master成功
9
nodeName_9节点宕机
Curator-TreeCache-1:我监听到了master删除事件
Curator-TreeCache-6:我监听到了master删除事件
Curator-TreeCache-2:我监听到了master删除事件
Curator-TreeCache-7:我监听到了master删除事件
Curator-TreeCache-8:我监听到了master删除事件
Curator-TreeCache-5:我监听到了master删除事件
Curator-TreeCache-3:我监听到了master删除事件
Curator-TreeCache-9:我监听到了master删除事件
Curator-TreeCache-4:我监听到了master删除事件
nodeName_5竞选master成功
5
nodeName_5节点宕机
Curator-TreeCache-3:我监听到了master删除事件
Curator-TreeCache-7:我监听到了master删除事件
Curator-TreeCache-4:我监听到了master删除事件
Curator-TreeCache-2:我监听到了master删除事件
Curator-TreeCache-1:我监听到了master删除事件
Curator-TreeCache-8:我监听到了master删除事件
Curator-TreeCache-6:我监听到了master删除事件
Curator-TreeCache-5:我监听到了master删除事件
nodeName_3竞选master成功