throttle_machines 0.2.4-x86_64-linux → 0.3.0-x86_64-linux

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b903b7c1264b244a780a3730bd09d7bdab11db1f28961910980d93d7722ff92e
-  data.tar.gz: d6229cb0f861acea1f76e99a8ec4bcd0c3217f10745467f6411c5383519fe84b
+  metadata.gz: 6930fcc4385b64ab5aaa9fd9ebbaa1b001fd8a55872e84eb70b94fbeb028f9bd
+  data.tar.gz: a3bdbd07e82ada0f96f3e402c0c4a9e59daeabc592f750dcca1e4e6af8d4bbc8
 SHA512:
-  metadata.gz: f35c2f3f16752f64e34105a201f5854b157a0f5092972acb47310802400502474d3d74ca52852056b77932c442af3d76ab0edec0becdd0142a07d4c17a11159e
-  data.tar.gz: 7c20314309fda5045cc7109b03b79dcdac901cb1ce9e83011cf59f6f1906c24f3fd9e6db7b86aff3091deebd21fdbd02118db08d9bce43d7a3f5be6c06111d0c
+  metadata.gz: 1a408d9c1aedf8b8ae9ec3aaa212c707dc1526907f401b8c1118e0b040eb4fc161cc17b205be3f098864c02364535103ccead629f58a9d3deef1fe32f240f1d3
+  data.tar.gz: 859411fc180acee7b70865e9f95f20b4de8fd14ba3f1e1c48b35d612ce2455152265c76335aa722b84ee0065aaf62ca95af029b7213a14a2bf591ae5b6f76797

data/lib/throttle_machines/storage/redis.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require 'digest/sha1'
+
 module ThrottleMachines
   module Storage
     class Redis < Base
@@ -13,17 +15,18 @@ module ThrottleMachines
       INCREMENT_COUNTER_SCRIPT = File.read(File.join(LUA_SCRIPTS_DIR, 'increment_counter.lua'))
       # Breaker scripts removed: breaker state is owned by BreakerMachines
 
+      # Redis caches scripts by the SHA1 of their source, so the SHAs are known
+      # without a SCRIPT LOAD round trip.
+      GCRA_SHA = Digest::SHA1.hexdigest(GCRA_SCRIPT)
+      TOKEN_BUCKET_SHA = Digest::SHA1.hexdigest(TOKEN_BUCKET_SCRIPT)
+      PEEK_GCRA_SHA = Digest::SHA1.hexdigest(PEEK_GCRA_SCRIPT)
+      PEEK_TOKEN_BUCKET_SHA = Digest::SHA1.hexdigest(PEEK_TOKEN_BUCKET_SCRIPT)
+
       def initialize(options = {})
         super
         @redis = options[:redis] || options[:client] || options[:pool]
         @prefix = options[:prefix] || 'throttle:'
 
-        # Cache scripts to avoid repeated script loads
-        @gcra_sha = nil
-        @token_bucket_sha = nil
-        @peek_gcra_sha = nil
-        @peek_token_bucket_sha = nil
-
         # Validate Redis connection
         validate_redis_connection!
       end
@@ -56,15 +59,9 @@ module ThrottleMachines
 
       # GCRA operations (atomic via Lua)
       def check_gcra_limit(key, emission_interval, delay_tolerance, ttl)
-        ensure_gcra_script_loaded!
-
-        result = with_redis do |redis|
-          redis.evalsha(
-            @gcra_sha,
-            keys: [prefixed(key)],
-            argv: [emission_interval, delay_tolerance, ttl, current_time]
-          )
-        end
+        result = run_script(GCRA_SCRIPT, GCRA_SHA,
+                            keys: [prefixed(key)],
+                            argv: [emission_interval, delay_tolerance, ttl, current_time])
 
         allowed = result[0] == 1
         tat = result[1]
@@ -75,24 +72,13 @@ module ThrottleMachines
           retry_after: allowed ? 0 : (tat - now - delay_tolerance),
           tat: tat
         }
-      rescue ::Redis::CommandError => e
-        raise unless e.message.include?('NOSCRIPT')
-
-        @gcra_sha = nil
-        retry
       end
 
       # Token bucket operations (atomic via Lua)
       def check_token_bucket(key, capacity, refill_rate, ttl)
-        ensure_token_bucket_script_loaded!
-
-        result = with_redis do |redis|
-          redis.evalsha(
-            @token_bucket_sha,
-            keys: [prefixed(key)],
-            argv: [capacity, refill_rate, ttl, current_time]
-          )
-        end
+        result = run_script(TOKEN_BUCKET_SCRIPT, TOKEN_BUCKET_SHA,
+                            keys: [prefixed(key)],
+                            argv: [capacity, refill_rate, ttl, current_time])
 
         allowed = result[0] == 1
         tokens = result[1]
@@ -102,24 +88,13 @@ module ThrottleMachines
           retry_after: allowed ? 0 : (1 - tokens) / refill_rate,
           tokens_remaining: tokens.floor
         }
-      rescue ::Redis::CommandError => e
-        raise unless e.message.include?('NOSCRIPT')
-
-        @token_bucket_sha = nil
-        retry
       end
 
       # Peek methods for non-consuming checks
       def peek_gcra_limit(key, emission_interval, delay_tolerance)
-        ensure_peek_gcra_script_loaded!
-
-        result = with_redis do |redis|
-          redis.evalsha(
-            @peek_gcra_sha,
-            keys: [prefixed(key)],
-            argv: [emission_interval, delay_tolerance, current_time]
-          )
-        end
+        result = run_script(PEEK_GCRA_SCRIPT, PEEK_GCRA_SHA,
+                            keys: [prefixed(key)],
+                            argv: [emission_interval, delay_tolerance, current_time])
 
         allowed = result[0] == 1
         tat = result[1]
@@ -130,23 +105,12 @@ module ThrottleMachines
           retry_after: allowed ? 0 : (tat - now - delay_tolerance),
           tat: tat
         }
-      rescue ::Redis::CommandError => e
-        raise unless e.message.include?('NOSCRIPT')
-
-        @peek_gcra_sha = nil
-        retry
       end
 
       def peek_token_bucket(key, capacity, refill_rate)
-        ensure_peek_token_bucket_script_loaded!
-
-        result = with_redis do |redis|
-          redis.evalsha(
-            @peek_token_bucket_sha,
-            keys: [prefixed(key)],
-            argv: [capacity, refill_rate, current_time]
-          )
-        end
+        result = run_script(PEEK_TOKEN_BUCKET_SCRIPT, PEEK_TOKEN_BUCKET_SHA,
+                            keys: [prefixed(key)],
+                            argv: [capacity, refill_rate, current_time])
 
         allowed = result[0] == 1
         tokens_remaining = result[1]
@@ -156,11 +120,6 @@ module ThrottleMachines
           retry_after: allowed ? 0 : (1 - tokens_remaining) / refill_rate,
           tokens_remaining: tokens_remaining.floor
         }
-      rescue ::Redis::CommandError => e
-        raise unless e.message.include?('NOSCRIPT')
-
-        @peek_token_bucket_sha = nil
-        retry
       end
 
       # No circuit breaker operations: breaker state is owned by BreakerMachines
@@ -192,20 +151,17 @@ module ThrottleMachines
         "#{@prefix}#{key}"
       end
 
-      def ensure_gcra_script_loaded!
-        @ensure_gcra_script_loaded ||= with_redis { |r| r.script(:load, GCRA_SCRIPT) }
-      end
-
-      def ensure_token_bucket_script_loaded!
-        @ensure_token_bucket_script_loaded ||= with_redis { |r| r.script(:load, TOKEN_BUCKET_SCRIPT) }
-      end
-
-      def ensure_peek_gcra_script_loaded!
-        @ensure_peek_gcra_script_loaded ||= with_redis { |r| r.script(:load, PEEK_GCRA_SCRIPT) }
-      end
+      # EVALSHA by the precomputed SHA. A server without the script cached (a
+      # restart, SCRIPT FLUSH, failover) answers NOSCRIPT; EVAL then runs the
+      # source once and caches it for later calls. Any other error propagates.
+      def run_script(script, sha, keys:, argv:)
+        with_redis do |redis|
+          redis.evalsha(sha, keys: keys, argv: argv)
+        rescue ::Redis::CommandError => e
+          raise unless e.message.include?('NOSCRIPT')
 
-      def ensure_peek_token_bucket_script_loaded!
-        @ensure_peek_token_bucket_script_loaded ||= with_redis { |r| r.script(:load, PEEK_TOKEN_BUCKET_SCRIPT) }
+          redis.eval(script, keys: keys, argv: argv)
+        end
       end
 
       def validate_redis_connection!

data/lib/throttle_machines/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ThrottleMachines
-  VERSION = '0.2.4'
+  VERSION = '0.3.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: throttle_machines
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.3.0
 platform: x86_64-linux
 authors:
 - Abdelkader Boudih
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-10-10 00:00:00.000000000 Z
+date: 2026-10-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 8.0.4
+        version: '8.1'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 8.0.4
+        version: '8.1'
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
   requirement: !ruby/object:Gem::Requirement
@@ -72,42 +72,42 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.19'
+        version: '0.20'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.19'
+        version: '0.20'
 - !ruby/object:Gem::Dependency
   name: chrono_machines
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.9'
+        version: '0.10'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.9'
+        version: '0.10'
 - !ruby/object:Gem::Dependency
   name: minitest
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '5.16'
+        version: '6.0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '5.16'
+        version: '6.0'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement