大家好，我是指北君。

通常，啟動(dòng)一個(gè)服務(wù)是很容易的。然而，有時(shí)我們需要有一個(gè)計(jì)劃來優(yōu)雅地關(guān)閉一個(gè)服務(wù)。

在本教程中，我們將看一下 JVM 應(yīng)用程序終止的不同方式。然后，我們將使用 Java APIs 來管理 JVM 關(guān)閉鉤子。

關(guān)閉 JVM

JVM 可以通過兩種不同的方式被關(guān)閉。

• 一種受控的方式
• 一種非受控的方式

一個(gè)受控的進(jìn)程在以下兩種情況下關(guān)閉 JVM。

• 最后一個(gè)非 daemon 線程終止。例如，當(dāng)主線程退出時(shí)，JVM 開始其關(guān)閉進(jìn)程
• 從操作系統(tǒng)發(fā)送一個(gè)中斷信號(hào)。例如，通過按 Ctrl + C 或注銷操作系統(tǒng)
• 從 Java 代碼中調(diào)用 System.exit()

雖然我們都在努力爭(zhēng)取優(yōu)雅的關(guān)閉，但有時(shí) JVM 可能會(huì)以突然和意外的方式關(guān)閉。JVM 會(huì)在以下情況下突然關(guān)閉。

• 從操作系統(tǒng)發(fā)送一個(gè) kill 信號(hào)。例如，通過發(fā)出 kill -9 的信號(hào)
• 從 Java 代碼中調(diào)用 Runtime.getRuntime().halt() 。
• 主機(jī)操作系統(tǒng)意外關(guān)閉，例如，在電源故障或操作系統(tǒng)崩潰的情況下

shutdown hook

JVM 允許在完成關(guān)機(jī)之前運(yùn)行注冊(cè)函數(shù)。這些函數(shù)通常是釋放資源或其他類似的內(nèi)部管理任務(wù)的好地方。在 JVM 的術(shù)語(yǔ)中，這些函數(shù)被稱為關(guān)閉鉤子。

關(guān)閉鉤子基本上是初始化但未啟動(dòng)的線程。當(dāng)JVM開始其關(guān)閉過程時(shí)，它將以一個(gè)未指定的順序啟動(dòng)所有注冊(cè)的鉤子。在運(yùn)行完所有鉤子后，JVM 將停止運(yùn)行。

添加鉤子

為了添加一個(gè)關(guān)閉鉤子，我們可以使用 Runtime.getRuntime().addShutdownHook() 方法。

Thread printingHook = new Thread(() -> System.out.println("我要關(guān)閉了"));
Runtime.getRuntime().addShutdownHook(printingHook);

在這里，我們只是在JVM自行關(guān)閉之前向標(biāo)準(zhǔn)輸出端打印一些東西。如果我們像下面這樣關(guān)閉JVM。

System.exit(123);

我要關(guān)閉了

然后我們會(huì)看到，鉤子實(shí)際上是將消息打印到標(biāo)準(zhǔn)輸出。

JVM負(fù)責(zé)啟動(dòng)鉤子線程。因此，如果給定的鉤子已經(jīng)被啟動(dòng)了，Java將拋出一個(gè)異常。

Thread longRunningHook = new Thread(() -> {
    try {
        Thread.sleep(300);
    } catch (InterruptedException ignored) {}
});
longRunningHook.start();

assertThatThrownBy(() -> Runtime.getRuntime().addShutdownHook(longRunningHook))
  .isInstanceOf(IllegalArgumentException.class)
  .hasMessage("鉤子正在運(yùn)行");

很明顯，我們也不能多次注冊(cè)一個(gè)鉤子。

Thread unfortunateHook = new Thread(() -> {});
Runtime.getRuntime().addShutdownHook(unfortunateHook);

assertThatThrownBy(() -> Runtime.getRuntime().addShutdownHook(unfortunateHook))
  .isInstanceOf(IllegalArgumentException.class)
  .hasMessage("鉤子已經(jīng)注冊(cè)");

刪除鉤子

Java 提供了一個(gè)孿生的移除方法，以便在注冊(cè)一個(gè)特定的關(guān)閉鉤子后將其移除。

Thread willNotRun = new Thread(() -> System.out.println("鉤子不會(huì)運(yùn)行的"));
Runtime.getRuntime().addShutdownHook(willNotRun);

assertThat(Runtime.getRuntime().removeShutdownHook(willNotRun)).isTrue();

當(dāng)關(guān)閉鉤子被成功刪除時(shí)，removeShutdownHook() 方法返回true。

注意事項(xiàng)

JVM 只在正常終止的情況下運(yùn)行關(guān)閉鉤子。因此，當(dāng)外部力量突然殺死JVM進(jìn)程時(shí)，JVM將沒有機(jī)會(huì)執(zhí)行關(guān)閉鉤子。此外，從Java代碼中停止JVM也會(huì)產(chǎn)生同樣的效果。

Thread haltedHook = new Thread(() -> System.out.println("強(qiáng)行終止"));
Runtime.getRuntime().addShutdownHook(haltedHook);

Runtime.getRuntime().halt(123);

halt 方法強(qiáng)行終止了當(dāng)前運(yùn)行的JVM。因此，注冊(cè)的關(guān)閉鉤子不會(huì)有機(jī)會(huì)執(zhí)行。

總結(jié)

在本教程中，我們研究了 JVM 應(yīng)用程序可能終止的不同方式。然后，我們使用一些運(yùn)行時(shí)API來注冊(cè)和取消注冊(cè)關(guān)閉鉤子。