赞
踩
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;
- }
漏桶算法思路比较简单,我们把水比作是请求,漏桶比作是系统处理能力极限,水先进入到漏桶里,漏桶会按照一定速率流出,当流出的速率小于流入的速率时,由于漏桶容量有限,后续进入的水直接溢出。
系统会维护一个令牌桶,以一个恒定的速度往桶里放入令牌(token),这时如果有请求进来想要被处理,则需要先从桶里获取一个令牌,当桶里没有令牌可取时,则该请求被拒绝服务。令牌桶算法通过控制桶容量、发放令牌速率,来达到对请求的限制。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。