Java中的关闭挂钩的有用示例?


125

我试图确保我的Java应用程序采取合理的步骤来增强功能,并且其中一部分涉及正常关闭。我正在阅读有关关机挂钩的信息,但实际上并没有实际使用它们的方法。

有没有实际的例子?

假设我有一个非常简单的应用程序,例如下面的程序,它以100的批数将数字写入文件,将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();
        }
    }
}

Answers:


160

您可以执行以下操作:

  • 让关闭挂钩将一些AtomicBoolean(或易失布尔)“ 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时拒绝所有客户端”的方式。


文档

当虚拟机开始其关闭序列时,它将以未指定的顺序启动所有已注册的关闭挂钩,并使其同时运行。当所有挂钩完成后,如果启用了退出时终结,则它将运行所有未调用的终结器。最后,虚拟机将停止。

也就是说,关闭挂钩使JVM一直运行,直到挂钩终止(从run()方法返回


14
嗯-如果我理解正确,那么您要说的要点是,关闭挂钩有机会与其他当前正在运行的线程进行通信,并阻止VM关闭(例如,通过使用Thread.join() ),直到确信已完成快速清除。这就是我所缺少的。谢谢!
杰森·S

2
是。你说对了。用文档中的一些句子更新了答案。
aioobe 2010年

1
预先抱歉,但GracefulShutdownTest1是否应实现Runnable?
维克多

在主线程中启动GUI也行吗?我知道这不是最好的方法(改用EDT),但是我正在处理旧代码。
阿戈斯蒂诺2014年

1
@MartynasJusevičius,当所有非守护程序线程都终止时,JVM终止。如果mainThreads是守护程序线程,则可以join保证在JVM关闭之前我们等待它完成。
aioobe

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.