多线程之注入hook钩子线程（二十一）

现在很多的软件包括MySql，Zookeeper，Kafka等都存在hook线程的校验机制，目的是校验进程是否已>经启动，防止重复启动我们的程序

Hook线程也称为钩子线程，当JVM推出的时候他会执行hook线程。经常在程序启动的时候创建一个.lock文件，用.lock文件校验程序是否启动,在程序退出(JVM退出时)在删除该.lock文件，在hook线程中除了防止重新启动进程外，还可以做资源释放，尽量避免在hook线程中复杂的操作

dome

package com.dome.hook;

import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.concurrent.TimeUnit;

/**
 * @author qb
 * @version 1.0
 * @Description
 * 通过hook线程防止程序重复启动
 * @date 2021/3/11 17:17
 */
public class Test01 {

    public static void main(String[] args) {

        //1.注入hook线程，在程序退出时删除.lock文件
        Runtime.getRuntime().addShutdownHook(new Thread(){
            @Override
            public void run() {
                System.out.println("JVM退出，会启动当前Hook线程，在Hook线程中删除.lock文件");
                getLockPath().toFile().delete();
            }
        });

        //2.程序运行时检查.lock文件是否存在，如果存在就抛出异常
        if(getLockPath().toFile().exists()){
            throw  new RuntimeException("程序已启动!");
        }else{
            //文件不存在说明程序第一次启动，就创建.lock文件
            try {
                getLockPath().toFile().createNewFile();
                System.out.println("程序在启动时创建了lock文件");
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        //3.模拟程序运行
        for (int i = 0; i < 10; i++) {
            System.out.println("程序正在运行1");

            try {
                TimeUnit.SECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }

    }

    private static Path getLockPath(){
        return Paths.get("","tmp.lock");
    }


}