当前位置：首页 > news >正文

彻底理解CountDownLatch

news 2026/3/30 15:22:18

CountDownLatch是 Java 并发包（java.util.concurrent）中一个非常经典且实用的同步工具类，由 Doug Lea 设计。它的核心思想是：

让一个或多个线程等待，直到其他线程完成一组操作（“倒计时归零”）后再继续执行。

下面从设计意图、内部实现、使用场景、内存语义四个维度帮你彻底理解它。

一、设计意图：解决什么问题？

场景1：主线程等所有子任务完成

启动 N 个 worker 线程去处理任务。
主线程不能继续，必须等这 N 个都做完。
✅CountDownLatch初始化为 N，每个 worker 完成后调用countDown()，主线程调用await()等待。

场景2：所有线程同时启动（“发令枪”）

所有 worker 线程先启动，但卡在await()。
主线程准备就绪后调用countDown()（初始 count=1），所有 worker 同时开始。
✅ 这就是“门闩”（latch）或“栅栏”的作用。

🔑关键特性：
一次性（one-shot）：计数器只能从 N 减到 0，不能重置。
如果你需要重复使用（比如循环等待），请用CyclicBarrier。

二、内部实现：基于 AQS 的共享模式

CountDownLatch的核心是内部类Sync，它继承自AbstractQueuedSynchronizer（AQS）。

1. 状态表示

setState(count);// AQS 的 state 字段 = 倒计数值

state > 0：还有任务未完成，调用await()的线程要阻塞。
state == 0：所有任务完成，所有等待线程被释放。

2. 获取（await）—— 共享模式

protectedinttryAcquireShared(intacquires){return(getState()==0)?1:-1;}

返回1：表示获取成功（可以通行）。
返回-1：表示获取失败（需要排队等待）。
注意：只要state == 0，所有调用await()的线程都能立即通过→ 这就是共享模式的体现！

3. 释放（countDown）—— 共享释放

protectedbooleantryReleaseShared(intreleases){for(;;){intc=getState();if(c==0)returnfalse;// 已经归零，无需再释放intnextc=c-1;if(compareAndSetState(c,nextc))returnnextc==0;// 只有归零时才触发唤醒}}

使用 CAS 循环安全地减 1。
只有当nextc == 0时，才返回true，通知 AQS：“现在可以唤醒所有等待者了！”

4. 唤醒机制

当countDown()使计数归零时，AQS 会调用doReleaseShared()（你之前问过的那个方法）。
因为是共享模式，所有等待线程都会被连续唤醒（传播机制确保不漏掉）。

✅ 所以：CountDownLatch是 AQS共享模式的经典应用。

三、典型使用方式（结合你的代码注释）

示例1：发令枪 + 完成信号

CountDownLatchstartSignal=newCountDownLatch(1);// 所有工人等“开始”信号CountDownLatchdoneSignal=newCountDownLatch(N);// 主线程等“完成”信号// 启动 N 个工人for(inti=0;i<N;i++)newThread(()->{startSignal.await();// 阻塞，直到 startSignal.countDown()doWork();doneSignal.countDown();// 完成一个，计数减1}).start();doSomethingElse();// 主线程做准备工作startSignal.countDown();// 发令：所有工人开始！doneSignal.await();// 等待所有工人完成

示例2：并行任务汇总

ExecutorServiceexec=Executors.newFixedThreadPool(4);CountDownLatchlatch=newCountDownLatch(10);for(inti=0;i<10;i++){exec.submit(()->{try{process(i);}finally{latch.countDown();// 必须保证执行！}});}latch.await();// 等待全部10个任务完成exec.shutdown();

⚠️ 注意：countDown()通常放在finally块中，防止异常导致计数不减，造成死锁。

四、内存可见性（Memory Consistency）

Java 内存模型（JMM）保证：

在某个线程调用countDown()之前的所有操作，对在await()成功返回后的线程是可见的。

即：happens-before 关系成立。

这意味着：

Worker 线程写入的数据（如计算结果到共享变量），
在主线程await()返回后，一定能读到最新值。

✅ 无需额外加volatile或synchronized！

五、与 CyclicBarrier 的区别

特性	`CountDownLatch`	`CyclicBarrier`
是否可重用	❌ 一次性	✅ 可重复使用
触发条件	计数归零	所有线程到达屏障点
谁等待	任意线程（通常是协调者）	所有参与线程互相等待
典型用途	等待 N 个任务完成 / 发令枪	多线程迭代同步（如并行计算每轮同步）