Semaphore 实现限流器

作者：知乎用户
链接：https://www.zhihu.com/question/586093481/answer/2917191929
来源：知乎
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

实现一个基于Semaphore的限流器，可以按照以下步骤进行：

1.创建Semaphore对象并初始化计数器

Semaphore semaphore = new Semaphore(permits);

其中，permits表示许可证的数量，即同时允许访问资源的线程数。

2.在需要限流的代码块中，尝试获取许可证

try {
    semaphore.acquire();
    // 执行需要限流的代码
} catch (InterruptedException e) {
    // 处理异常
} finally {
    semaphore.release();
}

其中，acquire()方法尝试获取一个许可证，如果没有可用的许可证，则会阻塞线程。release()方法释放一个许可证。

3.下面是一个基于Semaphore实现的限流器示例：

public class RateLimiter {
    private final Semaphore semaphore;

    public RateLimiter(int permits) {
        semaphore = new Semaphore(permits);
    }

    public void acquire() {
        try {
            semaphore.acquire();
            // 执行需要限流的代码
        } catch (InterruptedException e) {
            // 处理异常
        } finally {
            semaphore.release();
        }
    }
}

使用时，可以创建一个RateLimiter对象，并在需要限流的代码块中调用acquire()方法：

RateLimiter rateLimiter = new RateLimiter(10);
// 在需要限流的代码块中调用acquire()方法
rateLimiter.acquire();