07_redis_简单限流

当系统处理能力不足的时候，需要组织计划之外的请求继续对系统施压。
除了控制限流，还用于控制用户的行为。

Redis实现限流

用一个 zset 结构记录用户的行为历史，每一个行为都会作为 zset 中的一个key 保存下来。同一个用户同一种行为用一个 zset 记录。
为节省内存，我们只需要保留时间窗口内的行为记录，同时如果用户是冷用户，滑动时间窗口内的行为是空记录，那么这个 zset 就可以从内存中移除，不再占用空间。
通过统计滑动窗口内的行为数量与阈值 max_count 进行比较就可以得出当前的行为是否允许。
代码表示：

public class SimpleRateLimiter {
    private Jedis jedis;

    public SimpleRateLimiter(Jedis jedis) {
        this.jedis = jedis;
    }
    /**
     * 查看用户操作是否正常
     * @param userId 用户Id
     * @param actionKey 行为名
     * @param period 时间范围
     * @param maxCount 最大次数
     * @return 是否正常
     * @throws IOException
     */
    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();
        // 记录行为 value和score都是毫秒时间戳
        pipe.zadd(key, nowTs, "" + nowTs);
        // 移除超时行为
        pipe.zremrangeByScore(key, 0, nowTs - period * 1000*1000);
        // 查找时间窗口中的行为数量，获取period内的该操作的次数
        Response<Long> count = pipe.zcard(key);
        // 重制所有行为的过期时间，避免冷用户持续占用内存，过期时间应等于窗口长度，这里多宽限1秒
        pipe.expire(key, period + 1);
        pipe.exec();
        pipe.close();
        return count.get() <= maxCount;
    }

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

这段代码还是略显复杂，需要读者花一定的时间好好啃。它的整体思路就是：每一个行为到来时，都维护一次时间窗口。将时间窗口外的记录全部清理掉，只保留窗口内的记录。
zset 集合中只有 score 值非常重要，value 值没有特别的意义，只需要保证它是唯一的就可以了。
因为这几个连续的 Redis 操作都是针对同一个 key 的，使用 pipeline 可以显著提升Redis 存取效率。但这种方案也有缺点，因为它要记录时间窗口内所有的行为记录，如果这个量很大，比如限定 60s 内操作不得超过 100w 次这样的参数，它是不适合做这样的限流的，因为会消耗大量的存储空间。

最后编辑于：2020.07.07 16:34:52