java学习之使用redis简单实现限流功能

最新推荐文章于 2024-10-31 10:25:36 发布

一个正在改变的积极fw

最新推荐文章于 2024-10-31 10:25:36 发布

阅读量621

点赞数 9

CC 4.0 BY-SA版权

分类专栏： java学习之工作代码文章标签： java 学习 redis

本文链接：https://blog.youkuaiyun.com/qq_44922250/article/details/139347517

事先声明基于redis实现滑动时间窗实现限流这种方式我们工作中一般不会使用,因为目前市面上对于大体量的服务使用的都是springCloud netFlix或者spring CloundAlibaba(主流)他们都有各自的限流组件( Netflix Hystrix , sentinel), 小体量服务又基本用不到限流,这里可以当成是时间窗口算法的一种学习

滑动时间窗口限流是一种基于时间的流量控制策略,本质就是规定一个时间大小的窗口,值统计在这个窗口内的请求,然后滑动这个窗口;至于怎么滑动窗口,可以那请求的当前时间-设置的窗口大小来完成

具体实现:

主要步骤:

1.定义窗口的大小,如30秒

2.每次收到请求,就定义一个zset存储到redis中

3.然后通过ZREMVRANGBYSCORE命令来删除窗口起始时间之前的数据

4.最后通过ZCARD命令获取zset集合中的成员数量,及窗口内的请求量

public class SlidingWindowRateLimiter {
    private final Jedis jedis;
    public SlidingWindowRateLimiter(Jedis jedis) {
        this.jedis = jedis;
    }
    public boolean isAllowed(String key, int limit, long windowSize, long currentTime) {
        // 计算时间窗口的起始时间
        long windowStart = currentTime - windowSize;
        // 移除时间窗口之前的数据
        jedis.zremrangeByScore(k