欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-03-20 14:47:56
lua 优点 减少网络开销: 不使用 lua 的代码需要向 redis 发送多次请求, 而脚本只需一次即可, 减少网络传输; 原子操作: redis 将整个脚本...

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参数);

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。