在线上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.amugua.test;
import org.slf4j.Logger;
import com.amugua.log.factory.LoggerFactory;
/**
* TODO 类实现描述
* @Company 杭州木瓜科技有限公司
* @className: Test.java
* @author liutao baiying@dianjia.io
* @date 2017年9月4日 下午2:12:59
*/
public class WorkThread extends Thread {
static final Logger LOGGER = LoggerFactory.getLogger(WorkThread.class);
privateboolean flag = true;
class ShutdownCallbackThread extends Thread {
@Override
public void run() {
// flag = false;
LOGGER.error("hook");
while (true) {
}
}//设置关闭筏值
}
@Override
public void run() {
//regist hook
ShutdownCallbackThread hook = new ShutdownCallbackThread();
Runtime.getRuntime().addShutdownHook(hook);
// while (flag) {
// ....working;
LOGGER.info("working");
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
LOGGER.info("working");
// }
}
}
测试类的main:
WorkThread aa = new WorkThread();
aa.start();
Thread.sleep(5000);
运行后,可以模拟以上五种场景进行测试,只有kill -9 pid不会执行Hook里面的代码。