rapidity 0.0.7.362805 → 0.0.7.369741

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ec1c0e662f963921cf709fa7769897774ab2814b9c1c6b5017dc38379e25614a
-  data.tar.gz: 978b48d1149569859268054e0cbd8eeff7577c9b2fc5cceec2040a1361b974e1
+  metadata.gz: b9b41e5968e93c6bcb0d1a323f73890ea6f8ada6e47ff143a7c1fd631083e318
+  data.tar.gz: d6ec9a04b1c674362757670101dd82fa752c219ade4717a0f08d89cb97f811a2
 SHA512:
-  metadata.gz: ca19db773ea028961e6dd593faf29d27a0039b809e22b4be258cbbd1a237b5b14d306ef57928d53a5ffd7578e113e4701d933416bd1ed49ad86a29e65f9edd7b
-  data.tar.gz: 5d43836d52cda325188b2628e92037e54e44d58972c36ea2eefc024e64aa011191c16971326c01ad86b2eb7db84af6b8a8dc68e8a17374f5b901227202fec437
+  metadata.gz: 52c1cb448e7769412f83aa7636a1339d28b04288a3af798b3563ed8d1b2f0904077deb211b556b2660d32bda3f24971424df8ccf812f12f1032bb6d4e705234f
+  data.tar.gz: e8feb958fae4dbf853d669d75f1caa503aab0b2fb23a9df490c3deceea3b7865f17bdb44f89a7918a76f846018f03a76a8c44e0ff8a2d997e455beba4f4b352f

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rapidity (0.0.7.362805)
+    rapidity (0.0.7.369741)
       activesupport
       connection_pool
       ostruct

data/lib/rapidity/share/lua_scripts/acquire.lua

@@ -7,13 +7,13 @@
 -- данные между чтением и записью) и высокую производительность.
 --
 -- Описание функционала файла acquire.lua
--- 1. Атомарный захват по нескольким лимитам ("Всё или ничего"): 
+-- 1. Атомарный захват по нескольким лимитам ("Всё или ничего"):
 --   Скрипт принимает список ключей (лимитов) и запрошенное количество токенов.
 --   Он проверяет, есть ли нужное количество токенов во всех переданных лимитах одновременно.
 -- 2. Ленивое пополнение (Lazy Refill): Токены не пополняются каким-то фоновым процессом.
 --   Вместо этого при каждом обращении (в функции Limit:update) вычисляется разница во времени
 --   с момента последнего обращения (current_time - self.last_used) и добавляется количество токенов, пропорциональное скорости rate.
--- 3. Проверка условий (Validation): 
+-- 3. Проверка условий (Validation):
 --    - Если запрашивается <= 0 токенов или передан пустой список ключей — сразу возвращается ошибка.
 --    - Если хотя бы одного ключа не существует в Redis, возвращается key_not_found.
 --    - Если хотя бы в одном лимите не хватает токенов, скрипт возвращает ошибку not_limits и ничего не списывает из других лимитов (откат транзакции).
@@ -48,7 +48,7 @@ function Limit:new(limit_key)
   local obj = { key = limit_key, exists = false }
   setmetatable(obj, self)
 
-  -- Оптимизация: вместо EXISTS + HMGET делаем только HMGET. 
+  -- Оптимизация: вместо EXISTS + HMGET делаем только HMGET.
   -- Если ключа нет, values[1] будет nil.
   local values = redis.call("HMGET", limit_key,
     "max_tokens",
@@ -57,7 +57,7 @@ function Limit:new(limit_key)
     "last_used",
     "rate"
   )
-    
+
   if not values[1] then
     return obj -- Ключ не существует
   end
@@ -68,24 +68,28 @@ function Limit:new(limit_key)
   obj.last_used  = tonumber(values[4]) or 0
   obj.rate       = tonumber(values[5]) or 0
 
-  obj.exists = (obj.max_tokens > 0) and (obj.interval > 0)
-  
+  obj.exists     = (obj.max_tokens > 0) and (obj.interval > 0)
+
   return obj
 end
 
 -- "Ленивое" пополнение корзины токенов на основе прошедшего времени
 function Limit:update(current_time)
-  if not self.exists then return self end  
-  
+  if not self.exists then return self end
+
   local time_passed = current_time - self.last_used
   if time_passed <= 0 then return self end
-  
+
   local tokens_to_add = math.floor(time_passed * self.rate)
   self.tokens = math.min(self.tokens + tokens_to_add, self.max_tokens)
-  
+
   if tokens_to_add > 0 then
-    local time_consumed = tokens_to_add / self.rate
-    self.last_used = math.min(self.last_used + time_consumed, current_time)
+    if self.tokens == self.max_tokens then
+      self.last_used = current_time
+    else
+      local time_consumed = tokens_to_add / self.rate
+      self.last_used = math.min(self.last_used + time_consumed, current_time)
+    end
   end
 
   return self
@@ -94,10 +98,10 @@ end
 -- Сохранение обновленного стейта лимита в Redis
 function Limit:save()
   if not self.exists then return end
-  
+
   redis.call("HMSET", self.key,
-      "tokens", self.tokens, 
-      "last_used", self.last_used
+    "tokens", self.tokens,
+    "last_used", self.last_used
   )
 end
 
@@ -128,14 +132,14 @@ local function process_all_limits(limit_keys, requested, current_time)
   if #limit_keys == 0 then
     return { "result", "false", "retryable", "false", "error", "no keys passed" }
   end
-  
+
   local limits = {}
-  
+
   -- Фаза 1: Проверка всех лимитов (Read & Validate)
   for i = 1, #limit_keys do
     local key = limit_keys[i]
     local limit = Limit:new(key)
-    
+
     if not limit.exists then
       return { "result", "false", "retryable", "false", "error", "key_not_found", "key", key }
     end
@@ -153,22 +157,22 @@ local function process_all_limits(limit_keys, requested, current_time)
         "key", key
       }
     end
-    
+
     table.insert(limits, limit)
   end
-  
+
   -- Фаза 2: Применение изменений (Write)
   -- Выполняется только если всем лимитам хватило токенов
   for i = 1, #limits do
     local limit = limits[i]
     limit:acquire(requested)
     limit:save()
-    
-    -- Продлеваем жизнь ключу, чтобы он не удалился, если к нему активно обращаются. 
+
+    -- Продлеваем жизнь ключу, чтобы он не удалился, если к нему активно обращаются.
     -- "GT" обновляет TTL только если новый TTL больше текущего.
     redis.call("EXPIRE", limit.key, key_ttl, "GT")
   end
-  
+
   return { "result", "true", "keys", limit_keys }
 end
 

data/lib/rapidity/share/lua_scripts/available_in.lua

@@ -100,8 +100,9 @@ function Limit:available_in(needed)
   end
   
   local deficit_tokens = needed - self.tokens
+  local time_passed = current_time - self.last_used
   -- Округляем вверх (math.ceil), так как нам нужно дождаться появления ЦЕЛОГО токена
-  return math.ceil(deficit_tokens / self.rate)
+  return math.ceil(deficit_tokens / self.rate) - time_passed
 end
 
 -------------------------------------------------------------------------------

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rapidity
 version: !ruby/object:Gem::Version
-  version: 0.0.7.362805
+  version: 0.0.7.369741
 platform: ruby
 authors:
 - Yurusov Vlad
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-03-19 00:00:00.000000000 Z
+date: 2026-04-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport