SpringBoot實現(xiàn)優(yōu)雅停機的流程步驟

更新時間：2024年03月19日 08:41:40 作者：Java中文社群

優(yōu)雅停機（Graceful Shutdown）是指在服務器需要關(guān)閉或重啟時,能夠先處理完當前正在進行的請求,然后再停止服務的操作,本文給大家介紹了SpringBoot實現(xiàn)優(yōu)雅停機的流程步驟,需要的朋友可以參考下

前言

優(yōu)雅停機（Graceful Shutdown） 是指在服務器需要關(guān)閉或重啟時，能夠先處理完當前正在進行的請求，然后再停止服務的操作。

優(yōu)雅停機的實現(xiàn)步驟主要分為以下幾步：

停止接收新的請求：首先，系統(tǒng)會停止接受新的請求，這樣就不會有新的任務被添加到任務隊列中。
處理當前請求：系統(tǒng)會繼續(xù)處理當前已經(jīng)在處理中的請求，確保這些請求能夠正常完成。這通常涉及到等待正在執(zhí)行的任務完成，如處理HTTP請求、數(shù)據(jù)庫操作等。
釋放資源：在請求處理完成后，系統(tǒng)會釋放所有已分配的資源，如關(guān)閉數(shù)據(jù)庫連接、斷開網(wǎng)絡連接等。
關(guān)閉服務：最后，當所有請求都處理完畢且資源都已釋放后，系統(tǒng)會安全地關(guān)閉服務。

0.SpringBoot如何實現(xiàn)優(yōu)雅停機？

優(yōu)雅停機的實現(xiàn)步驟分為以下兩步：

使用合理的 kill 命令，給 Spring Boot 項目發(fā)送優(yōu)雅停機指令。
開啟 Spring Boot 優(yōu)雅停機/自定義 Spring Boot 優(yōu)雅停機的實現(xiàn)。

1.合理殺死進程

在 Linux 中 kill 殺死進程的常用命令有以下這些：

kill -2 pid：向指定 pid 發(fā)送 SIGINT 中斷信號，等同于 ctrl+c。也就說，不僅當前進程會收到該信號，而且它的子進程也會收到終止的命令。
kill -9 pid：向指定 pid 發(fā)送 SIGKILL 立即終止信號。程序不能捕獲該信號，最粗暴最快速結(jié)束程序的方法。
kill -15 pid：向指定 pid 發(fā)送 SIGTERM 終止信號。信號會被當前進程接收到，但它的子進程不會收到，如果當前進程被 kill 掉，它的的子進程的父進程將變成 init 進程 (init 進程是那個 pid 為 1 的進程)。
kill pid：等同于 kill 15 pid。

因此，在以上命令中，我們不能使用“kill -9”來殺死進程，使用“kill”殺死進程即可。

2.設置SpringBoot優(yōu)雅停機

在 Spring Boot 2.3.0 之后，可以通過配置設置開啟 Spring Boot 的優(yōu)雅停機功能，如下所示：

# 開啟優(yōu)雅停機，默認值：immediate 為立即關(guān)閉
server.shutdown=graceful

# 設置緩沖期，最大等待時間，默認：30秒
spring.lifecycle.timeout-per-shutdown-phase=60s

此時，應用在關(guān)閉時，Web 服務器將不再接受新請求，并等待正在進行的請求完成的緩沖時間。

然而，如果是 Spring Boot 2.3.0 之前，就需要自行擴展（線程池）來實現(xiàn)優(yōu)雅停機了。它的核心實現(xiàn)實現(xiàn)是在系統(tǒng)關(guān)閉時會調(diào)用 ShutdownHook，然后在 ShutdownHook 中阻塞 Web 容器的線程池，直到所有請求都處理完畢再關(guān)閉程序，這樣就實現(xiàn)自定義優(yōu)雅線下了。

但是，不同的 Web 容器（Tomcat、Jetty、Undertow）有不同的自定義優(yōu)雅停機的方法，以 Tomcat 為例，它的自定義優(yōu)雅停機實現(xiàn)如下。

2.1 Tomcat 容器關(guān)閉代碼

public class TomcatGracefulShutdown implements TomcatConnectorCustomizer, ApplicationListener<ContextClosedEvent> {
    private volatile Connector connector;

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

    public void onApplicationEvent(ContextClosedEvent contextClosedEvent) {
        this.connector.pause();
        Executor executor = this.connector.getProtocolHandler().getExecutor();
        if (executor instanceof ThreadPoolExecutor) {
            try {
                log.info("Start to shutdown tomcat thread pool");
                ThreadPoolExecutor threadPoolExecutor = (ThreadPoolExecutor) executor;
                threadPoolExecutor.shutdown();
                if (!threadPoolExecutor.awaitTermination(20, TimeUnit.SECONDS)) {
                    log.warn("Tomcat thread pool did not shutdown gracefully within 20 seconds. ");
                }
            } catch (InterruptedException e) {
                log.warn("Fail to shut down tomcat thread pool ", e);
            }
        }
    }
}

2.2 設置 Tomcat 自動裝配

@Configuration
@ConditionalOnClass({Servlet.class, Tomcat.class})
public static class TomcatConfiguration {
    @Bean
    public TomcatGracefulShutdown tomcatGracefulShutdown() {
        return new TomcatGracefulShutdown();
    }

    @Bean
    public EmbeddedServletContainerFactory tomcatEmbeddedServletContainerFactory(TomcatGracefulShutdown gracefulShutdown) {
        TomcatEmbeddedServletContainerFactory tomcatFactory = new TomcatEmbeddedServletContainerFactory();
        tomcatFactory.addConnectorCustomizers(gracefulShutdown);
        return tomcatFactory;
    }
}