转载:https://www.cnblogs.com/shuo1208/p/5871224.html
一、什么是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在被运行时,首先不接收新的请求,或者告诉调用方重定向到其他节点
等待当前的执行线程运行完毕,如果五秒后仍在运行,则强制退出