1、计数器
java内部可以使用原子计数器AtomicInteger\Semaphore信号量来做简单的限流
// 限流的个数
private int maxCount = 10;
// 指定的时间内
private long interval = 60;
// 原子类计数器
private AtomicInteger atomicInteger = new AtomicInteger(0);
// 起始时间
private long startTime = System.currentTimeMillis();
public boolean limit(int maxCount, int interval) {
atomicInteger.addAndGet(1);
if (atomicInteger.get() == 1) {
startTime = System.currentTimeMillis();
atomicInteger.addAndGet(1);
return true;
}
// 超过了间隔时间,直接重新开始计数
if (System.currentTimeMillis() - startTime > interval * 1000) {
startTime = System.currentTimeMillis();
atomicInteger.set(1);
return true;
}
// 还在间隔时间内,check有没有超过限流的个数
if (atomicInteger.get() > maxCount) {
return false;
}
return true;
}
2、漏桶算法
漏桶算法思路比较简单,我们把水比作是请求,漏桶比作是系统处理能力极限,水先进入到漏桶里,漏桶会按照一定速率流出,当流出的速率小于流入的速率时,由于漏桶容量有限,后续进入的水直接溢出。
3、令牌桶算法
系统会维护一个令牌桶,以一个恒定的速度往桶里放入令牌(token),这时如果有请求进来想要被处理,则需要先从桶里获取一个令牌,当桶里没有令牌可取时,则该请求被拒绝服务。令牌桶算法通过控制桶容量、发放令牌速率,来达到对请求的限制。