在Java中，`Semaphore` 类是一个强大的并发控制工具，它属于 `java.util.concurrent` 包。`Semaphore` 本质上是一个计数信号量，用于控制对共享资源的访问数量。它允许一个或多个线程同时访问某个特定资源或资源池，但总数不超过设置的许可数（permits）。这种机制在限制并发访问共享资源时非常有用，比如数据库连接池、线程池、或者任何需要限制访问次数的场景。 ### Semaphore 的基本工作原理 `Semaphore` 的核心在于其内部的计数器（permits 计数器），它表示当前可用的许可数。当线程需要访问共享资源时，它会尝试从 `Semaphore` 中获取一个许可（通过调用 `acquire()` 方法）。如果许可数大于0，则许可被减少一个，线程继续执行。如果许可数为0，则线程将被阻塞，直到有许可可用。线程完成资源访问后，应该释放许可（通过调用 `release()` 方法），这样其他等待的线程就可以获取许可并继续执行。 ### Semaphore 的主要方法 - `acquire()`: 尝试获取一个许可，如果当前没有可用的许可，则当前线程将被阻塞，直到有许可可用。 - `acquire(int permits)`: 尝试获取指定数量的许可，行为类似 `acquire()`，但会尝试一次性获取多个许可。 - `acquireUninterruptibly()`: 尝试获取一个许可，但与 `acquire()` 不同的是，这个方法在等待许可的过程中不会响应中断。 - `acquireUninterruptibly(int permits)`: 尝试获取指定数量的许可，同样不响应中断。 - `release()`: 释放一个许可，将其返回给 `Semaphore`，可能会唤醒一个或多个等待获取许可的线程。 - `release(int permits)`: 释放指定数量的许可。 - `availablePermits()`: 返回当前可用的许可数。 - `drainPermits()`: 获取并返回当前可用的所有许可，并立即将许可数设置为0。 ### Semaphore 在并发控制中的应用 #### 示例场景：数据库连接池 假设我们有一个数据库连接池，它最多可以管理10个数据库连接。任何需要访问数据库的操作都需要从这个连接池中获取连接。使用 `Semaphore` 可以很容易地实现这种限制。 ```java import java.util.concurrent.Semaphore; public class DatabaseConnectionPool { private final Semaphore semaphore; private final Connection[] connections; // 假设Connection是一个表示数据库连接的类 public DatabaseConnectionPool(int maxConnections) { this.semaphore = new Semaphore(maxConnections); this.connections = new Connection[maxConnections]; // 初始化connections数组，这里只是示意，实际中需要创建真实的数据库连接 } public Connection getConnection() throws InterruptedException { semaphore.acquire(); // 获取一个许可，表示占用一个连接 // 这里应实现某种策略来选择一个可用的连接，这里简化处理 return connections[getAvailableConnectionIndex()]; } private int getAvailableConnectionIndex() { // 简化的逻辑，实际中需要更复杂的逻辑来管理连接的分配和回收 return 0; // 示例中总是返回第一个连接 } public void releaseConnection(Connection connection) { // 假设connection对象可以安全地放回池中 semaphore.release(); // 释放一个许可，表示连接被释放 // 实际的连接回收逻辑会更复杂，包括检查连接是否仍然有效等 } // 其他方法和成员... } ``` 在上述示例中，`Semaphore` 被用来限制同时获取数据库连接的线程数。每当一个线程尝试从连接池中获取连接时，它都会尝试从 `Semaphore` 中获取一个许可。如果所有连接都在使用中，则线程将被阻塞，直到有连接被释放并返回许可。 #### 示例场景：限制并发任务数 另一个常见的使用场景是限制同时执行的任务数。比如，你有一个任务队列，但出于资源限制（如CPU、内存等），你希望同时执行的任务数不超过某个阈值。 ```java import java.util.concurrent.*; public class TaskExecutor { private final ExecutorService executorService; private final Semaphore semaphore; public TaskExecutor(int maxConcurrentTasks, int threadPoolSize) { this.executorService = Executors.newFixedThreadPool(threadPoolSize); this.semaphore = new Semaphore(maxConcurrentTasks); } public void executeTask(Runnable task) { try { semaphore.acquire(); // 尝试获取许可 executorService.submit(() -> { try { task.run(); } finally { semaphore.release(); // 确保在任务执行完毕后释放许可 } }); } catch (InterruptedException e) { // 处理中断异常，这里可以重新尝试获取许可或进行其他处理 Thread.currentThread().interrupt(); // 重新设置中断状态 } } // 关闭ExecutorService等清理工作... } ``` 在这个例子中，`Semaphore` 被用来限制同时提交给 `ExecutorService` 的任务数。每个任务在提交前都会尝试从 `Semaphore` 中获取一个许可。如果许可数不足，则提交任务的线程将被阻塞，直到有许可可用。这种方式可以确保即使在高负载情况下，系统的资源使用也不会超过预定的阈值。 ### 总结 `Semaphore` 是Java并发编程中一个非常有用的工具，它提供了一种灵活的方式来限制对共享资源的并发访问。通过控制许可的数量，`Semaphore` 可以帮助开发者设计出既高效又安全的并发系统。无论是在管理数据库连接池、限制并发任务数，还是在其他需要限制资源访问的场景中，`Semaphore` 都能发挥重要作用。在实际开发中，合理利用 `Semaphore` 可以显著提升系统的稳定性和性能。 希望这篇文章能够帮助你更好地理解 `Semaphore` 在Java并发控制中的应用，并能在你的项目中有效地利用它。如果你对Java并发编程的其他方面也有兴趣，不妨关注码小课网站，那里有更多的教程和实例等待你去探索。