细说CountDownLatch

CountDownLatch是Java中提供的一个同步辅助类，它允许一个或多个线程等待其他线程完成操作。在面试中，面试官经常会询问候选人是否在实际项目中使用过CountDownLatch，以评估其对多线程编程和并发控制的理解和经验。本文将详细介绍CountDownLatch的作用、工作原理及其在实际项目中的应用。
什么是 CountDownLatch？
CountDownLatch是一个同步工具类，允许一个或多个线程一直等待，直到其他线程执行完毕后再继续执行。它通过一个计数器实现，这个计数器初始化为一个给定的数量。每当一个线程完成了它的任务后，这个计数器的值就会减一。当计数器的值达到零时，等待的线程会被唤醒，继续执行后续任务。
CountDownLatch 的工作原理
CountDownLatch的工作原理可以通过以下几个步骤来概括：
初始化：创建 CountDownLatch 实例时，设置一个初始计数值。
等待：一个或多个线程调用await()方法进入等待状态，直到计数器减到零。
计数减一：完成任务的线程调用countDown()方法，计数器减一。
唤醒：当计数器减到零时，所有调用await()方法等待的线程被唤醒，继续执行。

实战中 CountDownLatch 的应用场景
以下是几个实际项目中使用CountDownLatch的典型场景：
场景一：并行任务处理
在实际开发中，有时需要将一个大任务分解成若干个小任务并行处理，待所有小任务完成后，再进行后续处理。例如，在文件处理、数据处理、网络请求等场景中，可以将大文件分割成多个小文件并行处理，或同时向多个服务器发起请求，待所有任务完成后再合并结果。

public class ParallelTaskExample {public static void main(String[] args) throws InterruptedException {int numberOfTasks = 5;CountDownLatch latch = new CountDownLatch(numberOfTasks);for (int i = 0; i < numberOfTasks; i++) {new Thread(new Worker(latch)).start();}latch.await(); // 等待所有任务完成System.out.println("所有任务已完成，继续处理后续操作。");}
}class Worker implements Runnable {private final CountDownLatch latch;Worker(CountDownLatch latch) {this.latch = latch;}@Overridepublic void run() {try {// 执行任务System.out.println(Thread.currentThread().getName() + " 执行任务");} finally {latch.countDown();}}
}

场景二：服务启动检查
在分布式系统中，系统启动时需要依赖多个服务。使用CountDownLatch可以确保所有依赖服务都启动完成后，主线程才继续执行，保证系统的稳定性。

import java.util.concurrent.CountDownLatch;public class ServiceStartupCheck {public static void main(String[] args) throws InterruptedException {int numberOfServices = 3;CountDownLatch latch = new CountDownLatch(numberOfServices);for (int i = 0; i < numberOfServices; i++) {new Thread(new Service(latch, "Service-" + (i + 1))).start();}latch.await(); // 等待所有服务启动System.out.println("所有服务已启动，系统启动完成。");}
}class Service implements Runnable {private final CountDownLatch latch;private final String serviceName;Service(CountDownLatch latch, String serviceName) {this.latch = latch;this.serviceName = serviceName;}@Overridepublic void run() {try {// 模拟服务启动System.out.println(serviceName + " 正在启动...");Thread.sleep((long) (Math.random() * 3000)); // 模拟不同启动时间System.out.println(serviceName + " 启动完成。");} catch (InterruptedException e) {e.printStackTrace();} finally {latch.countDown();}}
}

场景三：并行计算结果汇总
在一些计算密集型任务中，可以将计算任务分解到多个线程中并行处理，待所有线程完成计算后，再汇总结果。例如，在大数据处理和统计分析中，可以使用CountDownLatch来同步各个计算线程，确保计算结果的准确性和一致性。

public class ParallelComputation {public static void main(String[] args) throws InterruptedException {int numberOfThreads = 4;CountDownLatch latch = new CountDownLatch(numberOfThreads);int[] results = new int[numberOfThreads];for (int i = 0; i < numberOfThreads; i++) {final int index = i;new Thread(() -> {try {// 模拟计算任务results[index] = (int) (Math.random() * 100);System.out.println("线程 " + Thread.currentThread().getName() + " 计算结果: " + results[index]);} finally {latch.countDown();}}).start();}latch.await(); // 等待所有计算完成int totalResult = 0;for (int result : results) {totalResult += result;}System.out.println("所有线程计算结果汇总: " + totalResult);}
}

使用CountDownLatch的注意事项
1、正确使用await()和countDown()：避免死锁和计数错误。确保countDown()在finally块中调用，以防止线程在任务失败时无法减少计数器。
2、合理设置计数值：计数值应与任务数量相匹配，防止出现无法唤醒等待线程的问题。
3、避免重复使用：CountDownLatch是一次性的，计数器归零后无法重置。如果需要重复使用，可以考虑使用CyclicBarrier或其他同步工具。
总结
CountDownLatch是一个强大的并发工具，适用于各种需要线程同步的场景。通过本文的介绍，相信你对CountDownLatch的工作原理和实际应用有了更深入的理解。在面试中，当被问及是否使用过CountDownLatch 时，可以结合上述示例和自己的项目经验，详细阐述其应用场景和使用方法，以展示自己在并发编程方面的能力和经验。