CountDownLatch通过计数器实现线程同步,初始化时设定计数值,调用await()的线程阻塞等待,其他线程执行countDown()使计数减一,计数归零后等待线程被唤醒,适用于主线程等待多个任务完成的场景。
CountDownLatch 是 Java 并发包(java.util.concurrent)中一个实用的同步工具,用于让一个或
多个线程等待其他线程完成一组操作后再继续执行。它的核心机制基于一个计数器,这个计数器在初始化时设定,每当一个任务完成就减 1,直到计数器归零,等待的线程才被唤醒。
CountDownLatch 内部维护一个 volatile 类型的整数计数器,使用 AQS(AbstractQueuedSynchronizer)实现线程的阻塞与唤醒:
适用于“等待N个任务完成”的场景,比如:
以下是一个简单的例子,展示主线程等待 3 个工作线程完成任务:
public class CountDownLatchExample {
public static void main(String[] args) throws InterruptedException {
CountDownLatch latch = new CountDownLatch(3);
for (int i = 1; i <= 3; i++) {
new Thread(() -> {
System.out.println(Thread.currentThread().getName() + " 开始工作");
try { Thread.sleep(1000); } catch (InterruptedException e) {}
System.out.println(Thread.currentThread().getName() + " 完成");
latch.countDown(); // 计数减一
}, "Worker-" + i).start();
}
System.out.println("主线程等待所有工作线程完成...");
latch.await(); // 阻塞直到计数为0
System.out.println("所有任务完成,主线程继续执行");
}
}
理解以下几个要点有助于正确使用 CountDownLatch: