Java 如何實現優雅停服？刨根問底

一猿小講發表於2020-05-20

Java

在 Java 的世界裡遨遊，如果能擁有一雙善於發現的眼睛，有很多東西留心去看，外加耐心助力，仔細去品，往往會品出不一樣的味道。

通過本次分享，能讓你輕鬆 get 如下幾點，絕對收穫滿滿。

a）如何讓 Java 程式實現優雅停服？有思想才是硬道理！

b）addShutdownHook 的使用場景？會用才是王道！

c）addShutdownHook 鉤子函式到底是個啥？刨根問底！

1. 如何讓 Java 程式實現優雅停服？

無論是自研基礎服務框架，還是分析開源專案原始碼，細心的 Java 開發同學，都會發現 Runtime.getRuntime().addShutdownHook 這麼一句程式碼的身影，這句到底是幹什麼用的？

接下來就一起細品，看看它香不香？

阿里開源的資料同步神器 Canal 啟動時的部分原始碼：

Apache 麾下的用於海量日誌收集的 Flume 啟動時的部分原始碼：

仰望了一下開源的專案，不妨從中提煉一下共性（同樣的程式碼遇到多次，勢必會品出味道），寫段程式碼跑跑看（站在 flume 原始碼的肩膀上，起飛）。

 1 import java.util.concurrent.ScheduledThreadPoolExecutor;
 2 import java.util.concurrent.TimeUnit;
 3 
 4 /**
 5  * 體驗 Java 優雅停服
 6  *
 7  * @author 一猿小講
 8  */
 9 public class Application {
10 
11     /**
12      * 監控服務
13      */
14     private ScheduledThreadPoolExecutor monitorService;
15 
16     public Application() {
17         monitorService = new ScheduledThreadPoolExecutor(1);
18     }
19 
20     /**
21      * 啟動監控服務，監控一下記憶體資訊
22      */
23     public void start() {
24         System.out.println(String.format("啟動監控服務 %s", Thread.currentThread().getId()));
25         monitorService.scheduleWithFixedDelay(new Runnable() {
26             @Override
27             public void run() {
28                 System.out.println(String.format("最大記憶體: %dm  已分配記憶體: %dm  已分配記憶體中的剩餘空間: %dm  最大可用記憶體: %dm",
29                         Runtime.getRuntime().maxMemory() / 1024 / 1024,
30                         Runtime.getRuntime().totalMemory() / 1024 / 1024,
31                         Runtime.getRuntime().freeMemory() / 1024 / 1024,
32                         (Runtime.getRuntime().maxMemory() - Runtime.getRuntime().totalMemory() +
33                                 Runtime.getRuntime().freeMemory()) / 1024 / 1024));
34             }
35         }, 2, 2, TimeUnit.SECONDS);
36     }
37 
38     /**
39      * 釋放資源（程式碼來源於 flume 原始碼）
40      * 主要用於關閉執行緒池（看不懂的同學莫糾結，當做黑盒去對待）
41      */
42     public void stop() {
43         System.out.println(String.format("開始關閉執行緒池 %s", Thread.currentThread().getId()));
44         if (monitorService != null) {
45             monitorService.shutdown();
46             try {
47                 monitorService.awaitTermination(10, TimeUnit.SECONDS);
48             } catch (InterruptedException e) {
49                 System.err.println("Interrupted while waiting for monitor service to stop");
50             }
51             if (!monitorService.isTerminated()) {
52                 monitorService.shutdownNow();
53                 try {
54                     while (!monitorService.isTerminated()) {
55                         monitorService.awaitTermination(10, TimeUnit.SECONDS);
56                     }
57                 } catch (InterruptedException e) {
58                     System.err.println("Interrupted while waiting for monitor service to stop");
59                 }
60             }
61         }
62         System.out.println(String.format("執行緒池關閉完成 %s", Thread.currentThread().getId()));
63     }
64 
65     /**
66      * 應用入口
67      */
68     public static void main(String[] args) {
69         Application application = new Application();
70         // 啟動服務（每隔一段時間監控輸出一下記憶體資訊）
71         application.start();
72 
73         // 新增鉤子，實現優雅停服（主要驗證鉤子的作用）
74         final Application appReference = application;
75         Runtime.getRuntime().addShutdownHook(new Thread("shutdown-hook") {
76             @Override
77             public void run() {
78                 System.out.println("接收到退出的訊號，開始打掃戰場，釋放資源，完成優雅停服");
79                 appReference.stop();
80             }
81         });
82         System.out.println("服務啟動完成");
83     }
84 }