基于Spring HandlerInterceptor和Google RateLimiter 的简单throttling

项目中的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,可以换成其他存储。

猜你喜欢

转载自blog.csdn.net/u013275741/article/details/51752470