fluxguard 0.1.1

package/scripts/sliding_window_counter.lua

@@ -0,0 +1,22 @@
+-- KEYS[1]=prev KEYS[2]=curr
+-- ARGV: limit, window_ms, now_ms
+local limit = tonumber(ARGV[1])
+local window = tonumber(ARGV[2])
+local now = tonumber(ARGV[3])
+local prev = tonumber(redis.call('GET', KEYS[1])) or 0
+local curr = tonumber(redis.call('GET', KEYS[2])) or 0
+local elapsed = now % window
+local weight = 1 - (elapsed / window)
+local estimated = math.floor(prev * weight) + curr
+if estimated >= limit then
+  local currStart = now - elapsed
+  local reset = currStart + window
+  return {0, estimated, limit, reset, reset - now}
+end
+redis.call('INCR', KEYS[2])
+redis.call('PEXPIRE', KEYS[2], window * 2)
+redis.call('PEXPIRE', KEYS[1], window * 2)
+local nextEst = estimated + 1
+local currStart = now - elapsed
+local reset = currStart + window
+return {1, limit - nextEst, limit, reset, 0}

package/scripts/sliding_window_log.lua

@@ -0,0 +1,20 @@
+-- KEYS[1] = zset key
+-- ARGV: limit, window_ms, now_ms, member unique id
+local limit = tonumber(ARGV[1])
+local window = tonumber(ARGV[2])
+local now = tonumber(ARGV[3])
+local member = ARGV[4]
+local cutoff = now - window
+redis.call('ZREMRANGEBYSCORE', KEYS[1], '-inf', cutoff)
+local count = redis.call('ZCARD', KEYS[1])
+if count >= limit then
+  local oldest = redis.call('ZRANGE', KEYS[1], 0, 0, 'WITHSCORES')
+  local ts = tonumber(oldest[2])
+  local reset = ts + window
+  return {0, 0, limit, reset, reset - now}
+end
+redis.call('ZADD', KEYS[1], now, member)
+redis.call('PEXPIRE', KEYS[1], window * 2)
+local newCount = count + 1
+local reset = now + window
+return {1, limit - newCount, limit, reset, 0}

package/scripts/token_bucket.lua

@@ -0,0 +1,26 @@
+-- KEYS[1]=tokens KEYS[2]=last_refill
+-- ARGV: limit(capacity), window_ms, now_ms
+local capacity = tonumber(ARGV[1])
+local window = tonumber(ARGV[2])
+local now = tonumber(ARGV[3])
+local refill = capacity / window
+local tokens = tonumber(redis.call('GET', KEYS[1]))
+local last = tonumber(redis.call('GET', KEYS[2]))
+if not tokens then tokens = capacity end
+if not last then last = now end
+local delta = math.max(0, now - last)
+tokens = math.min(capacity, tokens + delta * refill)
+if tokens < 1 then
+  local need = 1 - tokens
+  local retry = math.ceil(need / refill)
+  local full = math.ceil(capacity / refill)
+  return {0, 0, capacity, now + full, retry}
+end
+tokens = tokens - 1
+redis.call('SET', KEYS[1], tostring(tokens))
+redis.call('SET', KEYS[2], tostring(now))
+redis.call('PEXPIRE', KEYS[1], window * 3)
+redis.call('PEXPIRE', KEYS[2], window * 3)
+local remaining = math.floor(tokens)
+local full = math.ceil((capacity - tokens) / refill)
+return {1, remaining, capacity, now + full, 0}