CountDownLatch 使用详情

news2024/10/7 12:28:38

在这里插入图片描述
CountDownLatch 是 Java.util.concurrent 包下的一个类，它可以用来实现一个或多个线程等待其他线程完成后再继续执行的场景。

CountDownLatch 类中有一个计数器，每次调用 countDown() 方法计数器的值减1，当计数器的值变为0时，调用 await() 方法的线程就会被唤醒，继续执行。

CountDownLatch 的使用一般包括以下几个步骤：

创建一个 CountDownLatch 对象，并指定计数器的初始值。
在需要等待的线程中调用 await() 方法，进入等待状态。
在其他线程中完成任务后，调用 countDown() 方法，计数器的值减1。
当计数器的值变为0时，被等待的线程会被唤醒，继续执行。

下面是一个使用 CountDownLatch 的简单案例：

import java.util.concurrent.CountDownLatch;

public class CountDownLatchExample {
    public static void main(String[] args) throws InterruptedException {
        // 创建一个 CountDownLatch 对象，初始计数器的值为3
        CountDownLatch latch = new CountDownLatch(3);

        // 创建三个线程，每个线程执行任务后调用 countDown() 方法
        Thread thread1 = new Thread(new Task(latch, "Task 1"));
        Thread thread2 = new Thread(new Task(latch, "Task 2"));
        Thread thread3 = new Thread(new Task(latch, "Task 3"));

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

        // 主线程调用 await() 方法等待计数器变为0
        latch.await();

        // 当计数器变为0时，主线程继续执行
        System.out.println("All tasks have been completed.");
    }
}

class Task implements Runnable {
    private CountDownLatch latch;
    private String name;

    public Task(CountDownLatch latch, String name) {
        this.latch = latch;
        this.name = name;
    }

    @Override
    public void run() {
        System.out.println(name + " is running.");

        // 模拟任务执行时间
        try {
            Thread.sleep(2000);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        System.out.println(name + " is completed.");

        // 任务完成后调用 countDown() 方法
        latch.countDown();
    }
}

在上面的例子中，我们创建了一个 CountDownLatch 对象，计数器的初始值为3。然后创建了三个线程，每个线程执行任务后调用 countDown() 方法。在主线程中调用 await() 方法等待计数器的值变为0，当计数器的值变为0时，主线程继续执行并输出 “All tasks have been completed.”。每个线程的任务模拟了一个耗时的操作，当任务完成后调用 countDown() 方法，计数器的值减1。当计数器的值减到0时，主线程被唤醒，继续执行。

CountDownLatch 的使用场景包括：