CountDownLatch的监听异步回调成功应用

最新推荐文章于 2025-01-02 06:00:00 发布

原创最新推荐文章于 2025-01-02 06:00:00 发布 · 452 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用CountDownLatch确保ZooKeeper连接建立后再执行后续代码的机制。通过单例模式创建ZooKeeper会话，并利用自定义Watcher监听连接状态，确保连接成功后才继续执行业务逻辑。

通过CountDownLatch 实现等待zookeeper连接建立完毕后，代码再向后执行的逻辑。

package com.wolf.cache.zk;


import lombok.extern.slf4j.Slf4j;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooKeeper;

import java.io.IOException;
import java.util.concurrent.CountDownLatch;

@Slf4j
public class ZookeeperSession {

    private static CountDownLatch connectedSemaphore = new CountDownLatch(1);

    private ZooKeeper zooKeeper;

    private ZookeeperSession(){
        // 去连接zookeeper server, 创建会话的时候，是异步进行的
        // 所有要给一个监听器，说告诉我们什么时候是真正的完成跟zookeeper server 连接
        try {
            this.zooKeeper = new ZooKeeper("192.168.43.10:2181,192.168.43.11:2181,192.168.43.12:2181",
                    5000,
                    new ZookeeperWather());
            log.info("### zookeeper server 连接成功， state={} ###", zooKeeper.getState());

            connectedSemaphore.await();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    private class ZookeeperWather implements Watcher {

        @Override
        public void process(WatchedEvent watchedEvent) {
            if (Event.KeeperState.SyncConnected == watchedEvent.getState()) {
                connectedSemaphore.countDown();
            }
        }
    }

    private static class SingleTon {
        private static ZookeeperSession instance = new ZookeeperSession();

        public static ZookeeperSession getInstance() {
            return SingleTon.instance;
        }
    }

    public static ZookeeperSession getInstance() {
        return SingleTon.getInstance();
    }

    public static void init() {
        getInstance();
    }

}