Java中一个有用的关闭钩子示例是什么？

Question

我正在努力确保我的Java应用程序采取合理的步骤来保证其健壮性，其中一部分涉及优雅地关闭。我正在阅读关机挂钩，但实际上不知道如何在实践中使用它们。

是否有一个实际的例子？

假设我有一个非常简单的应用程序，像下面这个示例一样，它将数字写入文件，每行10个数字，每批100个，并且如果程序被中断，我想确保给定的批次完成。我知道如何注册关机挂钩，但不知道如何将其集成到我的应用程序中。有什么建议吗？

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();
        }
    }
}

- Jason S

---

请参阅 https://dev59.com/-2oy5IYBdhLWcg3wKrCs - flow2k

Answer 1

你可以这样做：

• 让关闭挂钩设置一些原子布尔值（或易失性布尔值）"keepRunning"为false
• （可选）如果工作线程在某些阻塞调用中等待数据，则中断它们
• 通过在工作线程上调用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 的方式。

---

关于文档的说明：

当虚拟机开始其关闭顺序时，它将按某种未指定的顺序启动所有已注册的关闭钩子，并让它们并发运行。当所有钩子都完成后，如果启用了 exit 期间的 finalizer，则会运行所有未调用的 finalizer。最后，虚拟机将停止。

也就是说，关闭钩子使 JVM 保持运行状态，直到钩子终止（从 run() 方法返回）。

- aioobe

---

啊，如果我理解正确，你的意思是关闭挂钩有机会与其他正在运行的线程通信，并且可以防止VM关机（例如使用Thread.join()），直到它满意地完成了快速清理。这是我之前没想到的要点。谢谢！ - Jason S

是的。你懂了。根据文档更新了答案的几个句子。 - aioobe

抱歉提前说一声，GracefulShutdownTest1 应该实现 Runnable 吗？ - Victor

关闭挂钩与GUI代码无关。两者应该可以很好地协同工作。 - aioobe

@MartynasJusevičius，当所有非守护线程终止时，JVM终止。如果mainThread是一个守护线程，则join保证在让JVM关闭之前等待其完成。 - aioobe