java 线程池主线程等待

原创

mob64ca12eee07b 2024-04-27 04:45:01 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12eee07b的原创作品，请联系作者获取转载授权，否则将追究法律责任

实现“java 线程池主线程等待”教程

一、整体流程

首先，让我们来看一下整个过程的步骤：

步骤	描述
1	创建一个线程池
2	向线程池提交任务
3	等待所有任务执行完成
4	关闭线程池

二、具体步骤

1. 创建一个线程池

首先，我们需要创建一个线程池。Java提供了Executors工具类来创建线程池，我们可以选择其中一种线程池类型来创建。

// 创建一个固定大小的线程池，线程数为5
ExecutorService executor = Executors.newFixedThreadPool(5);

2. 向线程池提交任务

接下来，我们需要向线程池提交任务。我们可以使用submit方法来提交Runnable或Callable类型的任务。

// 提交一个Runnable任务
executor.submit(() -> {
    // 执行任务的代码
});

3. 等待所有任务执行完成

为了保证主线程等待所有任务执行完成，我们需要使用CountDownLatch来进行同步操作。

CountDownLatch latch = new CountDownLatch(5); // 任务数量为5

executor.submit(() -> {
    // 执行任务的代码
    latch.countDown(); // 每个任务执行完成后调用countDown方法
});

latch.await(); // 主线程等待所有任务执行完成

4. 关闭线程池

最后，在所有任务执行完成后，我们需要关闭线程池。

executor.shutdown();

三、状态图

stateDiagram
    [*] --> 创建线程池
    创建线程池 --> 提交任务
    提交任务 --> 等待任务执行完成
    等待任务执行完成 --> 关闭线程池
    关闭线程池 --> [*]

四、类图

classDiagram
    class ExecutorService {
        + submit()
        + shutdown()
    }
    class Executors {
        + newFixedThreadPool()
    }
    class CountDownLatch {
        + countDown()
        + await()
    }