lua 优点
减少网络开销: 不使用 Lua 的代码需要向 Redis 发送多次请求, 而脚本只需一次即可, 减少网络传输;
原子操作: Redis 将整个脚本作为一个原子执行, 无需担心并发, 也就无需事务;
复用: 脚本会永久保存 Redis 中, 其他客户端可继续使用.
计数器模式:
利用lua脚本一次性完成处理达到原子性,通过INCR自增计数,判断是否达到限定值,达到限定值则返回限流,添加key过期时间应该范围过度
$lua = '
local i = redis.call("INCR", KEYS[1])
if i > 10 then
return "wait"
else
if i == 1
then
redis.call("expire", KEYS[1], KEYS[2])
end
return redis.call("get", KEYS[3])
end
';
laravel 请求代码:
Redis::eval($lua, 3, sprintf(RedisKey::API_LIMIT, $key, $callService['service']), 60, $cache_key);
令牌桶模式
每次请求在桶内拿取一个令牌,有令牌则通过,否则返回,并且按照算法一定的慢慢把令牌放入桶内
注释:
KEY[1] = 查找数据key(这个是我存redis中的key你可以改成return true也可以,或者其他)
KEY[2] = 限流的key
KEY[3] = 桶内数量
KEY[4] = 时间戳
KEY[5] = 过期时间(这个就是你这个桶内的有效时间,时间越大漏铜恢复的越慢)
核心:
local newNum = math.min(KEYS[3], math.floor(((dataJson["limitVal"] - 1) + (KEYS[3]/KEYS[5]) * (KEYS[4] - dataJson["limitTime"]))))
大概是:
math.min(桶内数量, math.floor(((存入redis的桶数量 - 1) + (桶内数量 / 过期时间) * (时间戳 - 存入redis的时间戳))))
$lua = '
local data = redis.call("get", KEYS[2])
if data then
local dataJson = cjson.decode(data)
local newNum = math.min(KEYS[3], math.floor(((dataJson["limitVal"] - 1) + (KEYS[3]/KEYS[5]) * (KEYS[4] - dataJson["limitTime"]))))
if newNum > 0 then
local paramsJson = cjson.encode({limitVal=newNum,limitTime=KEYS[4]})
redis.call("set", KEYS[2], paramsJson)
return redis.call("get", KEYS[1])
end
return "wait"
end
local paramsJson = cjson.encode({limitVal=KEYS[3],limitTime=KEYS[4]})
redis.call("set", KEYS[2], paramsJson)
return redis.call("get", KEYS[1])
';
// 1. lua脚本, 2 KEYS数量, 3 查找数据key, 4 限制key, 5 桶内数量, 6 时间戳, 7 过期时间
Redis::eval(1,2,3,4,5,6,7参数);
小思路:
把限制key换成用户的某个唯一字段(维度)可针对用户限流
把桶内数量和时间戳换成动态的,根据请求的接口方法(维度)来设置多少,可根据不同的方法设置不一样的限流大小
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。