Java并發(fā)編程：優(yōu)雅的關(guān)閉鉤子（Shutdown Hook）
作者：寫bug的高哈哈
當程序即將退出時（例如釋放資源、關(guān)閉數(shù)據(jù)庫連接等），可以通過預(yù)先注冊一個或多個關(guān)閉鉤子線程（Shutdown Hook）來執(zhí)行相關(guān)操作。當 JVM 進程準備退出時，這些鉤子線程會被觸發(fā)并運行。
關(guān)閉鉤子簡介
當程序即將退出時（例如釋放資源、關(guān)閉數(shù)據(jù)庫連接等），可以通過預(yù)先注冊一個或多個關(guān)閉鉤子線程（Shutdown Hook）來執(zhí)行相關(guān)操作。當 JVM 進程準備退出時，這些鉤子線程會被觸發(fā)并運行。
示例代碼：
public class HookThreadDemo {
privatestaticclass HookRunnable implements Runnable {
@Override
public void run() {
try {
System.out.println("鉤子線程 " + Thread.currentThread().getName() + " 正在執(zhí)行...");
Thread.sleep(2000L);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("鉤子線程 " + Thread.currentThread().getName() + " 執(zhí)行結(jié)束");
}
}
public static void main(String[] args) {
HookRunnable hookRunnable = new HookRunnable();
// 添加鉤子線程 0
Runtime.getRuntime().addShutdownHook(new Thread(hookRunnable));
// 添加鉤子線程 1
Runtime.getRuntime().addShutdownHook(new Thread(hookRunnable));
System.out.println("主線程即將結(jié)束執(zhí)行");
}
}輸出結(jié)果：
主線程即將結(jié)束執(zhí)行
鉤子線程 Thread-0 正在執(zhí)行...
鉤子線程 Thread-1 正在執(zhí)行...
鉤子線程 Thread-1 執(zhí)行結(jié)束
鉤子線程 Thread-0 執(zhí)行結(jié)束當主線程執(zhí)行完畢后，JVM 進程退出前，所有注冊的鉤子線程會被啟動并執(zhí)行。
關(guān)閉鉤子應(yīng)用場景
- 釋放資源：關(guān)閉文件句柄、數(shù)據(jù)庫連接等，避免資源泄漏。
- 停止服務(wù)：安全關(guān)閉服務(wù)器，確保所有請求處理完畢。
- 發(fā)送通知：通過郵件或短信通知用戶服務(wù)已停止。
- 記錄日志：保存系統(tǒng)狀態(tài)或錯誤信息，便于后續(xù)排查問題。
數(shù)據(jù)庫連接實戰(zhàn)演示
以下代碼演示如何用關(guān)閉鉤子關(guān)閉數(shù)據(jù)庫連接：
public class DatabaseConnection {
privatestatic Connection conn;
public static void main(String[] args) {
System.out.println("主線程開始執(zhí)行");
initConnection(); // 初始化數(shù)據(jù)庫連接
System.out.println("執(zhí)行數(shù)據(jù)查詢與處理");
// 注冊關(guān)閉鉤子
Runtime.getRuntime().addShutdownHook(new Thread(() -> closeConnection()));
System.out.println("主線程結(jié)束執(zhí)行");
}
private static void initConnection() {
try {
Class.forName("com.mysql.cj.jdbc.Driver");
conn = DriverManager.getConnection(
"jdbc:mysql://localhost:3306/school_info?useSSL=true&",
"root", "root"
);
System.out.println("數(shù)據(jù)庫連接成功！");
} catch (ClassNotFoundException | SQLException e) {
e.printStackTrace();
}
}
private static void closeConnection() {
try {
conn.close();
System.out.println("數(shù)據(jù)庫連接已關(guān)閉！");
} catch (SQLException e) {
e.printStackTrace();
}
}
}輸出結(jié)果：
主線程開始執(zhí)行
數(shù)據(jù)庫連接成功！
執(zhí)行數(shù)據(jù)查詢與處理
主線程結(jié)束執(zhí)行
數(shù)據(jù)庫連接已關(guān)閉！使用關(guān)閉鉤子的注意事項
- 強制終止進程（如kill -9）不會觸發(fā)鉤子線程。
- 避免耗時操作：鉤子線程中不要執(zhí)行長時間任務(wù)，否則會延遲 JVM 退出。
- 禁止異常拋出：鉤子線程中的異?？赡軐е?JVM 無法正常退出。
- 注冊順序：按依賴關(guān)系注冊鉤子，先注冊簡單任務(wù)，后注冊復(fù)雜任務(wù)。
- 避免啟動新線程：在鉤子中啟動新線程可能導致 JVM 無法正常關(guān)閉。
開源框架中的關(guān)閉鉤子機制
1. Spring
在AbstractApplicationContext中，registerShutdownHook()方法注冊鉤子，用于關(guān)閉上下文：
public void registerShutdownHook() {
if (this.shutdownHook == null) {
this.shutdownHook = new Thread(() -> doClose());
Runtime.getRuntime().addShutdownHook(this.shutdownHook);
}
}2. Tomcat
Tomcat 通過注冊鉤子確保服務(wù)關(guān)閉時釋放資源：
public void registerShutdownHook() {
if (this.shutdownHook == null) {
this.shutdownHook = new Thread(() -> {
synchronized (startupShutdownMonitor) {
doClose();
}
});
Runtime.getRuntime().addShutdownHook(this.shutdownHook);
}
}關(guān)閉鉤子機制的原理
JVM 啟動時，主線程會創(chuàng)建一個關(guān)閉線程（Shutdown Thread），并將所有注冊的鉤子添加到其任務(wù)列表中。當 JVM 收到終止信號時：
- 停止所有用戶線程。
- 啟動關(guān)閉線程，按順序執(zhí)行鉤子任務(wù)。
- 等待所有鉤子執(zhí)行完畢或超時后退出。
鉤子的注冊與執(zhí)行
- 注冊：通過Runtime.getRuntime().addShutdownHook(Thread)將線程添加到ApplicationShutdownHooks的靜態(tài)列表中。
- 執(zhí)行：關(guān)閉線程按順序同步執(zhí)行系統(tǒng)級鉤子，異步執(zhí)行應(yīng)用級鉤子，并等待所有線程完成。
關(guān)閉鉤子的觸發(fā)時機
- 主動調(diào)用：通過Runtime.exit()或System.exit()觸發(fā)。
- 信號捕獲：JVM 注冊信號處理器（如INT、TERM），捕獲kill命令發(fā)送的信號后觸發(fā)。
示例代碼（捕獲信號）：
public class SignalHandlerTest implements SignalHandler {
public static void main(String[] args) {
Runtime.getRuntime().addShutdownHook(new Thread(() ->
System.out.println("關(guān)閉鉤子正在運行...")));
SignalHandler handler = new SignalHandlerTest();
Signal.handle(new Signal("INT"), handler); // 捕獲 Ctrl+C
Signal.handle(new Signal("TERM"), handler); // 捕獲 kill 命令
while (true) {
System.out.println("主線程運行中...");
Thread.sleep(2000);
}
}
@Override
public void handle(Signal signal) {
System.out.println("接收到信號：" + signal.getName() + "-" + signal.getNumber());
System.exit(0);
}
}輸出示例：
主線程運行中...
主線程運行中...
^C接收到信號：INT-2
關(guān)閉鉤子正在運行...信號處理與守護線程
- 信號不可捕獲的情況：KILL（9）和QUIT（3）無法被捕獲。
- 守護線程：JVM 在所有用戶線程結(jié)束后自動退出，守護線程（如 GC 線程）不會阻止 JVM 退出。
總結(jié)
Java 的關(guān)閉鉤子機制覆蓋了大部分退出場景，但以下情況例外：
- 使用kill -9強制終止進程時，鉤子不會執(zhí)行。
- 信號處理需調(diào)用System.exit()確保進程退出。
通過合理使用關(guān)閉鉤子，可以實現(xiàn)資源釋放、服務(wù)優(yōu)雅關(guān)閉等關(guān)鍵功能。
責任編輯：武曉燕
來源：
程序猿技術(shù)充電站
