130

我试图确保我的 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();
        }
    }
}
4

2 回答 2

168

您可以执行以下操作：

  • 让关闭钩子将一些 AtomicBoolean（或 volatile boolean）“keepRunning”设置为 false
  • （可选地，.interrupt如果工作线程在某些阻塞调用中等待数据）
  • 通过调用工作线程上的方法，等待工作线程（writeBatch在您的情况下正在执行）完成Thread.join()。
  • 终止程序

一些粗略的代码：

  • 添加一个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 时拒绝所有客户端”。

从文档：

当虚拟机开始其关闭序列时，它将以某种未指定的顺序启动所有已注册的关闭挂钩并让它们同时运行。当所有钩子都完成后，如果 finalization-on-exit 已启用，它将运行所有未调用的终结器。最后，虚拟机将停止。

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

于 2010-05-27T14:31:07.710 回答
-10

关闭挂钩是在 Runtime.addShutdownHook() 注册的未启动线程。JVM 不保证关闭挂钩的启动顺序。有关更多信息，请参阅http://techno-terminal.blogspot.in/2015/08 /shutdown-hooks.html

于 2015-08-30T09:42:40.297 回答