我试图确保我的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(); } } }
你可以执行以下操作:
AtomicBoolean
.interrupt
writeBatch
Thread.join()
一些粗略的代码:
static volatile boolean keepRunning = true
for (int i = 0; i < N && keepRunning; ++i) writeBatch(pw, i);
final Thread mainThread = Thread.currentThread(); Runtime.getRuntime().addShutdownHook(new Thread() { public void run() { keepRunning = false; mainThread.join(); } });
这大概就是我在终端中优雅地“击中Control-C时拒绝所有客户端”的方式。