本文旨在解决 Java ExecutorService 线程池在使用 shutdown() 和 awaitTermination() 方法时，可能遇到的线程未完成任务就被强制关闭的问题。我们将探讨如何确保线程池在所有任务执行完毕后再进行关闭，从而避免数据丢失或程序异常。
在使用 Java 的 ExecutorService 线程池时，一个常见的困惑是如何正确地关闭线程池，以确保所有提交的任务都已完成。特别是在任务之间存在依赖关系，或者任务会动态提交新的任务到线程池的情况下，这个问题变得更加复杂。 简单地调用 shutdown() 方法并不能保证所有任务都已完成，因此我们需要更细致地控制线程池的关闭流程。
理解 shutdown() 和 awaitTermination() 的作用
shutdown() 方法会阻止线程池接受新的任务，但不会中断正在执行的任务。它会将线程池的状态设置为“停止接受新任务”，并允许已提交的任务继续执行。
awaitTermination(long timeout, TimeUnit unit) 方法会阻塞当前线程，直到线程池中的所有任务都已完成执行，或者超时时间已过。如果所有任务在超时时间内完成，则返回 true；否则返回 false。
正确关闭线程池的策略
关键在于确保在调用 shutdown() 之前，所有任务都已经提交到线程池。如果任务 A 可能会提交新的任务 B 到同一个线程池，那么必须等待任务 A 完成，才能调用 shutdown()。
以下是两种常见的场景以及对应的解决方案：
场景一：任务 A 通过 submit() 方法提交到线程池
如果任务 A 是通过 submit() 方法提交到线程池的，可以使用 Future 对象来跟踪任务 A 的完成状态。
ExecutorService executorService = Executors.newFixedThreadPool(10);
Future> futureA = executorService.submit(taskA);
// 处理其他任务 t1, t2, t3...
try {
futureA.get(); // 阻塞直到任务 A 完成
} catch (InterruptedException | ExecutionException e) {
// 处理异常
e.printStackTrace();
}
executorService.shutdown();
try {
if (!executorService.awaitTermination(60, TimeUnit.MINUTES)) {
System.err.println("线程池关闭超时");
}
} catch (InterruptedException e) {
e.printStackTrace();
}
在这个例子中，futureA.get() 方法会阻塞当前线程，直到任务 A 完成。只有在任务 A 完成后，才会调用 executorService.shutdown() 方法，从而确保任务 A 提交的所有子任务（例如任务 B）都已经提交到线程池。
场景二：任务 A 是一个线程，它会提交任务到线程池
如果任务 A 本身就是一个线程，它会向线程池提交新的任务，那么需要使用 Thread.join() 方法等待任务 A 线程结束。
ExecutorService executorService = Executors.newFixedThreadPool(10);
Thread threadA = new Thread(taskA);
threadA.start();
// 处理其他任务 t1, t2, t3...
try {
threadA.join(); // 阻塞直到线程 A 完成
} catch (InterruptedException e) {
// 处理异常
e.printStackTrace();
}
executorService.shutdown();
try {
if (!executorService.awaitTermination(60, TimeUnit.MINUTES)) {
System.err.println("线程池关闭超时");
}
} catch (InterruptedException e) {
e.printStackTrace();
}
在这个例子中，threadA.join() 方法会阻塞当前线程，直到线程 A 结束。只有在线程 A 结束后，才会调用 executorService.shutdown() 方法，从而确保线程 A 提交的所有子任务都已经提交到线程池。
注意事项
- 异常处理： 在使用 futureA.get() 和 threadA.join() 时，需要捕获 InterruptedException 和 ExecutionException 异常，并进行适当的处理。
- 超时设置： awaitTermination() 方法的超时时间需要根据实际情况进行调整，以确保线程池有足够的时间完成所有任务。
- 资源释放： 在线程池关闭后，应该释放相关的资源，例如关闭文件流、数据库连接等。
- 拒绝策略： 在调用 shutdown() 之后，如果再向线程池提交新的任务，线程池会根据配置的拒绝策略来处理这些任务。 默认情况下，会抛出 RejectedExecutionException 异常。 可以通过 setRejectedExecutionHandler() 方法来设置自定义的拒绝策略。
总结
正确关闭 ExecutorService 线程池需要确保在调用 shutdown() 之前，所有任务都已经提交到线程池。 对于通过 submit() 方法提交的任务，可以使用 Future.get() 方法来等待任务完成。 对于线程本身提交任务的情况，可以使用 Thread.join() 方法来等待线程结束。 合理的异常处理、超时设置和资源释放也是确保线程池正确关闭的重要环节。
Java免费学习笔记：立即使用
解锁 Java 大师之旅：从入门到精通的终极指南
