基于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,可以换成其他存储。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值