Java - ScheduledExecutorService：在终止时执行任务

Question

我使用ScheduledExecutorService 以固定速率执行任务。下面是我的main方法的内容：

RemoteSync updater = new RemoteSync(config);
try  {
    updater.initialise();
    updater.startService(totalTime, TimeUnit.MINUTES);
} catch (Exception e) {
    e.printStackTrace();
}

RemoteSync实现了AutoCloseable（和Runnable）接口，所以我最初使用try-with-resources，像这样：

try (RemoteSync updater = new RemoteSync(config)) {
    ...
} catch (Exception e) {
   e.printStackTrace();
}

但是updater.startService()在调度任务后立即返回，所以updater.close()被提前调用，应用程序退出。

这里是RemoteSync的startService()方法：

public void startService(int rate, TimeUnit unit) {
    ScheduledExecutorService scheduledExecutorService = Executors.newScheduledThreadPool(1);
    service =
        scheduledExecutorService.scheduleWithFixedDelay(this, 1L,
        rate,
        unit);
}

理想情况下，我希望有这样的方法：

scheduledExecutorService.executeAtTermination(Runnable task)

这将允许我在调度程序实际停止时调用close()，不幸的是我不知道这种方法。

我能做的是阻止startService() 方法，如下所示：

while (!scheduledExecutorService.isTerminated()) {
    Thread.sleep(10000);
}

但这感觉肮脏和hackish。

欢迎提出任何建议。

Answer 1

也许您可以使用应用程序关闭挂钩。就像在this 讨论中一样。

您可以在应用程序初始化的某个阶段添加这样的代码：

Runtime.getRuntime().addShutdownHook(new Thread() {
  public void run() {
    >>> shutdown your service here <<<
  }
});

Answer 2

您可以尝试scheduledExecutorService.awaitTermination(Long.MAX_VALUE, TimeUnit.MILLISECONDS), 可能在单独的线程中