并发编程——Hook线程

Hook线程

Hook线程，通常也被成为钩子。

Hook介绍

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

Hook创建

Runtime.getRuntime().addShutdownHook(new Thread(() -> {
    ...
    //处理方法
}));

当方法运行结束之后，也就是jvm进程退出的时候，Hook线程，就会执行，当然如果有多个Hook线程，运行的是无序的。

Hook实例

import java.util.concurrent.TimeUnit;

/**
 * @Author: zcl
 * @Date: 2022-01-29 10:19
 */
public class HookThread {

    public static void main(String[] args) {
        createHook();
        System.out.println("The main process is over!");
    }

    public static void createHook() {
        Runtime.getRuntime().addShutdownHook(new Thread(() -> {

            System.out.println("This is a hook thread:" + Thread.currentThread().getName());
            try {
                TimeUnit.SECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("This is a hook exit" + Thread.currentThread().getName());
        }, "Thread-Hook-1"));

        Runtime.getRuntime().addShutdownHook(new Thread(() -> {
            System.out.println("This is a hook thread:" + Thread.currentThread().getName());
            try {
                TimeUnit.SECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("This is a hook exit" + Thread.currentThread().getName());
        }, "Thread-Hook-2"));
    }

}

运行结果如下：

The main process is over!
This is a hook thread:Thread-Hook-1
This is a hook thread:Thread-Hook-2
This is a hook exitThread-Hook-2
This is a hook exitThread-Hook-1

通过运行结果可以看出，hook线程运行是交替运行，也就是并行状态的。同时也是当主线程运行结束之后，才会调用，可以通过打上断点测试下。

Hook应用

开发中遇到的hook线程，比如为了防止某个程序被重复启动， 在进程启动时创建一个lock文件，进程收到中断信号之后删除这个文件。也可以执行一些资源释放的工作，比如关闭文件句柄、socket链接、数据库connection等。

注意点：

1. Hook线程只有在收到退出信号的时候会被执行，如果在kill的时候使用了参数 -9，那么Hook线程不会得到执行，进程将立即退出，强制结束。
2. 尽量不要在Hook线程中执行一些耗时非常长的操作，因为会导致程序迟迟不能退出。

学习自：
《Java高并发编程详解——多线程与架构设计》 汪文君