redis简单限流

需求：

          如果要保证一个用户一分钟内只能访问5次接口，超过就拒绝范围。

学习老钱的简单限流，这边重复阐述一下，以便更多人能借鉴。

这个限流需求中存在一个滑动时间窗口，想想 zset 数据结构的 score 值，是不是可以通过 score 来圈出这个时间窗口来。而且我们只需要保留这个时间窗口，窗口之外的数据都可以砍掉。那这个 zset 的 value 填什么比较合适呢？它只需要保证唯一性即可，用 uuid 会比较浪费空间，那就改用毫秒时间戳吧。

直接贴代码

public class SimpleRateLimiter {
    private Jedis jedis;

    public SimpleRateLimiter(Jedis jedis) {
        this.jedis = jedis;
    }

    public boolean isActionAllowed(String userId, String actionKey, int period, int maxCount) {
        String key = String.format("hist:%s:%s", userId, actionKey);
        long nowTs = System.currentTimeMillis();
        Pipeline pipe = jedis.pipelined();
        pipe.multi();
        pipe.zadd(key, nowTs, "" + nowTs);
        //移除超过5秒前的记录
        pipe.zremrangeByScore(key, 0, nowTs - period * 1000);
        //计算该key有多少条数据
        Response<Long> count = pipe.zcard(key);
        //设置过期时间
        pipe.expire(key, period + 1);
        pipe.exec();
        pipe.close();
        return count.get() <= maxCount;
    }

    public static void main(String[] args) {
        HttpDispatcher.getInstance().load("com.cloud.gold");
        Jedis jedis = RedisUtil.INSTANCE.getJedis();
        SimpleRateLimiter limiter = new SimpleRateLimiter(jedis);
        for(int i=0;i<20;i++) {
            System.out.println(limiter.isActionAllowed("laoqian", "reply", 60, 5));
        }
    }

}

再次重复，本文部分内容转载老钱的redis书籍