【智能排班系统】基于Redis的increment命令和lua脚本实现IP限流

-- 假设有一个名为"rate_limiter"的Redis键，用于跟踪用户的请求次数
-- 假设限制时间窗口为60秒，允许的最大请求数为100
 
-- 使用Lua脚本原子地检查和更新计数器
local key = "rate_limiter:" .. KEYS[1]
local limit = 100
local window = 60
 
-- 获取当前的计数
local count = redis.call('get', key)
if count == false then
    -- 如果键不存在，设置初始计数为0，并设置过期时间
    redis.call('set', key, 0, 'EX', window)
    count = 0
else
    count = tonumber(count)
end
 
-- 检查是否超出了限制
if count >= limit then
    return redis.error_reply('超出请求限制')
else
    -- 如果没有超出限制，增加计数
    redis.call('INCR', key)
    -- 返回当前计数
    return redis.status_reply(count + 1)
end

这段Lua脚本用于实现一个简单的IP限流器。它首先检查是否有对应的键来跟踪请求计数，如果没有，它会创建一个初始化为0的键，并设置一个60秒的过期时间。如果请求计数超过100，脚本会返回一个错误。否则，它会增加计数并允许请求继续。这个例子演示了如何使用Lua脚本和Redis的INCR命令来实现请求限流。