JAVA HOOK（钩子）线程介绍

当jvm进程退出的时候，或者受到了系统的中断信号，hook线程就会启动，一个线程可以注入多个钩子，这是一个实例：

import java.io.IOException;
import java.util.concurrent.TimeUnit;

public class HookTest {
    public static void main(String[] args) throws IOException {
        Runtime.getRuntime().addShutdownHook(new Thread(){
            @Override
            public void run() {
                try {
                    System.out.println("这个钩子启动");
                    TimeUnit.SECONDS.sleep(2);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("这个钩子退出");
            }
        });

    }
}

给这段程序注入了一个hook线程，当main结束时，jvm即将退出的时候，hook线程就会被运行

运行这段main函数，可以打印出两条语句：

这个钩子启动

这个钩子退出

一段程序可以注入多个钩子！

那么，hook线程有什么用作呢？

在开发的时候，为了防止某段程序被重复启动，在进程启动的时候会创建一个lock文件，当程序结束或者收到中断信号的时候，就会触发hook从而删除lock文件，在mysql，zookeeper等系统中都可以看到lock文件的存在，下面是一段示例代码：

package Hook;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.attribute.PosixFilePermission;
import java.nio.file.attribute.PosixFilePermissions;
import java.util.Set;
import java.util.concurrent.TimeUnit;

public class HookTest {
    private final static String LOCK_PATH = "/locks";
    private final static String LOCK_FILE = ".lock";
    private final static String  PERMISSIONS = "rw--------";
    public static void main(String[] args) throws IOException {
        Runtime.getRuntime().addShutdownHook(new Thread(()->{
            //程序结束时删除.lock文件             getLockFile().toFile().delete();
        }));
        //检查是否存在lock文件         checkRunning();
        for (;;){
            try {
                TimeUnit.MILLISECONDS.sleep(1);
                System.out.println("程序正在运行");
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    private static void checkRunning() throws IOException {
        Path path = getLockFile();
        if (path.toFile().exists()){
            throw new RuntimeException("程序正在运行!");
        }
        Set<PosixFilePermission> param = PosixFilePermissions.fromString(PERMISSIONS);
        Files.createFile(path,PosixFilePermissions.asFileAttribute(param));
    }

    //获取文件地址     private static Path getLockFile() {
        return Paths.get(LOCK_PATH,LOCK_FILE);
    }
}

运行这段程序会发生文件夹里面生成了.lock文件

当使用kill pid命令后，jvm会收到中断信号，运行hook线程，删除.lock文件。

注意

• hook线程需要收到中断信号才能开启，如果没有收到中断信号，就不会执行hook，导致.lock不能删除，比如kill -9命令，所以防止程序重复运行，还需要其他手段。
• hook线程也可以做一些资源释放的操作，类似于try-catch-finally，但不要在hook里面运行复杂的程序，这样会导致程序长时间无法退出。