Redisson限流器(RateLimiter)

尝试获取令牌lua脚本解析

return commandExecutor.evalWriteAsync(getName(), LongCodec.INSTANCE, command,
    // 获取缓存中的rate(令牌)
    "local rate = redis.call('hget', KEYS[1], 'rate');"
    // 获取缓存中的interval(时间间隔)
  + "local interval = redis.call('hget', KEYS[1], 'interval');"
  // 获取缓存中的type
  + "local type = redis.call('hget', KEYS[1], 'type');"
  + "assert(rate ~= false and interval ~= false and type ~= false, 'RateLimiter is not initialized')"
  
  // {lotus:sale:dashboard:exportRateLimiter}:value
  + "local valueName = KEYS[2];"
  // {lotus:sale:dashboard:exportRateLimiter}:permits
  + "local permitsName = KEYS[4];"
  // 如果type=1,要加上对应客户端的后缀
  + "if type == '1' then "
      + "valueName = KEYS[3];"
      + "permitsName = KEYS[5];"
  + "end;"

  // 如果缓存中的令牌数量>=用户要获取的令牌数量,才可以继续
  + "assert(tonumber(rate) >= tonumber(ARGV[1]), 'Requested permits amount could not exceed defined rate'); "

  // 获取当前已经有的令牌数量
  + "local currentValue = redis.call('get', valueName); "
  // 当前已经有的令牌数量存在
  + "if currentValue ~= false then "
         // 从0到当前时间-时间范围 分支下的 数量(过期的数据)
         + "local expiredValues = redis.call('zrangebyscore', permitsName, 0, tonumber(ARGV[2]) - interval); "
         + "local released = 0; "
         + "for i, v in ipairs(expiredValues) do "
              // 拆封,获取 随机数 和 需要归还的数量
              + "local random, permits = struct.unpack('fI', v);"
              + "released = released + permits;"
         + "end; "

        // 说明有已经过期的数据,需要删除
         + "if released > 0 then "
              + "redis.call('zremrangebyscore', permitsName, 0, tonumber(ARGV[2]) - interval); "
              // 增加当前可用的令牌数量
              + "currentValue = tonumber(currentValue) + released; "
              + "redis.call('set', valueName, currentValue);"
         + "end;"

         // 当前可用的令牌数量 < 要获取的数量
         + "if tonumber(currentValue) < tonumber(ARGV[1]) then "
             // 获取最近的要过期的信息
             + "local nearest = redis.call('zrangebyscore', permitsName, '(' .. (tonumber(ARGV[2]) - interval), '+inf', 'withscores', 'limit', 0, 1); "
             + "local random, permits = struct.unpack('fI', nearest[1]);"
             // 返回多久后再来重试
             + "return tonumber(nearest[2]) - (tonumber(ARGV[2]) - interval);"
         // 当前可用的令牌数量 >= 要获取的数量
         + "else "
             // 加入到 permits 获取成功的
             + "redis.call('zadd', permitsName, ARGV[2], struct.pack('fI', ARGV[3], ARGV[1])); "
             // 减少可用的令牌数
             + "redis.call('decrby', valueName, ARGV[1]); "
             + "return nil; "
         + "end; "
    // 当前已经有的令牌数量不存在
  + "else "
         + "redis.call('set', valueName, rate); "
         // struct.pack('fI', ARGV[3], ARGV[1]) 对参数进行包装
         + "redis.call('zadd', permitsName, ARGV[2], struct.pack('fI', ARGV[3], ARGV[1])); "
         + "redis.call('decrby', valueName, ARGV[1]); "
         + "return nil; "
  + "end;",
// 参数解析
// key {key}:value {key}:value:clientId {key}:permits {key}:permits:clientId
// 1 当前时间戳 随机数
Arrays.asList(getName(), getValueName(), getClientValueName(), getPermitsName(), getClientPermitsName()),
                value, System.currentTimeMillis(), ThreadLocalRandom.current().nextLong());

令牌桶配置(每分钟产生一个令牌)

https://cdn.jsdelivr.net/gh/beastmouth/blog-img/限流器配置.png

令牌桶占用情况

分值:当时的时间

元素:随机数+申请的数量,二进制内容

https://cdn.jsdelivr.net/gh/beastmouth/blog-img/令牌占用情况.png

剩余可用的令牌数

https://cdn.jsdelivr.net/gh/beastmouth/blog-img/当前可用令牌.png

你可能感兴趣的:(Redisson限流器(RateLimiter))