shutdownhook-优雅地停止服务

转载：https://www.cnblogs.com/shuo1208/p/5871224.html

  https://blog.csdn.net/wins22237/article/details/72758644#%E5%85%AD%E4%B8%BA%E4%BD%95%E9%87%8D%E5%90%AF%E6%97%B6%E6%9C%89%E6%97%B6%E4%BC%9A%E6%9C%89classnotfoundexception

 

一、什么是ShutdownHook？
在Java程序中可以通过添加关闭钩子，实现在程序退出时关闭资源、平滑退出的功能。 
使用Runtime.addShutdownHook(Thread hook)方法，可以注册一个JVM关闭的钩子，这个钩子可以在以下几种场景被调用： 
1. 程序正常退出 
2. 使用System.exit() 
3. 终端使用Ctrl+C触发的中断 
4. 系统关闭 
5. 使用Kill pid命令干掉进程

代码示例：
使用Timer模拟一个工作线程，该线程重复工作十次，使用System.exit()退出，在清理现场代码CleanWorkThread 中，取消timer运行，并输出必要的日志信息。

 1 package com.netease.test.java.lang;
 2 
 3 import java.util.Timer;
 4 import java.util.TimerTask;
 5 import java.util.concurrent.atomic.AtomicInteger;
 6 
 7 /**
 8  * Date: 14-6-18
 9  * Time: 11:01
10  * 测试ShutdownHook
11  */
12 public class TestShutdownHook {
13 
14     //简单模拟干活的
15     static Timer timer = new Timer("job-timer");
16 
17     //计数干活次数
18     static AtomicInteger count = new AtomicInteger(0);
19 
20     /**
21      * hook线程
22      */
23     static class CleanWorkThread extends Thread{
24         @Override
25         public void run() {
26             System.out.println("clean some work.");
27             timer.cancel();
28             try {
29                 Thread.sleep(2 * 1000);//sleep 2s
30             } catch (InterruptedException e) {
31                 e.printStackTrace();
32             }
33         }
34     }
35     public static void main(String[] args) throws InterruptedException {
36         //将hook线程添加到运行时环境中去
37         Runtime.getRuntime().addShutdownHook(new CleanWorkThread());
38         System.out.println("main class start ..... ");
39         //简单模拟
40         timer.schedule(new TimerTask() {
41             @Override
42             public void run() {
43                 count.getAndIncrement();
44                 System.out.println("doing job " + count);
45                 if (count.get() == 10) {  //干了10次退出
46                     System.exit(0);
47                 }
48             }
49         }, 0, 2 * 1000);
50 
51     }
52 }

运行后，可以模拟以上五种场景进行测试，只有kill -9 pid不会执行Hook里面的代码。

 

二、java进程平滑退出的意义
很多时候，我们会有这样的一些场景，比如说nginx反向代理若干个负载均衡的web容器，又或者微服务架构中存在的若干个服务节点，需要进行无间断的升级发布。 
在重启服务的时候，除非我们去变更nginx的配置，否则重启很可能会导致正在执行的线程突然中断，本来应该要完成的事情只完成了一半，并且调用方出现错误警告。 
如果能有一种简单的方式，能够让进程在退出时能执行完当前正在执行的任务，并且让服务的调用方将新的请求定向到其他负载节点，这将会很有意义。 
自己注册ShutdownHook可以帮助我们实现java进程的平滑退出。

三、java进程平滑退出的思路
在服务启动时注册自己的ShutdownHook
ShutdownHook在被运行时，首先不接收新的请求，或者告诉调用方重定向到其他节点
等待当前的执行线程运行完毕，如果五秒后仍在运行，则强制退出