详解CountDownlatch

引言

CountDownLatch 是 Java 并发包 java.util.concurrent 中的一个同步工具类，由著名的并发编程大师 Doug Lea 所开发。它允许一个或多个线程等待其他线程完成操作后再继续执行，其核心思想是通过一个计数器，对计数器进行调度来实现

原理

CountDownLatch 初始化时会设置一个计数器的值，该值表示需要等待完成的操作数量。每当一个线程完成了自己的任务后，会调用 countDown() 方法将计数器减 1。当计数器的值变为 0 时，意味着所有需要等待的操作都已完成，此时在 await() 方法上等待的线程将被唤醒，继续执行后续操作。

源码分析

内部类

private final Sync sync;private static final class Sync extends AbstractQueuedSynchronizer {private static final long serialVersionUID = 4982264981922014374L;Sync(int count) {setState(count);}int getCount() {return getState();}protected int tryAcquireShared(int acquires) {return (getState() == 0) ? 1 : -1;}protected boolean tryReleaseShared(int releases) {// Decrement count; signal when transition to zerofor (;;) {int c = getState();if (c == 0)return false;int nextc = c-1;if (compareAndSetState(c, nextc))return nextc == 0;}}
}

• sync：CountDownLatch 内部使用 Sync 类的实例，Sync 继自 AbstractQueuedSynchronizer（AQS），借助 AQS 的状态（state）来表示计数器的值。
• Sync 类：
  • 构造函数 Sync(int count)：在创建 Sync 实例时，会调用 setState(count) 方法将 AQS 的状态初始化为传入的计数器值。
  • getCount() 方法：返回 AQS 的当前状态，即计数器的当前值。
  • tryAcquireShared(int acquires) 方法：用于尝试以共享模式获取锁。当计数器值为 0 时返回 1，表示可以获取锁；否则返回 -1，表示无法获取锁。
  • tryReleaseShared(int releases) 方法：用于尝试以共享模式释放锁。使用 for(;;) 无限循环结合 compareAndSetState 方法（CAS 操作）来安全地将计数器减 1。当计数器减为 0 时返回 true，表示释放成功且可以唤醒等待线程；否则返回 false。

结合源码我们一般是主线程设计为等待线程，线程池去完成计数器的增减

内部方法

await()

public void await() throws InterruptedException {sync.acquireSharedInterruptibly(1);
}

• await() 方法会调用 sync 的 acquireSharedInterruptibly(1) 方法，该方法是 AQS 提供的以共享模式获取锁的可中断方法。其具体逻辑为：

  • 检查当前线程是否被中断，若被中断则抛出 InterruptedException 异常。

  • 调用 tryAcquireShared 方法尝试获取锁，若计数器为 0 则获取成功，线程继续执行；否则将当前线程放入等待队列并阻塞。

await(long timeout, TimeUnit unit) 

public boolean await(long timeout, TimeUnit unit)throws InterruptedException {return sync.tryAcquireSharedNanos(1, unit.toNanos(timeout));
}

该方法允许线程等待指定的时间。调用 sync 的 tryAcquireSharedNanos(1, unit.toNanos(timeout)) 方法，在指定时间内尝试以共享模式获取锁。若在超时时间内计数器变为 0 则返回 true；若超时则返回 false。

countDown() 

countDown() 方法调用 sync 的 releaseShared(1) 方法，以共享模式释放锁。该方法会调用 tryReleaseShared 方法尝试将计数器减 1，若减为 0 则唤醒所有在等待队列中的线程。

实战

下面是多线程实战CountDownlatch，依旧是主线程等待，在循环中创建线程调用await方法

import java.util.concurrent.CountDownLatch;public class CountDownLatchExample {public static void main(String[] args) throws InterruptedException {// 创建一个 CountDownLatch 实例，计数器初始值为 3CountDownLatch latch = new CountDownLatch(3);// 创建并启动三个工作线程for (int i = 0; i < 3; i++) {final int workerId = i;new Thread(() -> {try {System.out.println("Worker " + workerId + " is working.");// 模拟工作Thread.sleep((long) (Math.random() * 1000));System.out.println("Worker " + workerId + " has finished.");} catch (InterruptedException e) {e.printStackTrace();} finally {// 工作完成，计数器减 1latch.countDown();}}).start();}// 主线程等待所有工作线程完成System.out.println("Main thread is waiting for workers to finish.");latch.await();System.out.println("All workers have finished. Main thread can continue.");}
}