CountDownLatch是Java中用于线程同步的工具,通过计数器控制主线程等待多个子线程完成任务后再执行。初始化时设定计数值,每调用一次countDown()计数减一,await()使主线程阻塞直至计数为零。适用于主线程等待多任务并发完成的场景,如示例中三个任务并行执行,主线程在latch.await()处等待,所有任务调用latch.countDown()后主线程继续执行汇总操作。需注意countDown()应在finally块中调用以防止异常导致计数不全,且CountDownLatch不可重置,需重复使用时应选择CyclicBarrier。
在Java多线程编程中,CountDownLatch 是一个非常实用的同步工具类,位于 java.util.concurrent 包下。它允许一个或多个线程等待其他线程完成操作后再继续执行。这种“倒计数门闩”机制非常适合用于协调多个线程之间的执行顺序。
CountDownLatch 内部维护一个计数器,初始化时指定需要等待的事件数量。每当一个事件完成,调用 countDown() 方法将计数减一。其他线程通过调用 await() 方法阻塞自己,直到计数器归零,此时所有等待线程被唤醒并继续执行。
关键点:
假设我们需要启动多个线程并行处理数据,主线程需等待所有子任务完成后才进行汇总操作。这时 CountDownLatch 非常合适。
示例代码:
import java.util.concurrent.CountDownLatch; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors;public class TaskProcessor { public static void main(String[] args) throws InterruptedException { int taskCount = 3; CountDownLatch latch = new CountDownLatch(taskCount); ExecutorService executor = Executors.newFixedThreadPool(3);
for (int i = 1; i <= taskCount; i++) { final int taskId = i; executor.submit(() -> { try { System.out.println("任务 " + taskId + " 开始执行"); Thread.sleep(1000 * taskId); // 模拟不同耗时 System.out.println("任务 " + taskId + " 完成"); } catch (InterruptedException e) { Thread.currentThread().interrupt(); } finally { latch.countDown(); // 任务完成,计数减一 } }); } System.out.println("主线程等待所有任务完成..."); latch.await(); // 阻塞直到计数为0 System.out.println("所有任务已完成,开始汇总结果"); executor.shutdown(); }}
输出大致如下:
任务 1 开始执行虽然 CountDownLatch 使用简单,但实际开发中仍有一些细节需要注意:
基本上就这些。CountDownLatch 是实现线程协作的一种简洁方式,尤其适合“一个或多个线程等待一组操作完成”的场景。掌握其原理和使用技巧,能有效提升并发程序的可控性和可读性。