sorted set rate limiter

gistfile1.txt

local ts = tonumber(table.remove(ARGV))
local weight = tonumber(table.remove(ARGV))
local limit = tonumber(ARGV[1])

redis.call('zremrangebyscore', KEYS[1], 0, tonumber(ts) - 60)
local curr = redis.call('zrangebyscore', KEYS[1], tonumber(ts), tonumber(ts))
redis.call('zremrangebyscore', KEYS[1], tonumber(ts), tonumber(ts))
redis.call('zadd', KEYS[1], ts, tonumber(curr[1] or '0') + tonumber(weight))

local all = redis.call('zrange', KEYS[1], 0, -1)
local total = 0
for i,m in ipairs(all) do
  total = tonumber(total) + tonumber(m)
end

redis.call('expire', KEYS[1], 60)

return total > limit