《进大厂前先学会阅读源码》之shenyu网关-替换ZooKeeper客户端

相信大家碰到源码一开始都是比较无从下手的🙃,不知道从哪开始阅读,面对大量代码晕头转向,索性就读不下去了,又浪费了一次提升自己的机会😭。

我认为有一种方法 ,可以解决大家的困扰!那就是通过阅读某一次开源【commit】、某一次社区【ISSUE】,从这个入口出发去阅读源码!!

至此,我们发现自己开始从大量堆砌的源码中脱离开来😀。豁然开朗,柳暗花明又一村🍀。

一、前瞻

今天我们攻克的一次开源提交:commit链接

本次commit的核心内容就在下图红框中,意思很清晰明了:替换当前的ZooKeeper客户端。

看看Magic Header是什么

Magic Header 通常指的是文件开头的一段特定字节序列,用来标识或确认文件的格式和类型。这些字节序列是预先定义的,不同的文件格式有不同的 Magic Header。操作系统和应用程序通过读取这些特定的字节序列来识别文件的格式,即使文件扩展名被更改或丢失。

简单来说就是用来标识文件。

我们先整体看下本次所有提交的内容,虽然看起来涉及了大量 的模块、大量的代码,但核心其实就是红框对应的内容。既然是要替换当前的ZooKeeper客户端 ,那便是要新建Curator 这个新的客户端,同时修改调用端的调用对象!

下面我们就来看看贡献者是怎么实现替换的 !!同时上文提到并将数据保存在没有 magic header 的 zk 中,正确翻译应该是保存数据在zk时,不使用magic header来标识数据,那他又是怎么做的??

二、探索

我们参照上图 ,先看下ZookeeperClient类,该类很显眼的引入了本次提交的主角CuratorFramework,ZookeeperClient封装了CuratorFramework作为ZooKeeper新的客户端。

java 复制代码
public class ZookeeperClient {

    private static final Logger LOGGER = LoggerFactory.getLogger(ZookeeperClient.class);

    private final ZookeeperConfig config;

    private final CuratorFramework client;

    private final Map<String, CuratorCache> caches = new ConcurrentHashMap<>();

    public ZookeeperClient(final ZookeeperConfig zookeeperConfig) {
        this.config = zookeeperConfig;
        ExponentialBackoffRetry retryPolicy = new ExponentialBackoffRetry(config.getBaseSleepTimeMilliseconds(), config.getMaxRetries(), config.getMaxSleepTimeMilliseconds());

        CuratorFrameworkFactory.Builder builder = CuratorFrameworkFactory.builder()
                .connectString(config.getServerLists())
                .retryPolicy(retryPolicy)
                .connectionTimeoutMs(config.getConnectionTimeoutMilliseconds())
                .sessionTimeoutMs(config.getSessionTimeoutMilliseconds())
                .namespace(config.getNamespace());

        if (!StringUtils.isEmpty(config.getDigest())) {
            builder.authorization("digest", config.getDigest().getBytes(StandardCharsets.UTF_8));
        }

        this.client = builder.build();
    }

    /**
     * start.
     */
    public void start() {
        this.client.start();
        try {
            this.client.blockUntilConnected();
        } catch (InterruptedException e) {
            LOGGER.warn("Interrupted during zookeeper client starting.");
            Thread.currentThread().interrupt();
        }
    }
}

而ZookeeperInstanceRegisterRepository就是起到控制器 的作用,把ZookeeperConfig类的配置注册到ZookeeperClient来初始化我们新的客户端

我们在代码可以看到初始化方法init()。

java 复制代码
@Join
public class ZookeeperInstanceRegisterRepository implements ShenyuInstanceRegisterRepository {

    private static final Logger LOGGER = LoggerFactory.getLogger(ZookeeperInstanceRegisterRepository.class);

    private ZookeeperClient client;

    private final Map<String, String> nodeDataMap = new HashMap<>();

    @Override
    public void init(final InstanceConfig config) {
        Properties props = config.getProps();
        int sessionTimeout = Integer.parseInt(props.getProperty("sessionTimeout", "3000"));
        int connectionTimeout = Integer.parseInt(props.getProperty("connectionTimeout", "3000"));

        int baseSleepTime = Integer.parseInt(props.getProperty("baseSleepTime", "1000"));
        int maxRetries = Integer.parseInt(props.getProperty("maxRetries", "3"));
        int maxSleepTime = Integer.parseInt(props.getProperty("maxSleepTime", String.valueOf(Integer.MAX_VALUE)));

        ZookeeperConfig zkConfig = new ZookeeperConfig(config.getServerLists());
        zkConfig.setBaseSleepTimeMilliseconds(baseSleepTime)
                .setMaxRetries(maxRetries)
                .setMaxSleepTimeMilliseconds(maxSleepTime)
                .setSessionTimeoutMilliseconds(sessionTimeout)
                .setConnectionTimeoutMilliseconds(connectionTimeout);

        String digest = props.getProperty("digest");
        if (!StringUtils.isEmpty(digest)) {
            zkConfig.setDigest(digest);
        }

        this.client = new ZookeeperClient(zkConfig);
        this.client.getClient().getConnectionStateListenable().addListener((c, newState) -> {
            if (newState == ConnectionState.RECONNECTED) {
                nodeDataMap.forEach((k, v) -> {
                    if (!client.isExist(k)) {
                        client.createOrUpdate(k, v, CreateMode.EPHEMERAL);
                        LOGGER.info("zookeeper client register instance success: {}", v);
                    }
                });
            }
        });

        client.start();
    }
}

替换客户端的操作其实不会太复杂,主要就是把所有旧的客户端入参修改为新的客户端对象 ,可以看下贡献者的提交代码。

我们打开commit链接指向的ISSUE

意思其实是旧的zkClient 保存数据时默认会把magic header头信息 插入到数据中,导致了我们的序列化数据失败

而贡献者本次提交把客户端改为了Curator客户端,也就不会有上面插入magic header头信息的操作了。

大家能否感受通过commit、ISSUE这种方式来阅读源码的乐趣呢!

创作不易,不妨点赞、关注、收藏支持一下,各位的支持就是我创作的最大动力❤️

相关推荐
callmeSoon6 天前
Vue2 模板编译三部曲(三)|生成器 Generator
vue.js·源码阅读
Tans56 天前
Java ReentrantLock 源码阅读笔记(上)
java·源码阅读
程序猿阿越9 天前
ChaosBlade源码(一)blade命令行
java·后端·源码阅读
码农明明16 天前
Android源码分析:从源头分析View事件的传递
android·操作系统·源码阅读
欧阳码农2 个月前
vue3的defineAsyncComponent是如何实现异步组件的呢?
vue.js·源码·源码阅读
鳄鱼不怕_牙医不怕2 个月前
Flutter 源码梳理系列(三十七):OffsetLayer
flutter·源码阅读
鳄鱼不怕_牙医不怕2 个月前
Flutter 源码梳理系列(三十六):RenderObject:PAINTING
flutter·源码阅读
鳄鱼不怕_牙医不怕2 个月前
Flutter 源码梳理系列(三十四):ContainerLayer
flutter·源码阅读
鳄鱼不怕_牙医不怕2 个月前
Flutter 源码梳理系列(三十一):PaintingContext
flutter·源码阅读
callmeSoon2 个月前
Vue2 模板编译三部曲(一)|架构设计 & 解析器 Parser
vue.js·源码阅读