java 钩子线程_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 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里面运行复杂的程序，这样会导致程序长时间无法退出。

作者：不减商山(作者)