rapidity 0.0.7.362805 → 0.0.8.373200

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ec1c0e662f963921cf709fa7769897774ab2814b9c1c6b5017dc38379e25614a
-  data.tar.gz: 978b48d1149569859268054e0cbd8eeff7577c9b2fc5cceec2040a1361b974e1
+  metadata.gz: d3def70e3c5647129987af0c2be0161ec2b848dadbe6cbad46f75b8fea3af67e
+  data.tar.gz: 01dd8c0f29563020a1b6f6d0aa0e89ff3d809dbe5694ca257b478b32a8005d12
 SHA512:
-  metadata.gz: ca19db773ea028961e6dd593faf29d27a0039b809e22b4be258cbbd1a237b5b14d306ef57928d53a5ffd7578e113e4701d933416bd1ed49ad86a29e65f9edd7b
-  data.tar.gz: 5d43836d52cda325188b2628e92037e54e44d58972c36ea2eefc024e64aa011191c16971326c01ad86b2eb7db84af6b8a8dc68e8a17374f5b901227202fec437
+  metadata.gz: db79ce100ae836e9c992629c23fdb6c163b79e282ec5fbb98bbd60a3709ed85fe0968e61f808f444feb26858fda71a3812e89c429d06215eec2fd050dea045ef
+  data.tar.gz: 337db88fdb29d7f6a7a0eb5af05cdf1944a962fab384fcdb8ec1a9968029c2fecbde0cbd186223b08b38d6d6e60e1a1371474c466111bf0b8c867f7cb2cace9a

data/Gemfile.lock

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

data/lib/rapidity/share/base.rb

@@ -8,12 +8,15 @@ module Rapidity
       BASE_SCRIPTS = [:list, :info, :reset, :delete]
       DEFAULT_KEY_TTL = 6000
 
+      MX = Monitor.new
+
       def initialize(pool, ttl: DEFAULT_KEY_TTL.to_i, logger: nil)
         @pool = pool
         @ttl = ttl
         @logger = logger || Logger.new(STDOUT)
         @logger.level = Logger::DEBUG
         # load_redis_scripts
+        restore_lua_hashes
       end
 
       # Returns a list of limits matching the pattern
@@ -154,15 +157,45 @@ module Rapidity
       #
       # @return [void]
       def load_redis_scripts
-        @pool.with do |conn|
-          (BASE_SCRIPTS + self.class::LUA_SCRIPTS).each do |script|
-            instance_variable_set("@lua_#{script}".to_sym,
-              conn.with {|r| r.script(:load, File.read(File.join(__dir__, 'lua_scripts', "#{script.to_s}.lua"))) }
-            )
+        cls = self.class
+        MX.synchronize do 
+          @pool.with do |conn|
+            list_lua_scripts.each do |script|
+              cls.instance_variable_set(lua_script_var(script),
+                conn.with {|r| r.script(:load, File.read(File.join(__dir__, 'lua_scripts', "#{script.to_s}.lua"))) }
+              )
+            end
           end
         end
+        restore_lua_hashes
+      end
+
+      # Restore instance script hashes from class cache
+      def restore_lua_hashes
+        cls = self.class
+        list_lua_scripts.each do |script|
+          instance_variable_set(lua_script_var(script), cls.instance_variable_get(lua_script_var(script)))
+        end
       end
 
+
+      # это для тестов чтобы проверить перезугрузку сриптов
+      def reset_lua_scripts
+        cls = self.class
+        list_lua_scripts.each do |script|
+          instance_variable_set(lua_script_var(script), nil)
+        end
+      end
+
+      def list_lua_scripts
+        (BASE_SCRIPTS + self.class::LUA_SCRIPTS)
+      end
+
+      def lua_script_var(script)
+        "@lua_#{script}".to_sym
+      end
+
+
     end
   end
 end

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
 
 -------------------------------------------------------------------------------

data/lib/rapidity/version.rb

@@ -1,6 +1,6 @@
 module Rapidity
 
-  VERSION = '0.0.7'.freeze
+  VERSION = '0.0.8'.freeze
 
 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.8.373200
 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-05-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport