web-dev-qa-db-ja.com

Javaのシャットダウンフックの便利な例

私はJavaアプリケーションが堅牢になるために妥当な手順を踏むようにしているが、その一部は正常にシャットダウンすることを伴う。私は shutdown hooks について読んでいますが、実際にそれらを利用する方法は実際にはわかりません。

実用的な例はありますか？

次のような非常に単純なアプリケーションがあり、100個のバッチでファイルに10個、行に10個を書き込み、プログラムが中断された場合に特定のバッチが終了することを確認したいとします。シャットダウンフックを登録する方法はわかりましたが、それをアプリケーションに統合する方法がわかりません。助言がありますか？

package com.example.test.concurrency;

import Java.io.File;
import Java.io.FileNotFoundException;
import Java.io.FileOutputStream;
import Java.io.PrintWriter;

public class GracefulShutdownTest1 {
    final private int N;
    final private File f;
    public GracefulShutdownTest1(File f, int N) { this.f=f; this.N = N; }

    public void run()
    {
        PrintWriter pw = null;
        try {
            FileOutputStream fos = new FileOutputStream(this.f);
            pw = new PrintWriter(fos);
            for (int i = 0; i < N; ++i)
                writeBatch(pw, i);
        }
        catch (FileNotFoundException e) {
            e.printStackTrace();
        }
        finally
        {
            pw.close();
        }       
    }

    private void writeBatch(PrintWriter pw, int i) {
        for (int j = 0; j < 100; ++j)
        {
            int k = i*100+j;
            pw.write(Integer.toString(k));
            if ((j+1)%10 == 0)
                pw.write('\n');
            else
                pw.write(' ');
        }
    }

    static public void main(String[] args)
    {
        if (args.length < 2)
        {
            System.out.println("args = [file] [N] "
                    +"where file = output filename, N=batch count");
        }
        else
        {
            new GracefulShutdownTest1(
                    new File(args[0]), 
                    Integer.parseInt(args[1])
            ).run();
        }
    }
}
javashutdown-hook
119
Jason S

次のことができます。

  • シャットダウンフックでAtomicBoolean（またはvolatileブール値）の「keepRunning」をfalseに設定します
  • （オプションで、.interrupt何らかのブロッキング呼び出しでデータを待機する場合、作業スレッド）
  • 作業スレッドでThread.join()メソッドを呼び出して、作業スレッド（この場合はwriteBatchを実行）が終了するのを待ちます。
  • プログラムを終了する

いくつかの大ざっぱなコード：

  • static volatile boolean keepRunning = true;を追加します

  • run（）に変更します

    for (int i = 0; i < N && keepRunning; ++i)
    writeBatch(pw, i);

  • main（）に追加します：

    final Thread mainThread = Thread.currentThread();
Runtime.getRuntime().addShutdownHook(new Thread() {
    public void run() {
        keepRunning = false;
        mainThread.join();
    }
});

それはおおよそ、ターミナルで「Control-Cを押すとすべてのクライアントを拒否する」という優雅な方法です。

ドキュメント から：

仮想マシンがシャットダウンシーケンスを開始すると、登録されているすべてのシャットダウンフックを指定されていない順序で開始し、同時に実行します。 すべてのフックが終了すると、終了時のファイナライズが有効になっている場合、すべてのフックが起動されていないファイナライザーを実行します。最後に、仮想マシンが停止します。

つまり、シャットダウンフックは、フックが終了する（run（）-メソッドから返される）までJVMを実行し続けます。

150
aioobe