apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-06-07 07:55:16 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: distribute calling CLIs randomly

Browse source
This commit is contained in:
Abhishek Kumar 2026-03-19 21:18:18 +05:30
parent 522e69688a
commit c61a3843a5
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
api/services/campaign/rate_limiter.py
View file

@ -297,15 +297,19 @@ class RateLimiter:
redis.call('ZADD', key, 0, member)
end
-- Find an available number (score == 0)
local available = redis.call('ZRANGEBYSCORE', key, 0, 0, 'LIMIT', 0, 1)
-- Find all available numbers (score == 0)
local available = redis.call('ZRANGEBYSCORE', key, 0, 0)
if #available == 0 then
return nil
end
-- Pick a random number from the available pool for uniform distribution
local idx = math.random(#available)
local chosen = available[idx]
-- Mark as in-use with current timestamp
redis.call('ZADD', key, now, available[1])
return available[1]
redis.call('ZADD', key, now, chosen)
return chosen
"""
try: