首页 Java 如何在 Spring Boot 中实现优雅停机？

如何在 Spring Boot 中实现优雅停机？

Author: 潘志的研发笔记 · 分类： Java ·    2024年12月 · tags:

大家好，我是志哥，今天推荐一个特别实用的 Spring Boot 技术特性，它就是“如何实现服务优雅停机？”，下面我们一起来看看具体技术实践！

01、故事背景

什么叫优雅停机？

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

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

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

那么如何在不影响正在执行的业务的情况下，将应用程序安全的进行关闭呢？

02、方案实践

SpringBoot 官方文档上，已经告诉开发者只需要实现特定接口即可监听到项目启动成功与关闭时的事件，相关接口如下：

  • CommandLineRunner接口：当应用启动成功后但在开始接受流量之前，会回调此接口的实现类，也可以实现ApplicationRunner接口，工作的方式与CommandLineRunner与之类似
  • DisposableBean接口：当应用正要被销毁前，会回调此接口的实现类，也可以使用@PreDestroy注解，被标记的方法也会被调用

基于此流程，我们可以创建一个服务监听类，用于监听到项目启动成功与关闭时的回调服务，示例代码如下：

@Component
public class AppListener implements CommandLineRunner, DisposableBean {
    @Override
    public void run(String... args) throws Exception {
        System.out.println("应用启动成功，预加载相关数据");
    }
    @Override
    public void destroy() throws Exception {
        System.out.println("应用正在关闭，清理相关数据");
    }
}

每一个SpringApplication在启用的时候，都会向 JVM 注册一个关闭钩子shutdown hook，以确保ApplicationContext在退出的时候，通过这个勾子通知 JVM，实现服务正常的关闭，以下介绍的所有关闭服务的方法，都是基于这一原理进行实现的。

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

使用此方法，需要先添加spring-boot-starter-actuator监控服务依赖包，

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

默认配置下，shutdown端点是关闭的，需要在application.properties里配置里面开启：

management.endpoint.shutdown.enabled=true

虽然Actuator的端点，支持通过JMX或HTTP进行远程访问。而shutdown默认配置下是不支持HTTP进行Web访问的，所以使用HTTP请求进行关闭时的配置，也需要开启：

management.endpoints.web.exposure.include=shutdown

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

http://127.0.0.1:8080/actuator/shutdown

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

如果你不想添加spring-boot-starter-actuator监控服务依赖包来关停服务，也可以使用ApplicationContext的close方法来关停服务，他会自动销毁bean对象并关停服务。

只需要在应用启用的时候，获取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();
    }
}

2.3、方法三：监听服务pid，通过kill方式关闭服务

通过api方式来关停服务，在很多人看来并不安全，因为一旦接口泄漏了，意味着用户可以随便请求这个接口来关闭服务，其影响不言而喻，因此很多人建议在服务端，通过其他的方式来关闭服务，比如通过进程命令方式来关停。

在springboot启动的时候将应用进程 ID 写入一个app.pid文件，生成的路径可以指定，然后通过脚本命令方式来关闭服务。

启动示例代码如下：

@SpringBootApplication
public class Application {
    public static void main(String[] args) {
        SpringApplication application = new SpringApplication(Application.class);
        application.addListeners(new ApplicationPidFileWriter("/home/app/project1/app.pid"));
        application.run();
    }
}

通过如下命令方式，可以安全的关闭服务。

cat /home/app/project1/app.pid | xargs kill

这种方式，也是目前在linux操作系统中，使用较为普遍的一种解决方案，区别在于实现的方式可能不同，有的不用写文件，通过其他方式来获取应用进程 ID。

如果使用kill -9 <pid>的方式关闭服务，服务的监听器不会收到任何消息，类似于直接强杀应用进程，此方法不可取
！

2.4、方法四：使用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);
    }
}

03、其他监听介绍

3.1、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("关闭相关的定时任务");
        }
    }
}

3.2、PreDestroy

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

示例代码如下：

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

04、小结

本位主要围绕如何安全的关闭SpringBoot服务，进行了一些方案操作的介绍，如果有疏漏的地方，欢迎网友批评指出！

示例代码地址：

https://gitee.com/pzblogs/spring-boot-example-demo
3