【Nacos源码篇(三)】NotifyCenter源码剖析

Cocowwy.

已于 2022-01-29 16:24:30 修改

阅读量1.4k

点赞数 2

分类专栏： Nacos源码源码 springcloud alibaba 文章标签： java 后端

于 2022-01-24 22:11:27 首次发布

本文链接：https://blog.csdn.net/Pzzzz_wwy/article/details/122675984

版权

源码同时被 3 个专栏收录

11 篇文章 0 订阅

订阅专栏

Nacos源码

10 篇文章 7 订阅

订阅专栏

springcloud alibaba

5 篇文章 0 订阅

订阅专栏

文章目录

先看看具体属性
它的静态代码块
接着看看他的核心方法

先上一些个人阅读的时候觉得的核心代码，至于 NotifyCenter已经在上文当中简要阐述了一下。

先看看具体属性

    // Notify状态
    private static final AtomicBoolean CLOSED = new AtomicBoolean(false);
    // 单例申明
    private static final NotifyCenter INSTANCE = new NotifyCenter();
    // 上文讲过，存储事件的发布者们
    private final Map<String, EventPublisher> publisherMap = new ConcurrentHashMap<>(16);

目前只放了这三个属性，可以看出来几个点：
⚡️ 通过JUC的原子Boolan来判定是否启用
⚡️ NotifyCenter 是单例的
⚡️ publisherMap 来保存对应事件的发布者

它的静态代码块

static {
    // ignore ..
    INSTANCE.sharePublisher = new DefaultSharePublisher();
	ThreadUtils.addShutdownHook(NotifyCenter::shutdown);
}

debug的时候主要就这两步，其余的暂时忽略了
⚡️设置了默认的事件发布者
⚡️ 添加了一个shutdown的钩子，也就是shutdown的时候NotifyCenter的停用

接着看看他的核心方法

shutdown

终止NotifyCenter

     /**
     * 调用所有的事件发布者的shutdown  ---》Closeable 该接口定义的
     */
    public static void shutdown() { 
        if (!CLOSED.compareAndSet(false, true)) {
            return;
        }
        for (Map.Entry<String, EventPublisher> entry : INSTANCE.publisherMap.entrySet()) {
            try {
                EventPublisher eventPublisher = entry.getValue();
                eventPublisher.shutdown();
            } catch (Throwable e) {
               // ..
            }
        }
        try {
            INSTANCE.sharePublisher.shutdown();
        } catch (Throwable e) {
            // ..
        }
    }

⚡️ 通过CAS来校验NotifyCenter的状态，保证并发安全
⚡️ 调用每个 EventPublisher 的 shutdown() 方法，该方法实现自 Closeable接口，该方法的具体实现细节会在对 EventPublisher 的源码分析篇细化
所以对于shutdown方法而言，主要是调用了每个发布者的shutdown方法

publishEvent

添加事件订阅者的同时注册事件发布者

     /**
     * 向publishmap添加事件发布者，同时添加订阅者
     */
    private static void addSubscriber(final Subscriber consumer, Class<? extends Event> subscribeType,
                                      EventPublisherFactory factory) {
        final String topic = ClassUtils.getCanonicalName(subscribeType);
        synchronized (NotifyCenter.class) {
            // MapUtils.computeIfAbsent is a unsafe method.
            // 向map里面添加了事件的发布者
            MapUtil.computeIfAbsent(INSTANCE.publisherMap, topic, factory, subscribeType, ringBufferSize);
        }
        EventPublisher publisher = INSTANCE.publisherMap.get(topic);
        if (publisher instanceof ShardedEventPublisher) {
            ((ShardedEventPublisher) publisher).addSubscriber(consumer, subscribeType);
        } else {
            // 将事件订阅者添加进刚刚的事件发布者里面
            publisher.addSubscriber(consumer);
        }
    }

⚡️ 主要的入参是一个订阅者，以及事件发布工厂（通过它来创建事件发布者，因为可能事件添加到订阅者的时候，该订阅者尚未初始化，即publish的value还未有值）
⚡️ 加synchronized锁，不过本身publisherMap已经是ConcurrentHashMap，不太明白为什么这里需要加锁？注释上写的是computeIfAbsent该方法不安全，可能是为了保证并发状态下同时读取，存在读到为null，导致publish同时创建，所以加上的锁
⚡️ computeIfAbsent通过EventPublisherFactory(实现了函数式接口BiFunction的apply方法)以及传入的订阅者的类型，来构造一个事件的发布者，然后丢到publishMap里面，具体apply的代码可以看下面贴出来的，

    @Override
    public EventPublisher apply(final Class<? extends Event> eventType, final Integer maxQueueSize) {
        // Like ClientEvent$ClientChangeEvent cache by ClientEvent
        Class<? extends Event> cachedEventType =
                eventType.isMemberClass() ? (Class<? extends Event>) eventType.getEnclosingClass() : eventType;
        publisher.computeIfAbsent(cachedEventType, eventClass -> {
            NamingEventPublisher result = new NamingEventPublisher();
            result.init(eventClass, maxQueueSize);
            return result;
        });
        return publisher.get(cachedEventType);
    }

⚡️接着之前的代码向publishmap添加了事件发布者之后，publisher.addSubscriber(consumer)接着将事件订阅者添加给事件发布者
总的来说，就是添加事件订阅者的同时，如果publishmap没有事件的发布者，则向通过工厂创建一个事件发布者，并且给这个事件发布者添加事件，利用发布者来进行广播

registerToPublisher

向事件订阅者注册事件


     /**
     * 直接向事件发布者注册订阅者
     *
     * @param eventType    class Instances type of the event type.
     * @param factory      publisher factory.
     * @param queueMaxSize the publisher's queue max size.
     */
    public static EventPublisher registerToPublisher(final Class<? extends Event> eventType,
                                                     final EventPublisherFactory factory, final int queueMaxSize) {
        if (ClassUtils.isAssignableFrom(SlowEvent.class, eventType)) {
            return INSTANCE.sharePublisher;
        }

        final String topic = ClassUtils.getCanonicalName(eventType);
        synchronized (NotifyCenter.class) {
            // MapUtils.computeIfAbsent is a unsafe method.
            MapUtil.computeIfAbsent(INSTANCE.publisherMap, topic, factory, eventType, queueMaxSize);
        }
        return INSTANCE.publisherMap.get(topic);
    }
}

⚡️其实阅读过上面的addSubscriber方法之后，就能够很好的理解这一块的内容了，事件发布者注册事件，同理

deregisterPublisher

有添加自然有删除

    /**
     * Deregister publisher.
     *
     * @param eventType class Instances type of the event type.
     */
    public static void deregisterPublisher(final Class<? extends Event> eventType) {
        final String topic = ClassUtils.getCanonicalName(eventType);
        EventPublisher publisher = INSTANCE.publisherMap.remove(topic);
        try {
            publisher.shutdown();
        } catch (Throwable ex) {
           // ignore ..
        }
    }

⚡️逻辑很清晰，先从publishmap移除掉事件的发布者
⚡️再调用事件发布者的shutdown方法，当然，事件发布者publish的源码分析会在下一章节阐述
至此，NotifyCenter的源码剖析到此差不多了

Nacos源码篇

语雀版文档

事件通知

Nacos源码注释

Cocowwy.

关注

2
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
【Nacos源码篇(三)】NotifyCenter源码剖析

文章目录先看看具体属性它的静态代码块接着看看他的核心方法shutdownpublishEventregisterToPublisherderegisterPublisher先上一些个人阅读的时候觉得的核心代码，至于NotifyCenter已经在上文当中简要阐述了一下。本人原文笔迹：戳这里本人源码注释：戳这里先看看具体属性 // Notify状态 private static final AtomicBoolean CLOSED = new AtomicBoolean(false);
复制链接

扫一扫

专栏目录