前言
在使用Spring的过程中,我们通常会用@PostConstruct和@PreDestroy在Bean初始化或销毁时执行一些操作,这些操作属于Bean声明周期级别的。
那么,就存在一些遗漏的场景,比如我们想在容器本身的生命周期(比如容器启动、停止)的事件上做一些工作,很典型的就是Spring Boot中启动内嵌的Web容器。该怎么办?
这就需要用到Spring提供的另外一个接口Lifecycle。这篇文件就介绍一下Lifecycle接口,以及比它更聪明(Smart)的SmartLifecycle。
Lifecycle
Lifecycle是一个接口,它的作用是让开发者可以在所有的bean都创建完成(getBean)之后执行自己的初始化工作,或者在退出时执行资源销毁工作。
Lifecycle定义了三个方法,任何Bean实现了Lifecycle方法,当ApplicationContext收到start、stop和restart等信号时,就会调用对应的方法。因此可以通过实现Lifecycle接口获得容器生命周期的回调,实现业务扩展。
LifeCycle定义如下:
public interface Lifecycle { void start(); void stop(); boolean isRunning(); }
自定义Lifecycle实现类
首先我们来自定义一个类,实现Lifecycle接口,来看看具体的实践效果:
@Component public class MyLifeCycle implements Lifecycle { /** * 运行状态 */ private volatile boolean running = false; /** * 容器启动后调用 */ @Override public void start() { System.out.println("容器启动后执行MyLifeCycle操作..."); running = true; } /** * 容器停止时调用 */ @Override public void stop() { System.out.println("收到关闭容器的信号MyLifeCycle操作..."); running = false; } /** * 检查此组件是否正在运行。 * 1. 只有该方法返回false时,start方法才会被执行。 * 2. 只有该方法返回true时,stop(Runnable callback)或stop()方法才会被执行。 */ @Override public boolean isRunning() { System.out.println("检查MyLifeCycle组件的运行状态:" + running); return running; } }
单纯的将上述代码添加的Spring Boot项目当中,你会发现启动时并没有打印出任何相关的日志,只有在关闭应用时会打印出:
检查MyLifeCycle组件的运行状态:false
1
这是因为,在SpringBoot或Spring应用中如果没有调用AbstractApplicationContext#start方法,只是实现了Lifecycle接口,是不会执行Lifecycle接口中的启动方法和isRunning方法的。但在应用退出时会执行Lifecycle#isRunning方法判断该Lifecycle是否已经启动,如果返回true则调用Lifecycle#stop()停止方法。
这个实例有一个很明显的问题,那就是需要使用者显式的调用容器的start()和stop()方法,Lifecycle的接口方法才会被执行。
而在一般的项目中,我们很少这样显式的去调用,所以就需要一个更“聪明”的类来处理,这就是SmartLifecycle。
SmartLifecycle
SmartLifecycle继承自Lifecycle,提供了更丰富的功能:第一,start()方法无需容器显式调用就可以被执行;第二,可以控制多SmartLifecycle实例的执行顺序。
先来看一下SmartLifecycle接口的源码:
public interface SmartLifecycle extends Lifecycle, Phased { int DEFAULT_PHASE = 2147483647; default boolean isAutoStartup() { return true; } default void stop(Runnable callback) { this.stop(); callback.run(); } default int getPhase() { return 2147483647; } }
可以看出该接口除了继承Lifecycle接口外,还继承了Phased。其中getPhase方法便是来自Phased。也正是基于Phased接口的这个方法来控制SmartLifecycle的执行顺序的。
来看一下实例代码:
@Component public class MySmartLifecycle implements SmartLifecycle { private volatile boolean running = false; /** * 如果该`Lifecycle`类所在的上下文在调用`refresh`时,希望能够自己自动进行回调,则返回`true`, * false的值表明组件打算通过显式的start()调用来启动,类似于普通的Lifecycle实现。 */ @Override public boolean isAutoStartup() { return true; } /** * SmartLifecycle子类的才有的方法,当isRunning方法返回true时,该方法才会被调用。 * 很多框架中的源码中,都会把真正逻辑写在stop()方法内。 * 比如quartz和Redis的spring支持包。 */ @Override public void stop(Runnable callback) { System.out.println("MySmartLifecycle容器停止,执行回调函数"); stop(); // 如果你让isRunning返回true,需要执行stop这个方法,那么就不要忘记调用callback.run()。 // 否则在程序退出时,Spring的DefaultLifecycleProcessor会认为这个MySmartLifecycle没有stop完成,程序会一直卡着结束不了,等待一定时间(默认超时时间30秒)后才会自动结束。 callback.run(); } /** * 1. 主要在该方法中启动任务或者其他异步服务,比如开启MQ接收消息<br/> * 2. 当上下文被刷新(所有对象已被实例化和初始化之后)时,将调用该方法, * 默认生命周期处理器将检查每个SmartLifecycle对象的isAutoStartup()方法返回的布尔值。 * 如果为“true”,则该方法会被调用,而不是等待显式调用自己的start()方法。 */ @Override public void start() { System.out.println("MySmartLifecycle容器启动完成 ..."); running = true; } /** * 接口Lifecycle子类的方法,只有非SmartLifecycle的子类才会执行该方法。<br/> * 1. 该方法只对直接实现接口Lifecycle的类才起作用,对实现SmartLifecycle接口的类无效。<br/> * 2. 方法stop()和方法stop(Runnable callback)的区别只在于,后者是SmartLifecycle子类的专属。 */ @Override public void stop() { System.out.println("MySmartLifecycle容器停止 ..."); running = false; } /** * 1. 只有该方法返回false时,start方法才会被执行。<br/> * 2. 只有该方法返回true时,stop(Runnable callback)或stop()方法才会被执行。 */ @Override public boolean isRunning() { System.out.println("MySmartLifecycle检查运行状态 ..."); return running; } /** * 如果有多个实现接口SmartLifecycle的类,则这些类的start的执行顺序按getPhase方法返回值从小到大执行。<br/> * 例如:1比2先执行,-1比0先执行。stop方法的执行顺序则相反,getPhase返回值较大类的stop方法先被调用,小的后被调用。 * */ @Override public int getPhase() { return 0; } }
关于每个方法的功能,注释部分已经明确说明了,下面启动SpringBoot项目,打印日志如下:
MySmartLifecycle检查运行状态 ...
MySmartLifecycle容器启动完成 ...
1
2
关闭SpringBoot项目,打印日志如下:
MySmartLifecycle检查运行状态 ...
MySmartLifecycle容器停止,执行回调函数
MySmartLifecycle容器停止 ...
1
2
3
通过上述实例可以看出:如果一个Bean实现了SmartLifecycle接口,则会执行启动方法。SmartLifecycle#isRunning判断是否已经执行,返回false表示还未执行,则调用SmartLifecycle#start()执行。
当关闭时,同样先检查运行状态,如果正在运行,则执行关闭操作。关闭时,还可以处理对应的回调函数。
其中,Phased返回值越小,优先级越高。
小结
当需要基于Spring容器的生命周期来处理一些逻辑时,通常可以实现SmartLifecycle接口来完成。像Spring Cloud,Spring Boot中都有大量的实践案例。所以,无论实战或阅读源码,不了解Lifecycle相关接口,都是一种损失。本文的产生也是在遇到Spring Cloud集成Nacos的源码中获得的灵感。