目录
一、CountDownLatch简介
CountDownLatch 是 Java 并发包中的一个同步工具类,位于 java.util.concurrent 包中。允许一个或多个线程等待直到一组操作完成。 CountDownLatch 的核心思想是通过一个计数器来控制线程的执行。 CountDownLatch 维护一个计数器,初始值为一个正整数。当计数器的值减为零时,表示所有线程的操作已完成,等待的线程将被唤醒。
CountDownLatch 的主要方法包括:
await() : 使当前线程等待,直到计数器的值为零。
countDown():将计数器的值减一。当计数器的值减为零时,所有等待的线程将被唤醒。
二、CountDownLatch使用场景
1. 并行任务的协调
当有多个线程执行任务,并希望在所有线程完成后再继续执行主线程时,可以使用 CountDownLatch 。
2. 完成初始化
在某些情况下,主线程需要等待多个初始化任务完成后再继续执行,比如在启动一个服务时等待所有依赖的组件初始化完毕。
3. 控制线程的执行顺序
可以使用 CountDownLatch 来控制线程的执行顺序。
三、CountDownLatch用法
下面使用CountDownLatch实现如何在多个线程完成任务后再继续执行主线程:
import java.util.concurrent.CountDownLatch;
public class CountDownLatchExample {
private static final int THREAD_COUNT = 3;
public static void main(String[] args) {
CountDownLatch latch = new CountDownLatch(THREAD_COUNT);
// 创建并启动多个线程
for (int i = 0; i < THREAD_COUNT; i++) {
final int threadIndex = i;
new Thread(() -> {
try {
// 模拟任务执行
System.out.println("线程 " + threadIndex + " 正在工作...");
Thread.sleep((long) (Math.random() * 1000)); // 随机睡眠
System.out.println("线程 " + threadIndex + " 已经完成.");
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
// 任务完成,计数器减一
latch.countDown();
}
}).start();
}
try {
// 主线程等待,直到计数器减为零
latch.await();
System.out.println("所有线程都完成,主线程开始启动.");
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
结果如下:
如上,在循环中创建多个线程,每个线程模拟一个任务的执行,每个线程在完成任务后调用 latch.countDown() ,将计数器减一,主线程调用 latch.await() ,等待计数器减到零,表示所有线程都已完成任务。