如何優雅地中止 Spring Boot 應用？

首先來介紹下什麼是優雅地中止，簡而言之，就是對應用進程發送中止指令以後，能保證正在執行的業務操做不受影響，能夠繼續完成已有請求的處理，可是中止接受新請求。

在 Spring Boot 2.3 中增長了新特性優雅中止，目前 Spring Boot 內置的四個嵌入式 Web 服務器（Jetty、Reactor Netty、Tomcat 和 Undertow）以及反應式和基於 Servlet 的 Web 應用程序都支持優雅中止。

下面，咱們先用新版本嘗試下：

Spring Boot 2.3 優雅中止

首先建立一個 Spring Boot 的 Web 項目，版本選擇 2.3.0.RELEASE，Spring Boot 2.3.0.RELEASE 版本內置的 Tomcat 爲 9.0.35。

而後須要在 application.yml 中添加一些配置來啓用優雅中止的功能：

# 開啓優雅中止 Web 容器，默認爲 IMMEDIATE：當即中止
server:
  shutdown: graceful

# 最大等待時間
spring:
  lifecycle:
    timeout-per-shutdown-phase: 30s

其中，平滑關閉內置的 Web 容器（以 Tomcat 爲例）的入口代碼在 org.springframework.boot.web.embedded.tomcat 的 GracefulShutdown 裏，大概邏輯就是先中止外部的全部新請求，而後再處理關閉前收到的請求，有興趣的能夠本身去看下。

內嵌的 Tomcat 容器平滑關閉的配置已經完成了，那麼如何優雅關閉 Spring 容器了，就須要 Actuator 來實現 Spring 容器的關閉了。

而後加入 actuator 依賴，依賴以下所示：

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

而後接着再添加一些配置來暴露 actuator 的 shutdown 接口：

# 暴露 shutdown 接口
management:
  endpoint:
    shutdown:
      enabled: true
  endpoints:
    web:
      exposure:
        include: shutdown

其中經過 Actuator 關閉 Spring 容器的入口代碼在 org.springframework.boot.actuate.context 包下 ShutdownEndpoint 類中，主要的就是執行 doClose() 方法關閉並銷燬 applicationContext，有興趣的能夠本身去看下。

配置搞定後，而後在 controller 包下建立一個 WorkController 類，並有一個 work 方法，用來模擬複雜業務耗時處理流程，具體代碼以下：

@RestController
public class WorkController {

    @GetMapping("/work")
    public String work() throws InterruptedException {
        // 模擬複雜業務耗時處理流程
        Thread.sleep(10 * 1000L);
        return "success";
    }
}

而後，咱們啓動項目，先用 Postman 請求 http://localhost:8080/work 處理業務：

而後在這個時候，調用 http://localhost:8080/actuator/shutdown 就能夠執行優雅地中止，返回結果以下：

{
    "message": "Shutting down, bye..."
}

若是在這個時候，發起新的請求 http://localhost:8080/work，會沒有反應：

再回頭看第一個請求，返回告終果：success。

其中有幾條服務日誌以下：

2020-05-20 23:05:15.163  INFO 102724 --- [     Thread-253] o.s.b.w.e.tomcat.GracefulShutdown        : Commencing graceful shutdown. Waiting for active requests to complete
2020-05-20 23:05:15.287  INFO 102724 --- [tomcat-shutdown] o.s.b.w.e.tomcat.GracefulShutdown        : Graceful shutdown complete
2020-05-20 23:05:15.295  INFO 102724 --- [     Thread-253] o.s.s.concurrent.ThreadPoolTaskExecutor  : Shutting down ExecutorService 'applicationTaskExecutor'

從日誌中也能夠看出來，當調用 shutdown 接口的時候，會先等待請求處理完畢後再優雅地中止。

到此爲止，Spring Boot 2.3 的優雅關閉就講解完了，是否是很簡單呢？若是是在以前不支持優雅關閉的版本如何去作呢？

Spring Boot 舊版本優雅中止

在這裏介紹 GitHub 上 issue 裏 Spring Boot 開發者提供的一種方案：

選取的 Spring Boot 版本爲 2.2.6.RELEASE，首先要實現 TomcatConnectorCustomizer 接口，該接口是自定義 Connector 的回調接口：

@FunctionalInterface
public interface TomcatConnectorCustomizer {

	void customize(Connector connector);
}

除了定製 Connector 的行爲，還要實現 ApplicationListener<ContextClosedEvent> 接口，由於要監聽 Spring 容器的關閉事件，即當前的 ApplicationContext 執行 close() 方法，這樣咱們就能夠在請求處理完畢後進行 Tomcat 線程池的關閉，具體的實現代碼以下：

@Bean
public GracefulShutdown gracefulShutdown() {
    return new GracefulShutdown();
}

private static class GracefulShutdown implements TomcatConnectorCustomizer, ApplicationListener<ContextClosedEvent> {
    private static final Logger log = LoggerFactory.getLogger(GracefulShutdown.class);

    private volatile Connector connector;

    @Override
    public void customize(Connector connector) {
        this.connector = connector;
    }

    @Override
    public void onApplicationEvent(ContextClosedEvent event) {
        this.connector.pause();
        Executor executor = this.connector.getProtocolHandler().getExecutor();
        if (executor instanceof ThreadPoolExecutor) {
            try {
                ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) executor;
                threadPoolExecutor.shutdown();
                if (!threadPoolExecutor.awaitTermination(30, TimeUnit.SECONDS)) {
                    log.warn("Tomcat thread pool did not shut down gracefully within 30 seconds. Proceeding with forceful shutdown");
                }
            } catch (InterruptedException ex) {
                Thread.currentThread().interrupt();
            }
        }
    }
}

有了定製的 Connector 回調，還須要在啓動過程當中添加到內嵌的 Tomcat 容器中，而後等待監聽到關閉指令時執行，addConnectorCustomizers 方法能夠把定製的 Connector 行爲添加到內嵌的 Tomcat 中，具體代碼以下：

@Bean
public ConfigurableServletWebServerFactory tomcatCustomizer() {
    TomcatServletWebServerFactory factory = new TomcatServletWebServerFactory();
    factory.addConnectorCustomizers(gracefulShutdown());
    return factory;
}

到此爲止，內置的 Tomcat 容器平滑關閉的操做就完成了，Spring 容器優雅中止上面已經說過了，再次就再也不贅述了。

經過測試，一樣能夠達到上面那樣優雅中止的效果。

總結

本文主要講解了 Spring Boot 2.3 版本和舊版本的優雅中止，避免強制中止致使正在處理的業務邏輯會被中斷，進而致使產生業務異常的情形。

另外使用 Actuator 的同時要注意安全問題，好比能夠經過引入 security 依賴，打開安全限制並進行身份驗證，設置單獨的 Actuator 管理端口並配置只對內網開放等。

本文的完整代碼在 https://github.com/wupeixuan/SpringBoot-Learn 的 graceful-shutdown 目錄下。

最好的關係就是互相成就，你們的在看、轉發、留言三連就是我創做的最大動力。

參考

https://github.com/spring-projects/spring-boot/issues/4657

https://github.com/wupeixuan/SpringBoot-Learn