ExecutorService线程池正确关闭方法

本文旨在解决Java ExecutorService线程池在使用过程中，shutdown()方法提前结束导致任务未完成的问题。通过控制shutdown()的调用时机，确保所有提交的任务（包括子任务）执行完毕后再关闭线程池，从而避免数据不一致等问题。

在使用Java的ExecutorService时，一个常见的困扰是如何确保所有提交的任务，包括那些由其他任务提交的子任务，都执行完毕后再关闭线程池。如果在主线程中调用shutdown()过早，可能会导致部分任务尚未完成就被中断，从而导致数据不一致或其他问题。本文将介绍几种确保所有任务完成再关闭线程池的方法。

问题分析：shutdown()的调用时机

问题的核心在于shutdown()方法的调用时机。shutdown()方法会阻止线程池接受新的任务，但它并不会立即停止正在执行的任务。它会等待所有已提交的任务执行完毕后才关闭线程池。如果在所有任务（包括子任务）都提交到线程池之前就调用了shutdown()，那么后续提交的任务可能无法执行，或者正在执行的任务可能被中断。

解决方案：控制shutdown()的调用时机

解决这个问题的关键在于确保在所有任务都提交到线程池之后再调用shutdown()。具体来说，需要根据任务的提交方式采取不同的策略。

1. 主线程提交任务，任务中包含子任务

如果主线程直接向线程池提交任务，并且这些任务又会向同一个线程池提交子任务，那么可以采用以下方法：

• 使用Future.get()等待任务完成： 如果主线程提交的任务返回一个Future对象，可以使用Future.get()方法阻塞主线程，直到该任务执行完毕。这样可以确保所有由该任务提交的子任务都已加入线程池。

  ExecutorService execService = Executors.newFixedThreadPool(10);
  Future t0fut = execService.submit(t0); // t0 是一个 Runnable 或 Callable 对象
  
  // 处理其他任务 t1, t2, t3...
  // ...
  
  try {
      t0fut.get(); // 阻塞直到 t0 完成
  } catch (InterruptedException | ExecutionException e) {
      // 处理异常
      e.printStackTrace();
  }
  
  execService.shutdown();
  try {
      if (!execService.awaitTermination(60, TimeUnit.MINUTES)) {
          System.err.println("线程池关闭超时!");
      }
  } catch (InterruptedException ie) {
      // (Re-)Cancel if current thread also interrupted
      execService.shutdownNow();
      // Preserve interrupt status
      Thread.currentThread().interrupt();
  }
  
  // 写入文件等操作
  writeFile();
  closeFile();

  在这个例子中，t0fut.get()会阻塞主线程，直到t0任务执行完毕。由于t0任务执行完毕意味着其提交的所有子任务（如t4和t5）都已经加入线程池，因此可以安全地调用execService.shutdown()。

• 注意事项：

  • Future.get()方法会抛出InterruptedException和ExecutionException，需要进行适当的异常处理。
  • 确保t0任务内部正确处理异常，避免异常导致子任务无法提交。

2. 其他线程提交任务，任务中包含子任务

如果t0本身就是一个线程（而不是通过submit提交的任务），并且它会向线程池提交任务，那么可以使用Thread.join()方法等待该线程结束。

ExecutorService execService = Executors.newFixedThreadPool(10);
Thread t0 = new Thread(runnableT0); // runnableT0 是一个 Runnable 对象

t0.start();

// 处理其他任务 t1, t2, t3...
// ...

try {
    t0.join(); // 阻塞直到 t0 线程结束
} catch (InterruptedException e) {
    // 处理异常
    e.printStackTrace();
}

execService.shutdown();
try {
    if (!execService.awaitTermination(60, TimeUnit.MINUTES)) {
        System.err.println("线程池关闭超时!");
    }
} catch (InterruptedException ie) {
    // (Re-)Cancel if current thread also interrupted
    execService.shutdownNow();
    // Preserve interrupt status
    Thread.currentThread().interrupt();
}

// 写入文件等操作
writeFile();
closeFile();

在这个例子中，t0.join()会阻塞主线程，直到t0线程执行完毕。由于t0线程执行完毕意味着其提交的所有子任务都已经加入线程池，因此可以安全地调用execService.shutdown()。

• 注意事项：
  • Thread.join()方法会抛出InterruptedException，需要进行适当的异常处理。
  • 确保t0线程内部正确处理异常，避免异常导致子任务无法提交。

总结

正确关闭ExecutorService线程池并确保所有任务完成的关键在于控制shutdown()的调用时机。根据任务提交的方式，可以使用Future.get()或Thread.join()方法等待所有任务（包括子任务）都提交到线程池之后再调用shutdown()。此外，还需要注意异常处理，确保任务能够正常执行并提交子任务。最后，在调用shutdown()之后，务必使用awaitTermination()方法等待线程池关闭，并处理可能的超时情况。