Java 如何实现优雅停服?刨根问底
追忆似水年华 · 107浏览量 · 发布于2020-05-20 +关注

在 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             @Override27             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             @Override77             public void run() {78                 System.out.println("接收到退出的讯号,开始打扫战场,释放资源,完成优雅停服");79                 appReference.stop();80             }81         });82         System.out.println("服务启动完成");83     }84 }

复制代码

经常读文的我很清楚,耐心读文章中源码的同学应该很少,所以我还是用图给你简单捋一捋。

标注1:start 方法利用线程池启动一个线程去定时监控内存信息;

标注2:stop 方法用于在退出程序之前,进行关闭线程池进而释放资源。

程序跑起来,效果如下。

 

当进行 kill 操作时,程序确实进行了资源释放,效果确实很优雅。

 

一切看似那么自然,一切又是那么完美,这是真的吗?杀进程时候如果用 kill -9,这种情况下会发生什么现象呢?

 

呜呼!结果不会骗人的,当用 kill -9 的时候,就显得很粗暴了,压根不管什么资源释放,不管三七二十一,就是终止程序。

估计很多同学,都擅长用 kill -9 进行杀进程,为了线上的应用安全,还是用 kill -15 命令杀进程吧,这样会给应用留点时间去打扫一下战场,释放一下资源。

好了,通过仔细品味,借助 JDK 自带的 addShutdownHook 来助力应用,确实能让线上服务跑起来很优雅。

有思想才是硬道理!

2. addShutdownHook 的使用场景?

通过代码试验,能够感知 addShutdownHook(new Thread(){}) 是 JVM 销毁前要执行的一个线程,那么只要是涉及到资源回收的场景,应该都可以满足,下面简单列举几个。

a)数据同步神器 Canal 借助它,来进行关闭 socket 链接、释放 canal 的工作节点、清理缓存信息等;

b)海量日志收集 Flume 借助它,来实现线程池资源关闭、工作线程停止等;

c)在应用正常退出时,执行特定的业务逻辑、关闭资源等操作。

d)在 OOM 宕机、 CTRL+C、或执行 kill pid,导致 JVM 非正常退出时,加入必要的挽救措施成为可能。

其实,在 Java 的世界里遨游,只有想不到的,没有做不到的!

3. addShutdownHook 钩子函数是个啥?

刨根还要问到底!

 

Hook 翻译过来是「钩子」的意思,那顾名思义就是用来挂东西的。

 

如图所示,在现实生活中,要制作腊肉,首先用钩子把肉勾住,然后挂在竹竿上,这应该是钩子的作用。

生活如此,一切设计理念都源于生活,在 Java 的世界里,亦是如此。

 

如上图 Runtime 的源码所示,遵循 Java 的核心思想「一切皆是对象」,那么可以把 addShutdownHook 方法可以视作挂钩子,其实称之为钩子函数会好一些,而现实生活中的肉就可以抽象为释放资源的线程。

只要有这个钩子函数,对外就提供了扩展能力,研发人员就可以往钩子上挂各种自定义的场景实现,这种设计你细品那绝对是香!这也就是 Canal、Flume、Tomcat 等不同应用,在优雅停服时有着不同的实现的原因吧。

大白话,钩子函数有了,想挂什么东西,根据心情自己定就好了。

再深入去刨会发现,由于底层数据结构采用 Map 来进行存储,那么就支持研发人员挂多个 shutdownHook 的实现,又带来了无限的可能性(又带来了无限的「刺激」,自己好好去体会)。

 

好了,避免头大,就刨到这儿吧,感兴趣的可自行顺着思路继续刨下去。

4. 寄语,写在最后

作为研发人员:要拥有一双善于发现的眼睛,要善于发现代码之美。

作为研发人员:要时常思考面对当前的项目,是否能够简单重构让程序跑的更顺溜。

作为研发人员:要多看、多悟、多提炼、多实践。

作为研发人员:请不要放弃代码,因为程序终会铸就人生。

相关推荐

PHP实现部分字符隐藏

· 289浏览 · 2019-04-28 09:47:56
Java中ArrayList和LinkedList区别

kenrry1992 · 245浏览 · 2019-05-08 21:14:54
Tomcat 下载及安装配置

manongba · 277浏览 · 2019-05-13 21:03:56
JAVA变量介绍

manongba · 245浏览 · 2019-05-13 21:05:52
什么是SpringBoot

iamitnan · 207浏览 · 2019-05-14 22:20:36
加载中

0评论

评论