Java中关闭钩子的有用示例？

java shutdown-hook

2022-08-31 08:29:17

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

答案 1

您可以执行以下操作：

让关机钩子将一些原子布尔（或易失性布尔）“keepRunning”设置为 false
（可选地，如果工作线程在某些阻塞调用中等待数据）.interrupt
通过在工作线程上调用方法，等待工作线程（在您的情况下执行）完成。writeBatchThread.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时拒绝所有客户端”的方式。

从文档中：