青衣 发表于 2021-8-21 13:26:04

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

lua 优点
减少网络开销: 不使用 Lua 的代码需要向 Redis 发送多次请求, 而脚本只需一次即可, 减少网络传输;
原子操作: Redis 将整个脚本作为一个原子执行, 无需担心并发, 也就无需事务;
复用: 脚本会永久保存 Redis 中, 其他客户端可继续使用.
计数器模式:
利用lua脚本一次性完成处理达到原子性,通过INCR自增计数,判断是否达到限定值,达到限定值则返回限流,添加key过期时间应该范围过度


$lua = '
      local i = redis.call("INCR", KEYS)
      if i > 10 then
          return "wait"
      else
          if i == 1
          then
            redis.call("expire", KEYS, KEYS)
          end
          return redis.call("get", KEYS)
      end
      ';
laravel 请求代码:


Redis::eval($lua, 3, sprintf(RedisKey::API_LIMIT, $key, $callService['service']), 60, $cache_key);
令牌桶模式
每次请求在桶内拿取一个令牌,有令牌则通过,否则返回,并且按照算法一定的慢慢把令牌放入桶内


$lua = '
      local data = redis.call("get", KEYS)
      if data then
      
          local dataJson = cjson.decode(data)
          local newNum = math.min(KEYS, math.floor(((dataJson["limitVal"] - 1) + (KEYS/KEYS) * (KEYS - dataJson["limitTime"]))))
         
          if newNum > 0 then
         
            local paramsJson = cjson.encode({limitVal=newNum,limitTime=KEYS})
            redis.call("set", KEYS, paramsJson)
            return redis.call("get", KEYS)
         
          end
         return "wait"
      end
      
      local paramsJson = cjson.encode({limitVal=KEYS,limitTime=KEYS})
      redis.call("set", KEYS, paramsJson)
      return redis.call("get", KEYS)
      ';
      
      // 1. lua脚本, 2 KEYS数量, 3 查找数据key, 4 限制key, 5 桶内数量, 6 时间戳, 7 过期时间
      Redis::eval(1,2,3,4,5,6,7参数);
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持CodeAE代码之家。
原文链接:https://segmentfault.com/a/1190000018761106

文档来源:http://www.zzvips.com/article/179707.html
页: [1]
查看完整版本: php使用lua+redis实现限流,计数器模式,令牌桶模式