尝试获取令牌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());
令牌桶配置(每分钟产生一个令牌)
令牌桶占用情况
分值:当时的时间
元素:随机数+申请的数量,二进制内容
剩余可用的令牌数