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),这时如果有请求进来想要被处理,则需要先从桶里获取一个令牌,当桶里没有令牌可取时,则该请求被拒绝服务。令牌桶算法通过控制桶容量、发放令牌速率,来达到对请求的限制。