项目中的throttling只要是为了api限流。
由于是基于 Spring HandlerInterceptor + Google RateLimiter实现,代码量比较少。核心思想可以查阅漏桶算法。
由于是api,控制粒度未api的接口,所以这采用
static ConcurrentHashMap<String, RateLimiter> rateLimiterPool
来管理各个接口的流量。
HandlerInterceptor.preHandle 中,通过rateLimiter.tryAcquire() 控制流量是否超标。
if (rateLimiterPool.containsKey(key)) {//已经加入计数器
if (rateLimiterPool.get(key).tryAcquire()) {
return true;
}else {
this.handl429(arg1);
return false;
}
}else {
String limit = this.redisService.get(EncryptionUtil.md5L32(appId + uri));
if (StringUtils.isEmpty(limit)) {
this.logger.info("Missing configuration of uri:{}. AppId:{}.", limit, appId);
//未配置appId 控制时,查找DEFAULT配置
limit = this.redisService.get(EncryptionUtil.md5L32("DEFAULT" + uri));
if (StringUtils.isEmpty(limit)) {
this.logger.info("No default configuration.");
}else {
RateLimiter rateLimiter = RateLimiter.create(Double.valueOf(limit));
rateLimiterPool.put(key, rateLimiter);
}
}else {
RateLimiter rateLimiter = RateLimiter.create(Double.valueOf(limit));
rateLimiterPool.put(key, rateLimiter);
}
return true;
}
至于redis,可以换成其他存储。