2

源码级别的广播与监听实现

 2 years ago
source link: https://xie.infoq.cn/article/f613ec0d97ae324c6e830028b
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
neoserver,ios ssh client

源码级别的广播与监听实现

源码级别的广播与监听实现

近期疫情形势严峻，情形不容乐观，周末也不敢出去浪了，躲在家里“葛优躺”。闲来无事，又翻了遍Spring的源码。不翻不知道，一翻吓一跳，之前翻过的源码已经吃进了肚子里，再见亦是陌生人。

个人建议：为了以后能快速的捡起某个知识点，最好的方法还是形成文档，下次有遗漏的时候，直接读文档，按之前的思路捋一遍，“干净又卫生”。

之前的文章中我们已经介绍过如何在项目中快速上手“事件通知机制”，相信大家已经掌握了。但是我们作为高级javaer，要知其然，更要知其所以然。今天就带大家从源码的角度来分析一下广播与监听的底层实现原理。

源码导入教程也给你准备好了，不来试试吗？

版本号：spring-framework-5.0.x

为了实现广播与监听的功能，Spring为我们提供了两个重要的函数式接口：ApplicationEventPublisher和ApplicationListener。前者的publishEvent()方法为我们提供了发送广播的能力；后者的onApplicationEvent()方法为我们提供了监听并处理事件的能力。

接下来我们就来分析一下spring是如何运用这两种能力的。

不知道大家对单例对象的初始化调用过程是否熟悉？主要调用方法流程如下：

applyBeanPostProcessorsBeforeInitialization方法会去遍历该工厂创建的所有的Bean后置处理器，然后去依次执行后置处理器对应的postProcessBeforeInitialization方法。

在该方法的实现类中我们看到了两个熟悉的类名

不知道大家还记得不，这俩类是在beanFactory的准备工作过程中添加的两个bean的后置处理器，所以这个地方会依次去执行这两个类中的实现方法。

由于蓝框中类的实现方法是默认实现按照原样返回的给定的bean，所以此处不用过多分析，我们重点来看下红框中类的方法实现。

该方法中最重要的是invokeAwareInterfaces方法，它的作用是检测对应的bean是否实现了某个Aware接口，如果实现了的话就去进行相关的调用。

if (bean instanceof ApplicationEventPublisherAware) {    ((ApplicationEventPublisherAware) bean).setApplicationEventPublisher(this.applicationContext);}

我们发现在invokeAwareInterfaces方法中出现了如上代码，这不就是和广播发送相关的吗？所以只要我们写一个类来实现ApplicationEventPublisherAware接口，就可以在该bean中注入一个ApplicationEventPublisher对象，也就获得了发送广播的能力。

applyBeanPostProcessorsAfterInitialization方法也会去遍历该工厂创建的所有的Bean后置处理器，然后去依次执行后置处理器对应的postProcessAfterInitialization方法。

同样的，该方法的实现类中也有ApplicationContextAwareProcessor和ApplicationListenerDetector两个类，但是不同的是，前者的类的实现方法是默认实现按照原样返回的给定bean，而后者做了相关的处理。

this.applicationContext.addApplicationListener((ApplicationListener<?>) bean);

上述代码是将实现了ApplicationListener接口的bean添加到监听器列表中，最终是保存在AbstractApplicationEventMulticaster的成员变量defaultRetriever的集合applicationListeners中。

猜想：当发送广播消息时，就直接找到集合中的这些监听器，然后调用每个监听器的onApplicationEvent方法完成事件的处理。

在refresh()的finishRefresh()方法中，

publishEvent(new ContextRefreshedEvent(this));

发送一条事件类型为ContextRefreshedEvent的广播消息，用来代表Spring容器初始化结束。通过分析发现，该方法中最主要的就是如下代码：

//真正的广播交给 applicationEventMulticaster 来完成getApplicationEventMulticaster().multicastEvent(applicationEvent, eventType);

refresh()的initApplicationEventMulticaster()将applicationEventMulticaster初始化为SimpleApplicationEventMulticaster

在实现类SimpleApplicationEventMulticaster的方法中，会找到已注册的ApplicationListener列表，然后分别调用invokeListener方法（将监听和事件作为参数传到方法并执行的过程就是发送广播的过程）。

底层调用的是listener.onApplicationEvent(event);方法，也就是各个监听实现类单独处理广播消息的逻辑。

消息与监听绑定

看到这儿，你是不是已经发现了：消息类型和监听器的绑定发生在广播过程中。接下来就让我们去一探究竟

我们看一下multicastEvent()方法中的getApplicationListeners(event, type)方法。

在该方法中，用到了ConcurrentHashMap类型的缓存retrieverCache，所以每种类型的事件在广播的时候会触发一次绑定操作。它的 key 由事件的来源和类型确定，它的 value 中就包含了由事件来源和类型所确定的所有监听列表。

其中绑定的逻辑就出现在retrieveApplicationListeners方法中，大家可以去源码中查看。

纸上得来终觉浅，绝知此事要躬行。为了更好地理解广播与监听的流程，我们当然得用实战来加以辅佐！

自定义事件

public class MyEvent extends ApplicationContextEvent {    public MyEvent(ApplicationContext source) {        super(source);    }}

自定义广播

@Componentpublic class MyPublisher implements ApplicationEventPublisherAware, ApplicationContextAware {  private ApplicationEventPublisher applicationEventPublisher;  private ApplicationContext applicationContext;  @Override  public void setApplicationEventPublisher(ApplicationEventPublisher applicationEventPublisher) {    this.applicationEventPublisher = applicationEventPublisher;  }  @Override  public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {    this.applicationContext = applicationContext;  }  //发送广播消息  public void publishEvent(){    System.out.println("我要开始发送消息了。。。");    MyEvent myEvent = new MyEvent(applicationContext);    applicationEventPublisher.publishEvent(myEvent);  }}

MyPublisher实现了ApplicationEventPublisherAware接口 ，在spring初始化（见上文中的invokeAwareInterfaces）的时候会回调setApplicationEventPublisher方法，获取到初始化（添加bean后置处理器ApplicationContextAwareProcessor）时的AbstractApplicationContext，而AbstractApplicationContext又间接实现了ApplicationEventPublisher而获得发送能力。真正执行的是 AbstractApplicationContext 类中的 publishEvent 方法。

自定义监听

@Componentpublic class MyEventListener implements ApplicationListener<MyEvent> {    @Override    public void onApplicationEvent(MyEvent event) {        System.out.println("我监听到你的消息了");    }}

MyEventListener实现了ApplicationListener接口，在spring初始化（见上文中的addApplicationListener）的时候会添加到applicationListeners中，在执行publishEvent 方法时就会走MyEventListener中的onApplicationEvent方法。

@RestController@RequestMapping("/demo")public class DemoTest {    @Autowired    private MyPublisher myPublisher;    @RequestMapping("/test")    public void test() {        myPublisher.publishEvent();    }}

访问127.0.0.1:8008/demo/test就可以发送广播了，发送与监听内容如下：

我要开始发送消息了。。。我监听到你的消息了

看到这儿，相信你己经完全掌握了广播与监听的精髓了，赶快实践起来吧。阿 Q 将持续更新 java 实战方面的文章，如果你有不同的意见或者更好的 idea，欢迎联系阿 Q。

阿 Q 说代码，值得关注的公众号

文章风格多变，配图通俗易懂，故事生动有趣，来聊聊技术呀！


report

Recommend

About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK