Java 线程池执行完再执行主线程的实现
1. 介绍
在Java中，使用线程池可以有效地管理和使用线程。在有些情况下，我们需要确保线程池中的所有任务都完成后，才能执行之后的主线程代码。本文将为你提供清晰的步骤和实现代码，帮助你理解并实现这个功能。
2. 整体流程
在实现“线程池执行完再执行主线程”时，整体流程可以归纳如下：
| 步骤 | 描述 |
|---|---|
| 1 | 创建线程池 |
| 2 | 提交任务到线程池 |
| 3 | 使用ExecutorService的shutdown方法关闭线程池 |
| 4 | 使用awaitTermination等待所有任务完成 |
| 5 | 执行主线程的代码 |
3. 每一步的具体实现
以下是我们在每一步中需要实现的代码。
步骤 1: 创建线程池
在这一部分，我们将使用Java的Executors创建一个固定大小的线程池。
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
// 创建一个固定大小的线程池
ExecutorService executorService = Executors.newFixedThreadPool(3);
Executors.newFixedThreadPool(3)：创建一个有3个线程的线程池。
步骤 2: 提交任务到线程池
在这一步，我们会定义一些需要执行的任务，并提交到线程池中。
import java.util.concurrent.Callable;
// 定义任务
Callable<String> task1 = () -> {
Thread.sleep(1000); // 模拟任务处理时间
return "任务1完成";
};
// 提交任务到线程池
executorService.submit(task1);
Callable<String>：可以返回结果的任务。executorService.submit(task1)：将任务提交到线程池进行执行。
步骤 3: 关闭线程池
在这一步中，我们关闭线程池，告知线程池不再接收新任务。
executorService.shutdown(); // 关闭线程池，禁止提交新任务
shutdown()：关闭线程池的方法，会平稳地终止线程池，不会立即中断正在执行的任务。
步骤 4: 等待所有任务完成
我们在这一步使用awaitTermination阻塞当前线程，直到线程池中的所有任务都完成。
try {
if (!executorService.awaitTermination(5, TimeUnit.SECONDS)) {
executorService.shutdownNow(); // 超时后强制终止
}
} catch (InterruptedException e) {
executorService.shutdownNow(); // 若当前线程被中断，也强制终止
}
awaitTermination(5, TimeUnit.SECONDS)：最多等待5秒，等待所有任务完成。shutdownNow()：强制停止正在执行的任务。
步骤 5: 执行主线程的代码
最后，在确保所有线程池中的任务都已完成后，我们可以安全地执行主线程的代码。
System.out.println("所有任务完成，准备执行主线程代码...");
- 这里是简单的打印，但可以替换为你的实际业务逻辑。
4. 类图
下面是类图的展示，反映了线程池的主要组成部分。
classDiagram
class Main {
+void main(String[] args)
}
class ExecutorService {
+void submit(Callable)
+void shutdown()
+boolean awaitTermination(long, TimeUnit)
}
class Callable {
+T call() throws Exception
}
Main --> ExecutorService
ExecutorService --> Callable
5. 关系图
以下是关系图的展示，用于描述任务与线程的关系。
erDiagram
TASK {
string id
string name
}
THREAD {
int id
string status
}
THREAD ||--o{ TASK : executes
6. 总结
以上就是实现“Java线程池执行完再执行主线程”的完整步骤和代码示例。通过这个讲解，希望你能够理解线程池的基本使用，以及如何控制主线程的执行顺序。如果有任何问题，请随时询问。掌握线程池的使用将为你未来的Java开发带来更高的效率和更好的性能！
