Java并发工具库：CountDownLatch与Semaphore实战案例

原创青旅半醒 2024-11-21 13:18 17阅读 0赞

在Java的并发工具库中，`CountDownLatch`和`Semaphore`是两种常用的同步工具，它们可以帮助我们控制多线程之间的交互。

1. **CountDownLatch**：
   - CountDownLatch最初用于一个任务完成后通知所有等待它的任务。
   - 当调用`await(count)`时，如果count为0表示该任务已经完成，当前线程将被阻塞；否则继续执行。

2. **Semaphore**：
   - Semaphore是一种信号量，它维护了一个允许的最大计数器（通常称为“许可”）。
   - 当调用`acquire(count)`时，如果count小于等于0，表示资源充足，立即获取许可证；否则线程将被阻塞直到获得足够的许可证。

实战案例：

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

public class ConcurrentExample {

// 使用CountDownLatch模拟任务完成通知
    private CountDownLatch countDownLatch = new CountDownLatch(1);

// 使用Semaphore模拟资源限制，如线程数量
    private Semaphore semaphore = new Semaphore(2);

public void runTask() throws InterruptedException {
        // 获取Semaphore许可，表示任务开始执行
        semaphore.acquire();

try {
            // 模拟耗时任务
            Thread.sleep(500);  // 假设耗时500ms

// 任务完成后通知所有等待的CountDownLatch
            countDownLatch.countDown();
        } finally {
            // 任务结束释放Semaphore许可，表示任务已经完成
            semaphore.release();
        }
    }

public static void main(String[] args) throws InterruptedException {
        ConcurrentExample example = new ConcurrentExample();
        Thread taskThread = new Thread(example::runTask);
        taskThread.start();

// 等待所有CountDownLatch通知完成
        countDownLatch.await();

System.out.println("All tasks finished successfully!");
    }
}
```

这段代码创建了一个`ConcurrentExample`类，它包含一个任务（`runTask`）和两个并发工具：`CountDownLatch`用于任务完成后通知，以及`Semaphore`用于模拟资源限制。

在`main`方法中，我们启动了任务线程，并等待所有任务完成。当所有任务完成后，控制台将打印一条消息表示任务已经成功完成。