使用 Redis + Lua 实现分布式限流

local redis = require 'redis'
 
-- 连接Redis
local client = redis.connect('127.0.0.1', 6379)
 
-- Lua脚本用于实现分布式限流
local script = [[
    local key = KEYS[1]
    local limit = tonumber(ARGV[1])
    local current = tonumber(redis.call('get', key) or "0")
    if current + 1 > limit then
        return false
    else
        redis.call('INCR', key)
        redis.call('EXPIRE', key, ARGV[2])
        return true
    end
]]
 
-- 调用Lua脚本进行限流
-- 参数：
-- resource_name：资源的唯一标识符
-- max_permits：最大许可数
-- expiration_secs：限流器的过期时间（秒）
function is_allowed(resource_name, max_permits, expiration_secs)
    local result = client:eval(script, 1, resource_name, max_permits, expiration_secs)
    return result == true
end
 
-- 使用示例
local allowed = is_allowed("my_api_call", 100, 60)
if allowed then
    print("访问被允许")
else
    print("超出访问限制")
end

这段代码展示了如何使用Redis和Lua脚本来实现一个简单的分布式限流器。它首先连接到Redis，然后定义了一个Lua脚本，该脚本用于检查是否超过了对特定资源的访问限制。如果没有超过限制，则允许访问并更新计数器。最后，提供了一个使用该限流器的示例函数，并根据返回值判断是否允许访问。