分布式限流——Redis + Lua实现滑动窗口算法

-- 使用Redis和Lua脚本实现滑动窗口限流
 
-- 初始化限流配置
local limit_count = tonumber(ARGV[1]) -- 限流阈值
local limit_time_in_seconds = tonumber(ARGV[2]) -- 时间窗口
local current_time = tonumber(ARGV[3]) -- 当前时间戳
local cache_key = KEYS[1] -- 缓存键
 
-- 计算窗口开始时间和结束时间
local window_start_time = current_time - limit_time_in_seconds
local window_end_time = current_time
 
-- 检查是否超出限制
local count = redis.call('zcount', cache_key, window_start_time, window_end_time)
if count < limit_count then
    -- 未超出限制，添加当前请求到缓存，并设置过期时间等于窗口时长
    redis.call('zadd', cache_key, current_time, current_time)
    redis.call('expire', cache_key, limit_time_in_seconds)
    return true
else
    -- 超出限制，不允许通过
    return false
end

这段Lua脚本用于Redis中，通过Redis的zadd和zcount命令实现了滑动窗口限流算法。它会检查在指定的时间窗口内的请求数量是否超过了限制，如果没有超过，则允许通过当前请求并更新缓存。如果超过了限制，则不允许通过。这是一个简单而有效的分布式限流解决方案。