Java ExecutorService 用法

Question

我想制作一个如下所示的 2 线程应用程序:

1. 创建2个线程池
2. 在#1 线程上运行一个 foo 方法
3. 等待 2 秒
4. 在 #2 线程上运行 bar 方法

5. 2 线程池最多应等待 20 秒

   ExecutorService executor = Executors.newFixedThreadPool(2);
   executor.submit(() -> foo());
   executor.awaitTermination(2, TimeUnit.SECONDS);
   executor.submit(() -> bar());
   if (!executor.awaitTermination(20, TimeUnit.SECONDS))
       executor.shutdownNow();
   

它不工作，有什么问题吗？

Answer 1

好问题!我认为您对awaitTermination()没有清楚的认识。方法。让我们看看这个document .

Blocks until all tasks have completed execution after a shutdown request, or the timeout occurs, or the current thread is interrupted, whichever happens first.

awaitTermination()不应用于等待所有任务完成。如果那是您正在寻找的语义，请使用 invokeAll()阻塞直到所有任务完成。利用 awaitTermination() 没有事先shutdown()由于 awaitTermination 不当，可能会导致您等待一段时间不关闭执行器。

在这种情况下，您只是试图强制应用程序等待您的线程完成，但如果它没有在您指定的时间范围内结束，您就想继续。这个具体案例让我有点质疑多线程的使用，因为它似乎有点过度设计，但出于教育目的，让我们解释一下如何做到这一点。

基本解决方案

ExecutorService executor = Executors.newFixedThreadPool(2);
executor.submit(() -> foo());
// Thread.sleep(2000); // Wait for 2 seconds
TimeUnit.SECONDS.sleep(2); // Also waits for 2 seconds and a bit more readable
executor.submit(() -> bar());
if (!executor.awaitTermination(20, TimeUnit.SECONDS))
    executor.shutdownNow();

虽然这不是理想的代码，但它会为您提供您正在寻找的语义。

改进

executor.submit()将返回 Future<T>目的。这是 Java 中非常强大的对象，具有很多功能。让我们看看如何改进场景。

ExecutorService executor = Executors.newFixedThreadPool(2);
Future<T> foo = executor.submit(() -> foo());
T result = foo.get(2, TimeUnit.SECONDS); // Attempt to grab the result for 2 seconds, then move on
executor.submit(() -> bar());
if (!executor.awaitTermination(20, TimeUnit.SECONDS))
    executor.shutdownNow();

现在，这是更合适的代码，将为您提供具有更好代码结构的所需语义。此外，如果第一个 future 在 2 秒之前完成，它将提前解除阻塞，这是对基本解决方案的一点改进!

正确关机

假设您要为 executor 等待 20 秒最后，这是您可以做的额外改进。使用以下代码进行关机，因为它来自 ExecutorService 的文档作为关机的例子。在这里，我已将您的等待时间更新为 20 秒。

pool.shutdown(); // Disable new tasks from being submitted
try {
    // Wait a while for existing tasks to terminate
    if (!pool.awaitTermination(20, TimeUnit.SECONDS)) {
      pool.shutdownNow(); // Cancel currently executing tasks
      // Wait a while for tasks to respond to being cancelled
      if (!pool.awaitTermination(20, TimeUnit.SECONDS))
          System.err.println("Pool did not terminate");
    }
} catch (InterruptedException ie) {
    // (Re-)Cancel if current thread also interrupted
    pool.shutdownNow();
    // Preserve interrupt status
    Thread.currentThread().interrupt();
}