Java 中的关闭挂钩如何确保程序正常终止？

在 Java 中利用 Shutdown Hooks 优雅地终止程序

Shutdown hook 提供了一种机制来确保 Java 应用程序在优雅退出之前执行必要的清理任务。在本文中，我们将深入研究关闭挂钩的实际应用，探索我们希望确保在程序终止之前将数据刷新到文件的场景。

考虑以下 Java 应用程序：

<code class="java">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();
        }       
    }

    // Write a batch of numbers to the file
    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(' ');
        }
    }

    public static 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();
        }
    }
}</code>

在此应用程序中，我们将批量数字写入文件。为了确保在程序中断之前完成批处理，我们将使用关闭钩子。

集成关闭钩子

要集成关闭钩子，请按照以下步骤操作：

1. 注册 shutdown hook：使用 Runtime.getRuntime().addShutdownHook() 方法向 JVM 注册 shutdown hook。
2. 实现 shutdown hook: 定义一个线程（Thread 的子类），定义调用钩子时要执行的操作。
3. 在关闭钩子中： 将标志或原子变量设置为 false向主线程发出信号以停止写入文件。
4. 在主线程中： 检查写入循环内的标志或原子变量。当它变为 false 时，停止写入并完成任何挂起的操作。
5. 加入线程：写入完成后，加入主线程（或任何其他工作线程）以确保所有线程都有执行完成。

示例：

添加 keepRunning 变量：

<code class="java">final static volatile boolean keepRunning = true;</code>

修改 run()方法：

<code class="java">//...
for (int i = 0; i < N && keepRunning; ++i)
    writeBatch(pw, i);
//...</code>

在 main() 中添加关闭钩子：

<code class="java">//...
Runtime.getRuntime().addShutdownHook(new Thread() {
    public void run() {
        keepRunning = false;
        mainThread.join();
    }
});
//...</code>

当 JVM 收到关闭信号（例如 Ctrl C ），关闭钩子会将 keepRunning 设置为 false。 run() 方法将继续写入数据，直到 keepRunning 为 false，确保当前批次在程序退出之前完成。

以上是Java 中的关闭挂钩如何确保程序正常终止？的详细内容。更多信息请关注PHP中文网其他相关文章！