深入解析CountDownLatch原理，轻松掌握其使用过程

2024-02-02 19:25:31

CountDownLatch的原理和工作机制

CountDownLatch是一个计数器闭锁，主要用于等待多个线程完成各自任务后再继续执行。其工作原理非常简单，CountDownLatch首先创建一个计数器，并将其初始化为需要等待的线程数。当一个线程调用await方法时，该线程会阻塞，直到计数器的值变为0。

在等待期间，如果其他线程调用countDown方法，则计数器会减1。当计数器的值变为0时，因await方法阻塞的线程会被唤醒，继续执行。CountDownLatch的一个重要特性是，await方法会阻塞当前线程，直到计数器变为0，而countDown方法则不会阻塞任何线程。

CountDownLatch的用法示例

为了更好地理解CountDownLatch的使用方法，我们来看一个简单的示例：

public class CountDownLatchExample {

    public static void main(String[] args) {
        // 创建一个CountDownLatch对象，并将其初始化为2
        CountDownLatch latch = new CountDownLatch(2);

        // 创建两个线程，每个线程都调用await方法等待CountDownLatch变为0
        Thread thread1 = new Thread(() -> {
            try {
                latch.await();
                System.out.println("Thread 1 completed its task");
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        });

        Thread thread2 = new Thread(() -> {
            try {
                latch.await();
                System.out.println("Thread 2 completed its task");
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        });

        // 启动两个线程
        thread1.start();
        thread2.start();

        // 主线程调用countDown方法，使CountDownLatch变为0
        latch.countDown();
        latch.countDown();

        // 等待两个线程完成任务
        try {
            thread1.join();
            thread2.join();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.out.println("Main thread completed its task");
    }
}