mirror/kumomta
1
0
Fork 0
mirror of https://github.com/mailscope/kumomta.git synced 2026-10-30 10:06:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

throttle: another pass to try to figure out the integer/range issue

Browse Source
This commit is contained in:
Wez Furlong
2025-01-06 08:18:34 -07:00
parent 4064b4d6e5
commit aedf744edd
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/throttle/src/limit.rs
+4 -4
View File
@@ -13,13 +13,13 @@ static MEMORY: LazyLock<Mutex<MemoryStore>> = LazyLock::new(|| Mutex::new(Memory
static ACQUIRE_SCRIPT: LazyLock<Script> = LazyLock::new(|| {
Script::new(
r#"
local now_ts = tonumber(ARGV[1])
local expires_ts = tonumber(ARGV[2])
local now_ts = math.floor(tonumber(ARGV[1]))
local expires_ts = math.ceil(tonumber(ARGV[2]))
local limit = tonumber(ARGV[3])
local uuid = ARGV[4]
-- prune expired values
redis.call("ZREMRANGEBYSCORE", KEYS[1], "-inf", math.floor(now_ts-1))
redis.call("ZREMRANGEBYSCORE", KEYS[1], "-inf", now_ts-1)
local count = redis.call("ZCARD", KEYS[1])
if count + 1 > limit then
@@ -30,7 +30,7 @@ if count + 1 > limit then
if #smallest == 0 then
return 0
end
return math.floor(smallest[2] - now_ts)
return math.ceil(smallest[2] - now_ts)
end
redis.call("ZADD", KEYS[1], "NX", expires_ts, uuid)
return redis.status_reply('OK')