20 lines
441 B
Lua
20 lines
441 B
Lua
-- 限流Lua脚本
|
||
-- KEYS[1] 是键，ARGV[1] 是限流次数，ARGV[2] 是限流时间
|
||
local key = KEYS[1]
|
||
local limit = tonumber(ARGV[1])
|
||
local timeout = tonumber(ARGV[2])
|
||
|
||
-- 当前计数
|
||
local current = redis.call('incr', key)
|
||
if current == 1 then
|
||
-- 如果是第一次访问，设置过期时间
|
||
redis.call('expire', key, timeout)
|
||
return 1
|
||
end
|
||
|
||
-- 检查是否超过限流
|
||
if current > limit then
|
||
return 0
|
||
end
|
||
return 1
|