用Redis的Zset数据结构做接口限流

最新推荐文章于 2024-12-20 13:53:41 发布

继承java之父

最新推荐文章于 2024-12-20 13:53:41 发布

阅读量411

点赞数

文章标签： redis java 数据库

本文链接：https://blog.csdn.net/qq_52611659/article/details/131248175

版权

在处理高并发接口时，由于对数据库的读写需求以及服务器资源限制，采用Redis进行接口限流。文中示例展示了使用RedisSet实现的一分钟内限制5次请求的滑动窗口限流，以及如何切换到Redis内置的RateLimiter（令牌桶算法）进行更高效的限流，以避免数据结构过大并精细控制访问速率。

摘要由CSDN通过智能技术生成

在公司业务项目中遇到了一个高并发接口，因为那个接口对数据库有很高的写和读的需求，再加之公司服务器会跑很多项目，稍不注意就会使服务器宕机，所以想出了接口限流的方法，是项目正常流转. 下面就是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 "不好意思，请过十秒钟再来~~~~~~~";

}