Hook鉤子線程注入

現(xiàn)在很多軟件包括MySQL, Zookeeper, kafka等都存在Hook線程的校驗機制, 目的是校驗進程是否已啟動,防止重復(fù)啟動程序。

Hook線程也稱為鉤子線程, 當(dāng)JVM退出的時候會執(zhí)行Hook線程.經(jīng)常在程序啟動時創(chuàng)建一個.lock文件, 用.lock文件校驗程序是否啟動,在程序退出(JVM退出)時刪除該.lock文件, 在Hook線程中除了防止重新啟動進程外,還可以做資源釋放, 盡量避免在Hook線程中進行復(fù)雜的操作。

package com.wkcto.hook;

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

/**
 * 通過Hook線程防止程序重復(fù)啟動
 */
public class Test {
    public static void main(String[] args) {
        //1)注入Hook線程,在程序退出時刪除.lock文件
        Runtime.getRuntime().addShutdownHook(new Thread(){
            @Override
            public void run() {
                System.out.println("JVM退出,會啟動當(dāng)前Hook線程,在Hook線程中刪除.lock文件");
                getLockFile().toFile().delete();
            }
        });

        //2)程序運行時,檢查lock文件是否存在,如果lock文件存在,則拋出異常
        if ( getLockFile().toFile().exists()){
            throw  new RuntimeException("程序已啟動");
        }else {     //文件不存在,說明程序是第一次啟動,創(chuàng)建lock文件
            try {
                getLockFile().toFile().createNewFile();
                System.out.println("程序在啟動時創(chuàng)建了lock文件");
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        //模擬程序運行
        for (int i = 0; i < 10; i++) {
            System.out.println("程序正在運行");
            try {
                TimeUnit.SECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

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