redis.replicate_commands()
-- 記錄桶的剩余容量
local tokens_key = KEYS[1]
-- 記錄桶上次刷新時(shí)間，以此推算當(dāng)前需要填入的令牌數(shù)量
-- 第一次：需要新填入的令牌數(shù)量 = (當(dāng)前時(shí)間 - 0) * 速率
-- 其他后：需要新填入的令牌數(shù)量 = (當(dāng)前時(shí)間 - Key[2]) * 速率
local timestamp_key = KEYS[2]
-- 綜上：**當(dāng)前桶內(nèi)可用令牌數(shù) = 桶的剩余容量 + 需要新填入的令牌數(shù)量**
redis.log(redis.LOG_WARNING, "tokens_key " .. tokens_key)

local rate = tonumber(ARGV[1])
local capacity = tonumber(ARGV[2])
local now = redis.call('TIME')[1]
local requested = tonumber(ARGV[4])

local fill_time = capacity/rate
-- redis key[1]、key[2] 的過(guò)期時(shí)間
-- 令牌過(guò)期時(shí)間：填充時(shí)間*2
-- 返回小于參數(shù)x的最大整數(shù)
-- 這樣可以保證令牌桶中的令牌能夠被充分利用，并避免過(guò)早的過(guò)期。
-- 例如，如果填充時(shí)間的值為 10 秒，那么過(guò)期時(shí)間的值就應(yīng)該設(shè)置為 20 秒。這樣，在令牌桶的生存周期內(nèi)，用戶就有足夠的時(shí)間來(lái)使用令牌桶中的令牌。
local ttl = math.floor(fill_time*2)

redis.log(redis.LOG_WARNING, "rate " .. ARGV[1])
redis.log(redis.LOG_WARNING, "capacity " .. ARGV[2])
redis.log(redis.LOG_WARNING, "now " .. now)
redis.log(redis.LOG_WARNING, "requested " .. ARGV[4])
redis.log(redis.LOG_WARNING, "filltime " .. fill_time)
redis.log(redis.LOG_WARNING, "ttl " .. ttl)

local last_tokens = tonumber(redis.call("get", tokens_key))
if last_tokens == nil then
   last_tokens = capacity
end
redis.log(redis.LOG_WARNING, "last_tokens " .. last_tokens)

local last_refreshed = tonumber(redis.call("get", timestamp_key))
if last_refreshed == nil then
   last_refreshed = 0
end
redis.log(redis.LOG_WARNING, "last_refreshed " .. last_refreshed)

local delta = math.max(0, now-last_refreshed)
local filled_tokens = math.min(capacity, last_tokens+(delta*rate))
local allowed = filled_tokens >= requested
local new_tokens = filled_tokens
local allowed_num = 0
if allowed then
   new_tokens = filled_tokens - requested
   allowed_num = 1
end

--redis.log(redis.LOG_WARNING, "delta " .. delta)
--redis.log(redis.LOG_WARNING, "filled_tokens " .. filled_tokens)
--redis.log(redis.LOG_WARNING, "allowed_num " .. allowed_num)
--redis.log(redis.LOG_WARNING, "new_tokens " .. new_tokens)

if ttl > 0 then
   redis.call("setex", tokens_key, ttl, new_tokens)
   redis.call("setex", timestamp_key, ttl, now)
end

-- return { allowed_num, new_tokens, capacity, filled_tokens, requested, new_tokens }
return { allowed_num, new_tokens }