在Java中,实现限流有多种方式,以下是其中一些常见的方案:
1、数器限流: 使用一个计数器记录请求的数量,当达到设定的阈值时拒绝后续请求,可以通过AtomicInteger等原子类来实现。需要注意计数器的重置和线程安全问题。
import java.util.concurrent.atomic.AtomicInteger;
public class CounterRateLimiter {
private static final int MAX_REQUESTS = 10;
private static AtomicInteger counter = new AtomicInteger(0);
public boolean allowRequest() {
return counter.incrementAndGet() <= MAX_REQUESTS;
}
}
2、令牌桶算法: 令牌桶算法是一种漏桶算法,通过固定速率往令牌桶中放入令牌,请求需要获取令牌才能被处理。Guava库中提供了RateLimiter
类,实现了令牌桶算法。
import com.google.common.util.concurrent.RateLimiter;
public class TokenBucketRateLimiter {
private static final double REQUESTS_PER_SECOND = 10.0;
private static RateLimiter rateLimiter = RateLimiter.create(REQUESTS_PER_SECOND);
public bool