JAVA HOOK(钩子)线程介绍

Java基础

浏览数:503

2019-8-20

当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里面运行复杂的程序,这样会导致程序长时间无法退出。

作者:不减商山(作者)