Java 如何实现优雅停服？刨根问底

时间：2020-05-20 09:34:19 阅读：71 评论：0 收藏：0 [点我收藏+]

标签：不同研发理念设计理念 interrupt eem cond override ice

在 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 }