SpringBoot中SmartLifecycle的使用解析

更新時間：2023年11月15日 08:48:12 作者：阿孟呀

這篇文章主要介紹了SpringBoot中SmartLifecycle的使用解析,SmartLifecycle是一個擴展了Lifecycle接口,可以跟蹤spring容器ApplicationContext刷新或者關閉的接口,實現(xiàn)該接口的實現(xiàn)類有特定的執(zhí)行順序,需要的朋友可以參考下

前言

SmartLifecycle是一個擴展了Lifecycle接口，可以跟蹤spring容器ApplicationContext刷新或者關閉的接口，實現(xiàn)該接口的實現(xiàn)類有特定的執(zhí)行順序。

當ApplicationContext刷新時，所有bean都加載和初始化完成后，根據(jù)isAutoStartup和isRunning判斷是否自動執(zhí)行start()

1.新建TestSmartLifecycle類

public class TestSmartLifecycle implements SmartLifecycle {
   
  @Override
    public void start() {
        System.out.println("TestSmartLifecycle is start");
    }
//SmartLifecycle組件停止后的回調(diào)，程序被異步關閉
    @Override
    public void stop() {
        System.out.println("SmartLifecycle is stop");
    }
//為false，并且isAutoStartup為true，執(zhí)行start方法
//為true，執(zhí)行stop方法。
    @Override
    public boolean isRunning() {
        return false;
    }
//如果為true，實現(xiàn)了SmartLifecycle 接口的組件，能再ApplicationContext已經(jīng)刷新的時候，自動執(zhí)行start方法，
    @Override
    public boolean isAutoStartup() {
        return true;
    }
//當running為true時，再應用關閉時會執(zhí)行stop(Runnable callback)方法,
如果再執(zhí)行完stop后，running沒有設置為false時，應用會卡住一段時間，無法退出
    @Override
    public void stop(Runnable callback) {
        System.out.println("TestSmartLifecycle is stop");
        callback.run();
    }
 
    @Override
    public int getPhase() {
        return 0;
    }
}

2.配置Bean

@Configuration
public class TestConfiguration {
 
    @Bean
    TestSmartLifecycle testSmartLifecycle(){
        return new TestSmartLifecycle();
    }
}

3.debugger啟動應用程序

可以看見如下的調(diào)用鏈，大致是SpringApplication.run(CommonsTestApplication.class, args)-->refreshContext-->refresh-->startBeans-->start，其中，再startBean中獲取了所有實現(xiàn)SmartLifecycle的非懶加載的bean，并再start方法中，調(diào)用對應實現(xiàn)bean的start方法，執(zhí)行相應的邏輯，具體可看下面截圖。

private void startBeans(boolean autoStartupOnly) {
		Map<String, Lifecycle> lifecycleBeans = getLifecycleBeans();
		Map<Integer, LifecycleGroup> phases = new HashMap<>();
		lifecycleBeans.forEach((beanName, bean) -> {
			if (!autoStartupOnly || (bean instanceof SmartLifecycle && ((SmartLifecycle) bean).isAutoStartup())) {
				int phase = getPhase(bean);
				LifecycleGroup group = phases.get(phase);
				if (group == null) {
					group = new LifecycleGroup(phase, this.timeoutPerShutdownPhase, lifecycleBeans, autoStartupOnly);
					phases.put(phase, group);
				}
				group.add(beanName, bean);
			}
		});
		if (!phases.isEmpty()) {
			List<Integer> keys = new ArrayList<>(phases.keySet());
			Collections.sort(keys);
			for (Integer key : keys) {
				phases.get(key).start();
			}
		}
	}