如何在Java中使用CountDownLatch

CountDownLatch通过计数器实现线程等待,初始化后调用countDown()减一,await()阻塞直至计数为0,适用于主线程等待多个任务完成的场景。

CountDownLatch 是 Java 并发编程中一个实用的同步工具,它允许一个或多个线程等待其他线程完成操作后再继续执行。它的核心思想是通过一个计数器来控制等待逻辑:计数器初始化为某个正整数,每当一个线程完成任务就调用 countDown() 方法将计数器减一,而等待的线程调用 await() 方法阻塞自己,直到计数器变为 0。

基本使用场景

假设你有一个主线程需要等待多个工作线程全部完成后再继续处理结果,CountDownLatch 就非常适合这个场景。

例如:

  • 启动多个服务组件,主程序等待所有组件初始化完成。
  • 并行执行多个子任务,汇总结果前必须等所有任务结束。
  • 测试多线程逻辑时,确保所有线程都执行完毕再验证结果。

代码示例:主线程等待多个线程完成

下面是一个简单的例子,展示如何使用 CountDownLatch:

import java.util.concurrent.CountDownLatch;

public class CountDownLatchExample {
    public static void main(String[] args) {
        int threadCount = 3;
        CountDownLatch latch = new CountDownLatch(threadCount);

        for (int i = 1; i <= threadCount; i++) {
            new Thread(() -> {
                try {
                    System.out.println(Thread.currentThread().getName() + " 正在执行任务...");
                    Thread.sleep(2000); // 模拟耗时操作
                    System.out.println(Thread.currentThread().getName() + " 任务完成!");
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                } finally {
                    latch.countDown(); // 任务完成,计数减一
                }
            }).start();
        }

        try {
            System.out.println("主线程等待所有子线程完成...");
            latch.await(); // 阻塞,直到计数器为0
            System.out.println("所有线程已完成,主线程继续执行。");
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
}

输出可能如下:

主线程等待所有子线程完成...
Thread-0 正在执行任务...
Thread-1 正在执行任务...
Thread-2 正在执行任务...
Thread-0 任务完成!
Thread-1 任务完成!
Thread-2 任务完成!
所有线程已完成,主线程继续执行。

关键方法说明

  • CountDownLatch(int count):构造函数,传入计数器初始值。
  • await():使当前线程阻塞,直到计数器为0。可被中断。
  • await(long timeout, TimeUnit unit):带超时的等待,避免无限等待。
  • countDown():将计数器减一,可在多个线程中调用。

注意:CountDownLatch 的计数器不能重置,一旦变为0,所有后续的 await() 调用会立即返回。如果需要重复使用的计数器,考虑使用 CyclicBarrier。

常见注意事项

使用时要注意以下几点:

  • 确保 countDown() 在 finally 块中调用,防止异常导致计数未减少。
  • 不要遗漏调用 countDown(),否则会导致 await() 永远无法结束。
  • 适合“等待一组操作完成”的场景,不适用于线程间频繁协作。

基本上就这些。CountDownLatch 简单易用,是控制线程执行顺序的有力工具。掌握它有助于写出更清晰、可靠的并发程序。