SpringBoot优雅关机

背景

简单的说，就是向应用进程发出停止指令之后，能保证正在执行的业务操作不受影响，直到操作运行完毕之后再停止服务。应用程序接收到停止指令之后，会进行如下操作：

1. 停止接收新的访问请求
2. 正在处理的请求，等待请求处理完毕；对于内部正在执行的其他任务，比如定时任务、mq 消费等等，也要等当前正在执行的任务执行完毕，并且不再启动新的任务
3. 当应用准备关闭的时候，按需向外发出信号，告知其他应用服务准备接手，以保证服务高可用

如果暴力的关闭应用程序，比如通过kill -9 命令强制直接关闭应用程序进程，可能会导致正在执行的任务数据丢失或者错乱，也可能会导致任务所持有的全局资源等不到释放，比如当前任务持有 redis 的锁，并且没有设置过期时间，当任务突然被终止并且没有主动释放锁，会导致其他进程因无法获取锁而不能处理业务。

方案实践

方法一：通过Actuator的Endpoint机制关闭服务

需要先添加spring-boot-starter-actuator监控服务依赖包

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-actuator</artifactId>
</dependency>

shutdown端点是关闭的，需要在application.properties里配置里面开启：
而shutdown默认配置下是不支持HTTP进行Web访问的，所以使用HTTP请求进行关闭时的配置，也需要开启：

management.endpoint.shutdown.enabled=true
management.endpoints.web.exposure.include=shutdown

最后将SpringBoot服务启动之后，使用POST请求类型，调用以下接口，即可实现关闭服务！

http://127.0.0.1:8080/actuator/shutdown

方法二：使用ApplicationContext的close方法关闭服务

获取ApplicationContext对象，然后在相关的位置调用close方法，就可以关闭服务。

@SpringBootApplication
public class Application {

    public static void main(String[] args) {
      ConfigurableApplicationContext context = SpringApplication.run(Application.class, args);

      try {
         TimeUnit.SECONDS.sleep(10);
      } catch (InterruptedException e) {
         e.printStackTrace();
      }
      //启动10秒以后，自动关闭
      context.close();
    }
}

我们也可以自己写一个Controller，获取对应的ApplicationContext对象，通过api操作调用close方法关停服务，示例代码如下：

@RestController
public class ShutdownController implements ApplicationContextAware {

    private ApplicationContext context;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.context = applicationContext;
    }

    /**
     * 关闭服务
     */
    @GetMapping("/shutdown")
    public void shutdownContext() {
        ((ConfigurableApplicationContext) context).close();
    }
}

方法三：使用SpringApplication的exit方法关闭服务

通过调用一个SpringApplication.exit()方法也可以安全的退出程序，同时会返回一个退出码，这个退出码可以传递给所有的context，最后通过调用System.exit()可以将这个错误码也传给JVM。

@SpringBootApplication
public class Application {

    public static void main(String[] args) {
        ConfigurableApplicationContext context = SpringApplication.run(Application.class, args);

        try {
            TimeUnit.SECONDS.sleep(5);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        //5秒后，关闭服务
        exitApplication(context);
    }

    public static void exitApplication(ConfigurableApplicationContext context) {
     //获取退出码
        int exitCode = SpringApplication.exit(context, (ExitCodeGenerator) () -> 0);
        //退出码传递给jvm，安全退出程序
        System.exit(exitCode);
    }

}

其他监听介绍

ApplicationListener
如果有些服务，比如定时任务，我们想在SpringBoot关闭数据源连接池之前，将其关闭，可以通过实现ApplicationListener接口，监听bean对象的变化情况，在bean对象销毁之前，执行相关的关闭任务。

@Component
public class JobTaskListener implements ApplicationListener {

    @Override
    public void onApplicationEvent(ApplicationEvent applicationEvent) {
        // 在spring bean容器销毁之前执行的事件，防止数据库连接池在任务终止前销毁
        if (applicationEvent instanceof ContextClosedEvent) {
            System.out.println("关闭相关的定时任务");
        }
    }
}

PreDestroy
上文中，我们提到了实现DisposableBean接口，可以监听应用关闭前的回调处理，其实在自定义的方法上加@PreDestroy注解，也可以实现相同的效果。

@Component
public class AppDestroyConfig {

    @PreDestroy
    public void PreDestroy(){
        System.out.println("应用程序正在关闭。。。");
    }
}