redis+lua实现分布式限流

在Redis中使用Lua脚本实现分布式限流可以确保操作的原子性，以下是一个简单的Lua脚本示例，用于实现一个基于Redis的分布式限流器：

local key = KEYS[1]
local limit = tonumber(ARGV[1])
local current = tonumber(redis.call('get', key) or "0")
if current + 1 > limit then
    return 0
else
    redis.call('INCR', key)
    redis.call('EXPIRE', key, 10) -- 设置键的过期时间以避免内存泄漏
    return 1
end

在应用程序中，你可以通过Redis客户端执行这个Lua脚本。以下是一个使用Python和redis-py客户端的示例：

import redis
 
# 连接到Redis
r = redis.StrictRedis(host='localhost', port=6379, db=0)
 
# Lua脚本
lua_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 0
else
    redis.call('INCR', key)
    redis.call('EXPIRE', key, 10)
    return 1
end
"""
 
# 定义键和限制
key = 'rate_limiter'
limit = 100
 
# 使用Lua脚本执行限流
result = r.eval(lua_script, 1, key, limit)
if result == 0:
    print("请求超出限制")
else:
    print("请求被允许")

这个示例中，我们定义了一个名为rate_limiter的键和一个每十秒的请求限制（这里为100次）。每当一个请求发生时，Lua脚本会被执行，检查当前的请求数是否超过了限制。如果没有超过，请求数会增加，并允许请求通过；如果超过了限制，请求会被拒绝。