利用 java.lang.Runtime.addShutdownHook() 鈎子程序，保證java程序安全退出

以前在開發時只知道依靠數據庫事務來保證程序關閉時數據的完整性。

但有些時候一個業務上要求的原子操作，不一定只包括數據庫，比如外部接口或者消息隊列。此時數據庫事務就無能為力了。

這時我們可以依靠java提供的一個工具方法：java.lang.Runtime.addShutdownHook(Thread hook)

addShutdownHook方法可以加入一個鈎子，在程序退出時觸發該鈎子。

（退出是指ctrl+c或者kill -15，但如果用kill -9 那是沒辦法的，具體有關kill的signal機制有篇大牛的文章《Linux 信號signal處理機制》）

鈎子做什么操作都可以，甚至可以循環檢查某個線程的狀態，直到業務線程正常退出，再結束鈎子程序就可以保證業務線程的完整性

例子程序如下：

實例程序在執行過程中按下ctrl -c或者 kill -15，由於鈎子程序的循環檢測，能夠保證線程執行完畢后，程序才關閉。

/**
 * Created by IntelliJ IDEA.
 * User: Luo
 * Date: 13-7-11
 * Time: 下午3:12
 */

public class TestShutdownHook {

    /**
     * 測試線程，用於模擬一個原子操作
     */
    private static class TaskThread extends Thread {
        @Override
        public void run() {
            System.out.println("thread begin ...");
            TestShutdownHook.sleep(1000);
            System.out.println("task 1 ok ...");
            TestShutdownHook.sleep(1000);
            System.out.println("task 2 ok ...");
            TestShutdownHook.sleep(1000);
            System.out.println("task 3 ok ...");
            TestShutdownHook.sleep(1000);
            System.out.println("task 4 ok ...");
            TestShutdownHook.sleep(1000);
            System.out.println("task 5 ok ...");

            System.out.println("thread end\n\n");
        }
    }

    /**
     * 注冊hook程序，保證線程能夠完整執行
     * @param checkThread
     */
    private static void addShutdownHook(final Thread checkThread) {
        //為了保證TaskThread不在中途退出，添加ShutdownHook
        Runtime.getRuntime().addShutdownHook(new Thread() {
            public void run() {
                System.out.println("收到關閉信號，hook起動，開始檢測線程狀態 ...");
                //不斷檢測一次執行狀態，如果線程一直沒有執行完畢，超時后，放棄等待       \
                for (int i = 0; i < 100; i++) {
                    if (checkThread.getState() == State.TERMINATED) {
                        System.out.println("檢測到線程執行完畢，退出hook");
                        return;
                    }
                    TestShutdownHook.sleep(100);
                }
                System.out.println("檢測超時，放棄等待，退出hook，此時線程會被強制關閉");
            }
        });
    }


    private static void sleep(long millis) {
        try {
            Thread.sleep(millis);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) throws InterruptedException {
        final TaskThread taskThread = new TaskThread();
        //為了保證TaskThread不在中途退出，添加ShutdownHook
        addShutdownHook(taskThread);
        //執行TaskThread
        taskThread.start();
    }

}