Redis限流又踩坑了，真让人头大

程序员小义

发布于 2024-05-30 16:20:40

900

发布于 2024-05-30 16:20:40

文章被收录于专栏：小义思小义思

在分布式微服务架构中，限流、缓存、降级/熔断，是维护系统稳定运行的三大利器。而Redis天生支持分布式，可以很容易地作为分布式限流的解决方案。但redis限流还需谨慎，否则一不小就踩了坑，来看看下面的例子。

在业务场景中，对用户进行接口限流，在一小时内只允许客户访问3000次请求，核心代码大致如下。

private boolean ExceedingLimit(String userId) {
        String totalKey = "TOTAL_" + userId;
        if (stringRedisTemplate.hasKey(totalKey)) {
            // 对value进行加1操作
            stringRedisTemplate.opsForValue().increment(totalKey, 1);
            // 获取key所对应的value
            Integer num = Integer.parseInt((String) stringRedisTemplate.opsForValue().get(totalKey));
            //总额度3000
            if (num >= 3000) {
                return true;
            }
            return false;
        } else {
            // 如果没有key值，对他进行添加到redis中
            stringRedisTemplate.opsForValue().set(totalKey, "1", 1, TimeUnit.HOURS);
        }
        return false;
    }

代码写得很好，下次不要再写了。细心的小伙伴可能一下子就看出来了，redis的检查key命令和给value值加一命令是分开的，不能保证原子性，是会有bug的。

虽然说redisTemplate的increment方法不会修改key的过期时间，但是如果线程在执行完if判断key存在进入if语句后，key这时正好过期，那么increment就会新增这个key，并且值为1，而且没有过期时间。所以这个key一直存在，客户刷超了限额就一直无法访问，这不是等着被客户骂吗。另外在高并发条件下，可能也会存在num取值不准确的情况。

解决上述问题最好的方案还是用lua脚本来实现。将key判断和设置值的操作打包在一个原子操作中执行。这样即使是在高并发的环境下，也可以保证当key被设置后过期时间必定会被正确设置。

lua脚本代码：

local totalKey = KEYS[1]
local maxNum = tonumber(ARGV[1])
local expireSec = tonumber(ARGV[2])//过期时间
local current = redis.call('get', totalKey)
if current then
    current = tonumber(current) + 1
    redis.call('incrby', totalKey, "1")
else
    current = 1
    redis.call('set', totalKey, tostring(current), 'EX', expireSec)
end
return current < maxNum

执行lua脚本：

public Boolean permit(String personId, Integer maxNUm, Integer expireSecond) {
        String totalKey = "TOTAL_" + personId;
        DefaultRedisScript<Boolean> redisScript = new DefaultRedisScript<>();
        redisScript.setScriptSource(new ResourceScriptSource(new ClassPathResource("limit.lua")));
        redisScript.setResultType(Boolean.class);
        Boolean result = redisTemplate.execute(redisScript, Arrays.asList(totalKey), String.valueOf(maxNUm), String.valueOf(expireSecond));
        log.info("result:{}", result);
        return result;
    }

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2024-05-29，如有侵权请联系 cloudcommunity@tencent.com 删除

redis