springboot 2.3新增了优雅停机策略
见：https://www.jianshu.com/p/8358495f7e2f
但是只支持内置的四个嵌入式Web服务器（Jetty，Reactor Netty，Tomcat和Undertow）以及响应式和基于Servlet的Web应用程序，如果想让其他组件优雅停机，比如数据库的datasource，还是需要自己做点动作的。
问题：在进行大量的实时数据接入时，为了保证数据的入库效率，会将批量入库操作和接入操作解耦，使用独立线程批量入库，所以当程序关机时，会出现
- 数据已经接入到缓存队列，但是入库还没进行，导致数据丢失。
- spring容器已经关闭，入库线程获取datasource时发现连接已经关闭，报错“ dataSource already closed”
解决方案：
- 在主线程中定义ShutdownHook，当服务收到kill指令后，第一时间关闭“数据接入”逻辑
- 定义ApplicationEventListener，当收到ContextClosedEvent事件时，主动检查当前是否还有未执行完成的子线程，有则阻塞等待子线程执行完毕。
ShutdownHook:
@Slf4j
@Component
public class MainThread implements CommandLineRunner {
@Override
public void run(String... args) throws Exception {
// 在主线程中定义一次即可
Runtime.getRuntime().addShutdownHook(new Thread(() -> {
// 修改全局变量，告诉子线程中断while循环
StaticDefine.isShutDown = true;
log.info("ShutdownHook 执行完成，全局信号已经修改");
}));
}
}
ApplicationEventListener：
@Component
@Slf4j
public class ApplicationEventListener implements ApplicationListener<ApplicationEvent> {
@Override
public void onApplicationEvent(ApplicationEvent applicationEvent) {
if(applicationEvent instanceof ContextClosedEvent){
log.info("Spring容器即将执行关闭");
int time = 0;
while (ThreadPoolExecutorManager.haveActiveThread() && time <= 30){
try {
TimeUnit.SECONDS.sleep(1);
} catch (InterruptedException e) {
e.printStackTrace();
}
time++;
}
log.info("Spring容器关闭");
}
}
}
其中ThreadPoolExecutorManager.haveActiveThread()是根据ThreadPoolExecutor中的getActiveCount()来判断执行器中是否还有在运行的线程。
/**
* 判断是否有进程在工作
* @return 是否
*/
public static Boolean haveActiveThread(){
for (ThreadPoolExecutor executor : EXECUTOR_MAP.values()) {
if (executor.getActiveCount() > 0) {
return true;
}
}
return false;
}
