在System.exit(1);之前运行的线程Hook线程
private static final Thread EXIT_TRACK_THREAD = new Thread(){
@Override
public void run() {
//退出之前要做的事情
}
};
try {
Runtime.getRuntime().addShutdownHook(EXIT_TRACK_THREAD);
} catch (IllegalArgumentException e) {
//如果存在定义的HOOK线程则需要移除
try {
boolean cleanFlag = Runtime.getRuntime().removeShutdownHook(EXIT_TRACK_THREAD);
if (cleanFlag == true) {
Runtime.getRuntime().addShutdownHook(EXIT_TRACK_THREAD);
}
} catch (IllegalArgumentException innerE) {
Utils.printException(innerE);
} catch (IllegalStateException innerE) {
Utils.printException(innerE);
}
} catch (IllegalStateException e) {
Utils.printException(e);
}
在线上Java程序中经常遇到进程程挂掉,一些状态没有正确的保存下来,这时候就需要在JVM关掉的时候执行一些清理现场的代码。Java中得ShutdownHook提供了比较好的方案。
JDK在1.3之后提供了Java Runtime.addShutdownHook(Thread hook)方法,可以注册一个JVM关闭的钩子,这个钩子可以在以下几种场景被调用:
1)程序正常退出
2)使用System.exit()
3)终端使用Ctrl+C触发的中断
4)系统关闭
5)使用Kill pid命令干掉进程
注:在使用kill -9 pid是不会JVM注册的钩子不会被调用。
在JDK中方法的声明:
public void addShutdownHook(Thread hook)
参数
hook – 一个初始化但尚未启动的线程对象,注册到JVM钩子的运行代码。
异常
IllegalArgumentException – 如果指定的钩已被注册,或如果它可以判定钩已经运行或已被运行
IllegalStateException – 如果虚拟机已经是在关闭的过程中
SecurityException – 如果存在安全管理器并且它拒绝的RuntimePermission(“shutdownHooks”)
代码示例:
使用Timer模拟一个工作线程,该线程重复工作十次,使用System.exit()退出,在清理现场代码CleanWorkThread 中,取消timer运行,并输出必要的日志信息。
package com.netease.test.java.lang;
import java.util.Timer;
import java.util.TimerTask;
import java.util.concurrent.atomic.AtomicInteger;
/**
* Date: 14-6-18
* Time: 11:01
* 测试ShutdownHook
*/
public class TestShutdownHook {
//简单模拟干活的
static Timer timer = new Timer("job-timer");
//计数干活次数
static AtomicInteger count = new AtomicInteger(0);
/**
* hook线程
*/
static class CleanWorkThread extends Thread{
@Override
public void run() {
System.out.println("clean some work.");
timer.cancel();
try {
Thread.sleep(2 * 1000);//sleep 2s
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
public static void main(String[] args) throws InterruptedException {
//将hook线程添加到运行时环境中去
Runtime.getRuntime().addShutdownHook(new CleanWorkThread());
System.out.println("main class start ..... ");
//简单模拟
timer.schedule(new TimerTask() {
@Override
public void run() {
count.getAndIncrement();
System.out.println("doing job " + count);
if (count.get() == 10) { //干了10次退出
System.exit(0);
}
}
}, 0, 2 * 1000);
}
}
运行后,可以模拟以上五种场景进行测试,只有kill -9 pid不会执行Hook里面的代码。