redis lua限流写法

最新推荐文章于 2025-02-10 23:34:35 发布

原创最新推荐文章于 2025-02-10 23:34:35 发布 · 432 阅读

0 ·

CC 4.0 BY-SA版权

Redis 专栏收录该内容

3 篇文章

订阅专栏

这段代码使用Redis实现业务限流逻辑。定义了业务标志、限流数量、限流标志等参数，通过获取系统配置或默认值确定限流规则。统计当前时间窗口内的访问次数，若超出限流则返回0，未超出则增加访问次数并返回当前次数。

-- 业务标志
local businessTag=KEY[1]
-- 限流数量
local limitDefault=tonumber(KEY[2] or 100)
-- 限流标志
local limitOnOffTag=KEY[3]
-- 限流开关默认是开
local limitOnOff=tonumber(redis.call("get",limitOnOffTag..":REQ_LIMIT_ON_OFF") or 0)
if(limitOnOff>0){
return 1
}
-- 如果没有获取到系统配置的限流数量就获取传入的默认限流数量
local limit = tonumber(redis.call("get",limitOnOffTag..":req_limit") or limitDefault))
-- 当前时间窗口内访问次数统计缓存KEY
local key = businessTag..":counter:"..limitOnOffTag
-- 获取当前时间范围内已访问次数
local current=tonumber(redis.call("get",key) or 0));
-- 设置统计时间范围
local windowTimeOutSetting=tonumber(redis.call("get",limitOnOffTag..":request_window_timeout") or 1)
-- 如果历史还没有设置访问就设置一个key
if(current<1){
redis.call("set",key,0)
redis.call("expire",key,windowTimeOutSetting)
current=0;
}
-- 当前时间范围内访问次数已经大于限流则返回0
if(current+1>limitDefault) then
return 0
else
-- 当前时间范围内访问次数处于限流范围内则返回已访问次数
redis.call("INCRBY", key ,"1")
return (current+1)
end