当计划的进程完成时，如何正确终止我的 java 程序?

Question

我有一个简单的java程序，用于生成元素，并在特定时间内每X秒将它们插入数据库。

一代是用scheduleAtFixedRate完成的. 其中只有一个。

我希望我的程序在计划任务结束时完全退出。为此，我在取消任务时使用System.exit()，但这是正确的方法吗？

这是我当前的代码：

public static void main(String[] args) throws InterruptedException {
c = generateDbConnection(url, user, password);
if (c != null) {
s = generateDbStatement(c);
} else {
System.out.println("ERROR");
return;
}
initialTimestamp = new Date();
TimeUnit.SECONDS.sleep(1);
generateForAnHour();
}
private final static ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
/**
* Generator thread handler Uses Statement from main function
*/
public static void generateForAnHour() {
final Runnable runner = new Runnable() {
public void run() {
String[][] data = new String[numberOfIds][2];
for (int i = 0; i < numberOfIds; i++) {
data[i] = generateDevice();
insertDevice(s, data[i][0], data[i][1]);
}
quantityOfIds += numberOfIds;
}
};
final ScheduledFuture<?> generatorHandle = scheduler.scheduleAtFixedRate(runner, 0, 5, TimeUnit.SECONDS);
scheduler.schedule(new Runnable() {
public void run() {
generatorHandle.cancel(true);
System.out.println("Scheduled ID generator terminated.");
System.exit(0); //TODO Is it really correct way to do it
}
}, timeToRun, TimeUnit.SECONDS);
}

Answer 1

如果程序具有更多功能，我不确定这是否是停止程序执行的正确方法，但我个人认为这是一种不错的方法。 :D

因此，事实证明，ScheduledExecutorService似乎创建了具有默认ThreadFactory的非守护进程线程，也许我们需要为其提供一个守护进程。

但是，如果我们要调用ExecutorService#shutdown或强制ExecutorService#shutdownNow，它将阻止两个任务执行，从而删除阻止应用程序结束其工作的线程：

private final static ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
public static void main(String[] args) {
// Some service code here
generateForAnHour();
}
public static void generateForAnHour() {
// Some code that does work
final Runnable runner = () -> System.out.println("Running...");
final ScheduledFuture<?> generatorHandle = scheduler.scheduleAtFixedRate(runner, 0, 1, TimeUnit.SECONDS);
// Code that interrupts the worker after a specified time
scheduler.schedule(scheduler::shutdown, 5, TimeUnit.SECONDS);
}

输出：

Running...
Running...
Running...
Running...
Running...
Running...
Process finished with exit code 0

我希望这会有所帮助。 :D

Answer 2

这个答案是次要的，但它有不同的解决问题的方法，所以我认为创建一个单独的答案是合适的。

如果你想在未来有更多的任务，我相信这个解决方案更具可扩展性，更"正确"。

它为runner和interrupter创建守护进程线程。我认为最好为interrupter创建一个普通的线程工厂，但我未能使其正常工作，所以最好坚持我的第一个答案......

在这里，generateForAnHour返回一个用于等待所需时间的Future。

private final static ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1 , new ThreadFactory() {
@Override
public Thread newThread(final Runnable r) {
Thread t = Executors.defaultThreadFactory().newThread(r);
t.setDaemon(true);
return t;
}
});
public static void main(String[] args) throws InterruptedException, ExecutionException {
// Some service code here
generateForAnHour().get();
}
public static ScheduledFuture<Boolean> generateForAnHour() {
// Some code that does work
final Runnable runner = () -> System.out.println("Running...");
final ScheduledFuture<?> generatorHandle = scheduler.scheduleAtFixedRate(runner, 0, 1, TimeUnit.SECONDS);
// Code that interrupts the worker after a specified time
return scheduler.schedule(() -> generatorHandle.cancel(false), 5, TimeUnit.SECONDS);
}

如果您不打电话给Future#get，在最好的情况下，您只会收到一个Running...或根本不会收到。

如果你决定归还runner的未来，你可能会在get()电话中得到讨厌的CancellationException：

Exception in thread "main" java.util.concurrent.CancellationException
at java.base/java.util.concurrent.FutureTask.report(FutureTask.java:121)
at java.base/java.util.concurrent.FutureTask.get(FutureTask.java:191)
at com.xobotun.Test.main(Test.java:30)

我会使用ExecutorService::shutdown方法作为更稳定、更易于理解的方法。 :D