Java并发工具类CountDownLatch和Semaphore的实战应用

原创港控/mmm° 2024-09-25 17:24 58阅读 0赞

CountDownLatch和Semaphore是Java并发工具类，它们在多线程环境下有着广泛的应用。下面分别介绍这两种工具的实战应用：

1. CountDownLatch（倒计时锁）

CountDownLatch主要用于一个或多个任务完成之后通知其他等待的任务。它的使用场景包括：资源归还、事件触发等。

实战示例：
```java
import java.util.concurrent.CountDownLatch;

public class CounterExample {
    private final CountDownLatch signal = new CountDownLatch(1);

public void task() throws InterruptedException {
        System.out.println("Task started...");
        // 这里模拟一些耗时的操作
        Thread.sleep(500);
        System.out.println("Task completed...");

// 在任务完成之后通知信号量
        signal.countDown();
    }

public static void main(String[] args) {
        CounterExample example = new CounterExample();

// 创建一个线程来执行任务
        Thread taskThread = new Thread(example::task);
        taskThread.start();

try {
            // 等待任务完成之后释放信号量
            example.signal.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

System.out.println("All tasks completed!");
    }
}
```
在这个例子中，我们创建了一个CountDownLatch，然后在任务中使用它来通知等待的任务。

2. Semaphore（信号量）

Semaphore主要用于控制同时访问共享资源的线程数量。它的初始值是1，表示最多只能有一个线程进入临界区。

实战示例：
```java
import java.util.concurrent.Semaphore;

public class SemaphoreExample {
    private final Semaphore semaphore = new Semaphore(1);

// 在任务完成之后尝试获取信号量，如果失败则表示资源已被占用
        if (!semaphore.tryAcquire()) {
            System.out.println("Resource already in use. Task skipped.");
            return;
        }

// 如果成功获取信号量，则允许一个线程进入临界区
        semaphore.release();
    }

public static void main(String[] args) {
        SemaphoreExample example = new SemaphoreExample();

// 创建一个线程来执行任务
        Thread taskThread = new Thread(example::task);
        taskThread.start();

System.out.println("All tasks completed!");
    }
}
```
在这个例子中，我们使用Semaphore来限制同时进入临界区的线程数量。如果尝试获取信号量但失败，则表示资源已被占用。