java 线程钩子_利用 java.lang.Runtime.addShutdownHook() 钩子程序，保证java程序安全退出...

最新推荐文章于 2021-09-25 19:51:14 发布

肥白白

最新推荐文章于 2021-09-25 19:51:14 发布

阅读量142

点赞数

文章标签： java 线程钩子

本文链接：https://blog.csdn.net/weixin_32631179/article/details/115043694

版权

本文介绍了如何利用Java的Runtime.addShutdownHook()方法创建线程钩子，确保在程序退出时能执行特定操作，如检查外部接口或消息队列的原子操作。通过示例代码展示了如何设置钩子程序，使其在接收到关闭信号后循环检测业务线程的状态，直到线程执行完毕才结束程序，从而保证业务线程的完整性。

摘要由CSDN通过智能技术生成

以前在开发时只知道依靠数据库事务来保证程序关闭时数据的完整性。

但有些时候一个业务上要求的原子操作，不一定只包括数据库，比如外部接口或者消息队列。此时数据库事务就无能为力了。

这时我们可以依靠java提供的一个工具方法：java.lang.Runtime.addShutdownHook(Thread hook)

addShutdownHook方法可以加入一个钩子，在程序退出时触发该钩子。

(退出是指ctrl+c或者kill -15，但如果用kill -9 那是没办法的，具体有关kill的signal机制有篇大牛的文章《Linux 信号signal处理机制》)

钩子做什么操作都可以，甚至可以循环检查某个线程的状态，直到业务线程正常退出，再结束钩子程序就可以保证业务线程的完整性

例子程序如下：

实例程序在执行过程中按下ctrl -c或者 kill -15，由于钩子程序的循环检测，能够保证线程执行完毕后，程序才关闭。

937a03f92563

1/**2* Created by IntelliJ IDEA.3* User: Luo4* Date: 13-7-115* Time: 下午3:126*/78publicclassTestShutdownHook {910/**11* 测试线程，用于模拟一个原子操作12*/13privatestaticclassTaskThreadextendsThread {14@Override15publicvoidrun() {16System.out.println("thread begin ...");17TestShutdownHook.sleep(1000);18System.out.println("task 1 ok ...");19TestShutdownHook.sleep(1000);20System.out.println("task 2 ok ...");21TestShutdownHook.sleep(1000);22System.out.println("task 3 ok ...");23TestShutdownHook.sleep(1000);24System.out.println("task 4 ok ...");25TestShutdownHook.sleep(1000);26System.out.println("task 5 ok ...");2728System.out.println("thread end\n\n");29}30}3132/**33* 注册hook程序，保证线程能够完整执行34*@paramcheckThread35*/36privatestaticvoidaddShutdownHook(finalThread checkThread) {37//为了保证TaskThread不在中途退出，添加ShutdownHook38Runtime.getRuntime().addShutdownHook(newThread() {39publicvoidrun() {40System.out.println("收到关闭信号，hook起动，开始检测线程状态 ...");41//不断检测一次执行状态，如果线程一直没有执行完毕，超时后，放弃等待 \42for(inti = 0; i < 100; i++) {43if(checkThread.getState() ==State.TERMINATED) {44System.out.println("检测到线程执行完毕，退出hook");45return;46}47TestShutdownHook.sleep(100);48}49System.out.println("检测超时，放弃等待，退出hook，此时线程会被强制关闭");50}51});52}535455privatestaticvoidsleep(longmillis) {56try{57Thread.sleep(millis);58}catch(InterruptedException e) {59e.printStackTrace();60}61}6263publicstaticvoidmain(String[] args)throwsInterruptedException {64finalTaskThread taskThread =newTaskThread();65//为了保证TaskThread不在中途退出，添加ShutdownHook66addShutdownHook(taskThread);67//执行TaskThread68taskThread.start();69}7071}

937a03f92563