Java并发工具库：Semaphore和CountDownLatch实操示例

原创喜欢ヅ旅行 2024-12-03 09:12 10阅读 0赞

在Java的并发工具库中，`Semaphore` 和 `CountDownLatch` 是两种常用的同步工具。下面我将为你提供这两种工具的使用示例。

1. Semaphore（信号量）：
信号量是一种资源管理机制，用于控制多个线程对同一资源的访问。

示例1：允许一定数量的线程同时执行任务。
```java
Semaphore semaphore = new Semaphore(3); // 允许的最大线程数

// 线程任务
Thread task1 = new Thread(() -> {
    try {
        semaphore.acquire(); // 获得信号量，如果达到最大值会阻塞
        // 执行任务
        System.out.println("Task 1 is running.");
    } catch (InterruptedException e) {
        e.printStackTrace();
    } finally {
        semaphore.release(); // 释放信号量，允许其他线程获取
    }
}));
task1.start();

// 线程任务2-3，依此类推
Thread task2 = new Thread(() -> {
    try {
        semaphore.acquire(); // 获得信号量
        // 执行任务
        System.out.println("Task 2 is running.");
    } catch (InterruptedException e) {
        e.printStackTrace();
    } finally {
        semaphore.release(); // 释放信号量
    }
});
task2.start();

// ...
```
示例2：当所有线程完成任务后，计数信号量会自动释放。
```java
Semaphore semaphore = new Semaphore(0); // 初始化为0，表示计数器未开启

Thread task1 = new Thread(() -> {
    try {
        semaphore.acquire(); // 获得信号量，打开计数器
        // 执行任务
        System.out.println("Task 1 is running.");
    } catch (InterruptedException e) {
        e.printStackTrace();
    } finally {
        semaphore.release(); // 释放信号量，关闭计数器
        System.out.println("Semaphore released, counting finished.");
    }
}));
task1.start();

// ...
```
2. CountDownLatch（倒计时）：
CountDownLatch也是一种同步工具，它用于管理一个或多个线程需要等待的事件。

示例：当所有线程执行完任务后，会调用CountDownLatch的一个方法，将计数器减一，直到计数器为零，表示所有线程已完成任务。

```java
// 初始化CountDownLatch，设置初始值10（表示有10个任务需要完成）。
CountDownLatch countDownLatch = new CountDownLatch(10);

// 线程任务
Thread task1 = new Thread(() -> {
    try {
        // 当前线程执行完任务后，调用countDownLatch的down方法，将计数器减一。
        countDownLatch.countDown();
        System.out.println("Task 1 is done, decrementing the counter.");
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
});
task1.start();

// ...
```
以上就是`Semaphore` 和 `CountDownLatch` 在Java并发工具库中的基本使用示例。