细说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;
    }

    @Override
    public 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;
    }

    @Override
    public 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 时，可以结合上述示例和自己的项目经验，详细阐述其应用场景和使用方法，以展示自己在并发编程方面的能力和经验。