用Redis的Zset数据结构做接口限流
·
在公司业务项目中遇到了一个高并发接口,因为那个接口对数据库有很高的写和读的需求,再加之公司服务器会跑很多项目,稍不注意就会使服务器宕机,所以想出了接口限流的方法,是项目正常流转. 下面就是redis用set做的一个一分钟限制5次的限流处理.
//可以灵活设置时间得接口请求数 @GetMapping("/Sliding") @ApiOperation("滑动限流") public String testSlidingWindow() { Long currentTime = System.currentTimeMillis(); if (redisTemplate.hasKey("limit")) { // intervalTime是限流的时间 Long intervalTime = 60000L; Integer count = redisTemplate.opsForZSet().rangeByScore("limit", currentTime - intervalTime, currentTime).size(); System.out.println(count); if (count != null && count > 5) { System.out.println("静止访问"); return "每分钟最多只能访问5次"; } } redisTemplate.opsForZSet().add("limit", UUID.randomUUID().toString(), currentTime); System.out.println("访问成功"); return "访问成功"; }
下面是用swagger测试之后的图片

但是后续我们发现这样会使redis中zset的数据结构变大,但是也不影响使用,最好的方法还可以用Redis集成的Ression令牌桶去限流 这个方法
@Resource
private Redisson redisson;
@GetMapping("/Token")
@ApiOperation("桶限流")
public String testTokenBucket() {
RRateLimiter rateLimiter = redisson.getRateLimiter("myRateLimiter");
// 最大流速 = 每10秒钟产生1个令牌 10秒一个令牌
rateLimiter.trySetRate(RateType.OVERALL, 1, 10, RateIntervalUnit.SECONDS);
//需要1个令牌
if (rateLimiter.tryAcquire(1)) {
return "可以访问";
}
System.err.println("静止访问");
return "不好意思,请过十秒钟再来~~~~~~~";
}
DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐
所有评论(0)