本文章主要有两部分。
第一部分主要是提供限流器的一个构建方法,以及提供流速调节的方法。
第二部分是讲述怎么使用限流器。
第一部分:
这一部分主要是提供构建一个限流器,并且提供流速的实时调节
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import com.google.common.util.concurrent.RateLimiter;
/*
* 用于限流和流量调节
* 原理是基于令牌桶算法
* 使用方式是:
* 1.通过getRateLimiter获取rateLimiter.
* 2.然后根据rateLimiter.tryAcquire()的返回的布尔值来进行判断是否可以执行正常流程。
* 3.如果返回的是true,则表明获取到令牌可以执行正常的流程,否则就意味着没有获取令牌,被限流。
* 有两点需要注意的:
* 第一,RateLimiter是按照一定频率往桶里放入令牌,频率是1/QPS秒
* 第二,RateLimiter最多会存储QPS个令牌
*
*/
public class RateLimiterBuilder {
private static final double DEFALUT_RATE = 200;
staticObjectobj=newObject();
private static Map rateLimiterMap=new ConcurrentHashMap();
/*
* 根据key和rate获取一个rateLimiter
* rate是流速,也就是允许的QPS
* 1.根据key去map里取rateLimiter,然后还要判断现在的rate和传人的rate是否相等,如果已经有这样一个rateLimiter(key和rate)就直接返回
* 2.如果传人的rate和现在的不相等则调用changeRate进行调速度
* 3.否则创建一个新的rateLimiter
*/
public static RateLimiter getRateLimiter(String key,double rate) {
if(rate<= 0) {
rate = DEFALUT_RATE;
}
RateLimiter limiter=rateLimiterMap.get(key);
if(limiter!=null&&limiter.getRate() ==rate) {
return limiter;
}
synchronized (obj) {
limiter = rateLimiterMap.get(key);
if(limiter!=null&&limiter.getRate() ==rate) {
return limiter;
}
if(limiter!=null) {
return changeRate(limiter,rate);
}
limiter= RateLimiter.create(rate);
rateLimiterMap.put(key, limiter);
}
return limiter;
}
/*
* 调整RateLimiter的流量
*/
public static synchronized RateLimiter changeRate(RateLimiterlimiter,doublerate) {
if(limiter.getRate() ==rate){
returnlimiter;
}
limiter.setRate(rate);
returnlimiter;
}
}
第二部分:
使用示例:
double rate = get();//通过配置中心去获取一个限流的速率
RateLimiter limiter= RateLimiterBuilder.getRateLimiter("limiterName",rate);
if(limiter!=null) {
if(!limiter.tryAcquire()) {
return weatherForecastResp;//被限流了,直接返回
}
}
//下面是相关的业务代码。
也可以写一个拦截器,将这些代码包起来,在方法执行前,拦截一下,做限流。
网友评论