用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 "不好意思，请过十秒钟再来~~~~~~~";

}