前言
在使用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;
- }
-
-
- @Override
- public boolean isRunning() {
- System.out.println("检查MyLifeCycle组件的运行状态:" + running);
- return running;
- }
- }
单纯的将上述代码添加的Spring Boot项目当中,你会发现启动时并没有打印出任何相关的日志,只有在关闭应用时会打印出:
- 检查MyLifeCycle组件的运行状态:false
这是因为,在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;
-
-
- @Override
- public boolean isAutoStartup() {
- return true;
- }
-
-
- @Override
- public void stop(Runnable callback) {
- System.out.println("MySmartLifecycle容器停止,执行回调函数");
- stop();
- // 如果你让isRunning返回true,需要执行stop这个方法,那么就不要忘记调用callback.run()。
- // 否则在程序退出时,Spring的DefaultLifecycleProcessor会认为这个MySmartLifecycle没有stop完成,程序会一直卡着结束不了,等待一定时间(默认超时时间30秒)后才会自动结束。
- callback.run();
- }
-
-
- @Override
- public void start() {
- System.out.println("MySmartLifecycle容器启动完成 ...");
- running = true;
- }
-
-
- @Override
- public void stop() {
- System.out.println("MySmartLifecycle容器停止 ...");
- running = false;
- }
-
-
- @Override
- public boolean isRunning() {
- System.out.println("MySmartLifecycle检查运行状态 ...");
- return running;
- }
-
-
- @Override
- public int getPhase() {
- return 0;
- }
- }
关于每个方法的功能,注释部分已经明确说明了,下面启动SpringBoot项目,打印日志如下:
- MySmartLifecycle检查运行状态 ...
- MySmartLifecycle容器启动完成 ...
关闭SpringBoot项目,打印日志如下:
- MySmartLifecycle检查运行状态 ...
- MySmartLifecycle容器停止,执行回调函数
- MySmartLifecycle容器停止 ...
通过上述实例可以看出:如果一个Bean实现了SmartLifecycle接口,则会执行启动方法。SmartLifecycle#isRunning判断是否已经执行,返回false表示还未执行,则调用SmartLifecycle#start()执行。
当关闭时,同样先检查运行状态,如果正在运行,则执行关闭操作。关闭时,还可以处理对应的回调函数。
其中,Phased返回值越小,优先级越高。
小结
当需要基于Spring容器的生命周期来处理一些逻辑时,通常可以实现SmartLifecycle接口来完成。像Spring Cloud,Spring Boot中都有大量的实践案例。所以,无论实战或阅读源码,不了解Lifecycle相关接口,都是一种损失。本文的产生也是在遇到Spring Cloud集成Nacos的源码中获得的灵感。