当前位置:首页 > PHP教程 > php高级应用 > 列表

php使用lua+redis实现限流,计数器模式,令牌桶模式

发布:smiling 来源: PHP粉丝网  添加日期:2021-11-15 10:43:17 浏览: 评论:0 

这篇文章主要介绍了php使用lua+redis实现限流,计数器模式,令牌桶模式,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧。

lua 优点

减少网络开销: 不使用 Lua 的代码需要向 Redis 发送多次请求, 而脚本只需一次即可, 减少网络传输;

原子操作: Redis 将整个脚本作为一个原子执行, 无需担心并发, 也就无需事务;

复用: 脚本会永久保存 Redis 中, 其他客户端可继续使用.

计数器模式:

利用lua脚本一次性完成处理达到原子性,通过INCR自增计数,判断是否达到限定值,达到限定值则返回限流,添加key过期时间应该范围过度

  1. $lua = ' 
  2.         local i = redis.call("INCR", KEYS[1])  
  3.         if i > 10 then 
  4.           return "wait" 
  5.         else 
  6.           if i == 1 
  7.           then 
  8.             redis.call("expire", KEYS[1], KEYS[2]) 
  9.           end 
  10.           return redis.call("get", KEYS[3]) 
  11.         end 
  12.       '; 

laravel 请求代码:

Redis::eval($lua, 3, sprintf(RedisKey::API_LIMIT, $key, $callService['service']), 60, $cache_key);

令牌桶模式

每次请求在桶内拿取一个令牌,有令牌则通过,否则返回,并且按照算法一定的慢慢把令牌放入桶内。

  1. $lua = ' 
  2.         local data = redis.call("get", KEYS[2]) 
  3.         if data then 
  4.           
  5.           local dataJson = cjson.decode(data) 
  6.           local newNum = math.min(KEYS[3], math.floor(((dataJson["limitVal"] - 1) + (KEYS[3]/KEYS[5]) * (KEYS[4] - dataJson["limitTime"])))) 
  7.             
  8.           if newNum > 0 then 
  9.             
  10.             local paramsJson = cjson.encode({limitVal=newNum,limitTime=KEYS[4]}) 
  11.             redis.call("set", KEYS[2], paramsJson) 
  12.             return redis.call("get", KEYS[1]) 
  13.             
  14.           end 
  15.            return "wait" 
  16.         end 
  17.           
  18.         local paramsJson = cjson.encode({limitVal=KEYS[3],limitTime=KEYS[4]}) 
  19.         redis.call("set", KEYS[2], paramsJson) 
  20.         return redis.call("get", KEYS[1]) 
  21.       '; 
  22.         
  23.       // 1. lua脚本, 2 KEYS数量, 3 查找数据key, 4 限制key, 5 桶内数量, 6 时间戳, 7 过期时间 
  24.       Redis::eval(1,2,3,4,5,6,7参数);

Tags: lua+redis php计数器 php令牌桶

分享到: