Java并发工具：理解和使用CountDownLatch

原创灰太狼 2024-12-05 03:57 14阅读 0赞

`CountDownLatch`是Java并发库中的一个重要工具，它主要用于协调多个线程的执行。

**基本概念：**

- `CountDownLatch`有一个初始计数器值。
- 当一个或多个线程调用`await()`方法时，它们会将控制权释放给当前的调用者（通常是父线程）。
- 父线程在等待子线程完成任务后，可以通过调用`countDown()`方法来减少计数器的值。当计数器值归零时，所有等待的线程都会被唤醒。

**使用示例：**

```java
import java.util.concurrent.CountDownLatch;

public class CountDownLatchExample {

private CountDownLatch countDownLatch;
    private int initialCount = 5; // 初始化计数器为5

public CountDownLatchExample() {
        this.countDownLatch = new CountDownLatch(initialCount);
    }

public void task(int id) throws InterruptedException {
        System.out.println("Task " + id + " starting...");

countDownLatch.await(id); // 等待计数器到达当前任务的id值

if (id == 1 && initialCount > 0)) { // 如果是第一个任务并且计数器大于0
            System.out.println("Task " + id + " completed. Decreasing counter...");
            countDownLatch.countDown(); // 减少计数器
        } else {
            System.out.println("Task " + id + " completed. No counter decrease necessary.");
        }

System.out.println("Task " + id + " finished.");
    }

public void main(String[] args) throws InterruptedException {
        CountDownLatchExample example = new CountDownLatchExample();

// 创建多个线程执行任务
        for (int i = 1; i <= 5; i++) {
            Thread thread = new Thread(example::task, "Task " + i));
            thread.start();
        }

// 等待所有子线程完成任务，计数器归零
        countDownLatch.await();

System.out.println("All tasks completed. CountDownLatch reached zero.");
    }
}
```

在这个例子中，我们创建了一个`CountDownLatchExample`类，它有一个`countDownLatch`变量用于管理计数器。同时，我们还定义了一个`task`方法，该方法会在子线程中执行，并且根据任务的id值调用`await()`方法。

在`main`方法中，我们创建了5个任务线程，每个线程执行`task`方法。然后，我们等待所有子线程完成任务后，计数器归零。