Java程序优雅停机v2

springboot 2.3新增了优雅停机策略
见：https://www.jianshu.com/p/8358495f7e2f

但是只支持内置的四个嵌入式Web服务器（Jetty，Reactor Netty，Tomcat和Undertow）以及响应式和基于Servlet的Web应用程序，如果想让其他组件优雅停机，比如数据库的datasource，还是需要自己做点动作的。

问题：在进行大量的实时数据接入时，为了保证数据的入库效率，会将批量入库操作和接入操作解耦，使用独立线程批量入库，所以当程序关机时，会出现

1. 数据已经接入到缓存队列，但是入库还没进行，导致数据丢失。
2. spring容器已经关闭，入库线程获取datasource时发现连接已经关闭，报错“ dataSource already closed”

解决方案：

1. 在主线程中定义ShutdownHook，当服务收到kill指令后，第一时间关闭“数据接入”逻辑
2. 定义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;
   }