Java钩子函数 - Hook线程实战

1、前言

最近小老弟出去面试，被打击了，问到钩子函数是什么，结果答不上来。其实这并不是什么复杂技术，只要用过就会。

2、Hook线程介绍

JVM进程退出是由于JVM进程中没有活跃的非守护线程，或者收到了系统中断信号，向JVM程序注入一个Hook线程，在JVM进程退出的时候，Hook线程会启动执行，通过Runtime可以为JVM注入多个Hook线程。

3、代码示范

package com.摸鱼码长.service.threadDemo;
import java.util.concurrent.TimeUnit;

/**
 * 钩子函数实战
 * 业务场景：在进程退出之前做一些动作，比如资源的关闭，比如发送告警通知
 * 【钩子函数可以注册多个】
 */
public class HookThread {
    public static void main(String[] args) {
        // 钩子线程1
        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
            try {
                System.out.println("hook1线程正在运行");
                TimeUnit.SECONDS.sleep(2);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("hook1线程运行结束");
        }));
        // 钩子线程2
        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
            try {
                System.out.println("hook2线程正在运行");
                TimeUnit.SECONDS.sleep(2);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("hook2线程运行结束");
        }));
        System.out.println("主线程运行结束");
    }
}

核心代码：

Runtime.getRuntime().addShutdownHook(new Thread(() -> { do something… }));

执行结果：

4、应用场景及注意事项

Hook业务场景应用

• 开发中遇到的hook线程，比如为了防止某个程序被重复启动，在进程退出之前做一些动作，比如资源的关闭，比如发送告警通知

注意点：

• Hook线程只有在收到退出信号的时候会被执行，如果在kill的时候使用了参数 -9，那么Hook线程不会得到执行，进程将立即退出，强制结束

• 尽量不要在Hook线程中执行一些耗时非常长的操作，因为会导致程序迟迟不能退出