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);
令牌桶模式
每次请求在桶内拿取一个令牌, 有令牌则通过, 否则返回, 并且按照算法一定的慢慢把令牌放入桶内
- $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 参数);
来源: https://www.jb51.net/article/159068.htm