throttle_machines 0.3.0-x86_64-linux → 0.3.1-x86_64-linux

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6930fcc4385b64ab5aaa9fd9ebbaa1b001fd8a55872e84eb70b94fbeb028f9bd
-  data.tar.gz: a3bdbd07e82ada0f96f3e402c0c4a9e59daeabc592f750dcca1e4e6af8d4bbc8
+  metadata.gz: c0e7b82b0a0e6aee13b28e5dff8d398583123ae16859094db1f5731563b6756a
+  data.tar.gz: 55eae369f9ff8d44e771ce1d30af98e81468c4e4590cb283121236dad134d726
 SHA512:
-  metadata.gz: 1a408d9c1aedf8b8ae9ec3aaa212c707dc1526907f401b8c1118e0b040eb4fc161cc17b205be3f098864c02364535103ccead629f58a9d3deef1fe32f240f1d3
-  data.tar.gz: 859411fc180acee7b70865e9f95f20b4de8fd14ba3f1e1c48b35d612ce2455152265c76335aa722b84ee0065aaf62ca95af029b7213a14a2bf591ae5b6f76797
+  metadata.gz: 9e2199d9db95a4b57fc42ed9d702a464e7efccc6a43312ef0e46e33b16cba5ec19bf6a1b684a8eaab2463d0ae89a4fcae063571a198a48b2d53a526f7a5730c5
+  data.tar.gz: b10d373aa0a749dddea669a8f50d1118c3101fb68b84a1411eed792457edae55e0f8cb0ecf43a389da389bc5884a9cead734df1a4a89687eb1027765ddc2bd56

data/lib/throttle_machines/storage/base.rb

@@ -57,8 +57,28 @@ module ThrottleMachines
         nil
       end
 
+      # Rounding slack for time comparisons, in units of the clock's precision.
+      # Matches the native core.
+      SLACK_ULPS = 64
+
       protected
 
+      # Times are running sums and differences of float seconds, so an
+      # on-schedule arrival can miss an exact boundary by a few ulps. The slack
+      # scales with the clock's magnitude.
+      def rounding_slack(time)
+        Float::EPSILON * SLACK_ULPS * [time.abs, 1.0].max
+      end
+
+      def gcra_allowed?(tat, now, delay_tolerance)
+        tat - now <= delay_tolerance + rounding_slack(tat)
+      end
+
+      # Whether a token is available, allowing for rounding in the refill.
+      def token_available?(tokens, now, refill_rate)
+        tokens >= 1 - (rounding_slack(now) * refill_rate)
+      end
+
       def current_time
         # Use monotonic time for consistency with BreakerMachines
         ThrottleMachines.monotonic_time

data/lib/throttle_machines/storage/memory.rb

@@ -86,7 +86,7 @@ module ThrottleMachines
           state = @gcra_states[key] || { tat: 0.0 }
 
           tat = [state[:tat], now].max
-          allow = tat - now <= delay_tolerance
+          allow = gcra_allowed?(tat, now, delay_tolerance)
 
           if allow
             new_tat = tat + emission_interval
@@ -107,7 +107,7 @@ module ThrottleMachines
           state = @gcra_states[key] || { tat: 0.0 }
 
           tat = [state[:tat], now].max
-          allow = tat - now <= delay_tolerance
+          allow = gcra_allowed?(tat, now, delay_tolerance)
 
           {
             allowed: allow,
@@ -130,8 +130,8 @@ module ThrottleMachines
           bucket[:last_refill] = now
 
           # Check if we can consume a token
-          if bucket[:tokens] >= 1
-            bucket[:tokens] -= 1
+          if token_available?(bucket[:tokens], now, refill_rate)
+            bucket[:tokens] = [bucket[:tokens] - 1, 0.0].max
             @token_buckets[key] = bucket.merge(expires_at: now + ttl)
 
             {
@@ -161,11 +161,11 @@ module ThrottleMachines
           tokens_to_add = elapsed * refill_rate
           current_tokens = [bucket[:tokens] + tokens_to_add, capacity].min
 
-          if current_tokens >= 1
+          if token_available?(current_tokens, now, refill_rate)
             {
               allowed: true,
               retry_after: 0,
-              tokens_remaining: (current_tokens - 1).floor
+              tokens_remaining: [current_tokens - 1, 0].max.floor
             }
           else
             retry_after = (1 - current_tokens) / refill_rate

data/lib/throttle_machines/storage/redis/gcra.lua

@@ -12,11 +12,15 @@ else
 end
 
 tat = math.max(tat, now)
-local allow = (tat - now) <= delay_tolerance
+-- The TAT is a running sum of emission intervals, so an on-schedule arrival
+-- can overshoot the tolerance by a few ulps; the slack scales with the clock.
+local slack = 2.220446049250313e-16 * 64 * math.max(math.abs(tat), 1)
+local allow = (tat - now) <= delay_tolerance + slack
 
 if allow then
   local new_tat = tat + emission_interval
   redis.call('SET', key, new_tat, 'EX', ttl)
 end
 
-return { allow and 1 or 0, tat }
+-- Redis truncates Lua numbers in replies to integers; send the TAT as text.
+return { allow and 1 or 0, string.format('%.17g', tat) }

data/lib/throttle_machines/storage/redis/peek_gcra.lua

@@ -11,6 +11,10 @@ else
 end
 
 tat = math.max(tat, now)
-local allow = (tat - now) <= delay_tolerance
+-- The TAT is a running sum of emission intervals, so an on-schedule arrival
+-- can overshoot the tolerance by a few ulps; the slack scales with the clock.
+local slack = 2.220446049250313e-16 * 64 * math.max(math.abs(tat), 1)
+local allow = (tat - now) <= delay_tolerance + slack
 
-return { allow and 1 or 0, tat }
+-- Redis truncates Lua numbers in replies to integers; send the TAT as text.
+return { allow and 1 or 0, string.format('%.17g', tat) }

data/lib/throttle_machines/storage/redis/peek_token_bucket.lua

@@ -12,7 +12,13 @@ local elapsed = now - last_refill
 local tokens_to_add = elapsed * refill_rate
 tokens = math.min(tokens + tokens_to_add, capacity)
 
-local allow = tokens >= 1
-local tokens_after = allow and (tokens - 1) or 0
+-- `elapsed` carries the clock's rounding; a token due exactly now may come
+-- out a hair short of 1. The slack scales with the clock, as in the native core.
+local slack = 2.220446049250313e-16 * 64 * math.max(math.abs(now), 1) * refill_rate
+local allow = tokens >= 1 - slack
+-- Admitted: tokens left after this request. Refused: the fractional balance,
+-- which retry_after is computed from.
+local tokens_after = allow and math.max(tokens - 1, 0) or tokens
 
-return { allow and 1 or 0, tokens_after }
+-- Redis truncates Lua numbers in replies to integers; send the count as text.
+return { allow and 1 or 0, string.format('%.17g', tokens_after) }

data/lib/throttle_machines/storage/redis/token_bucket.lua

@@ -13,11 +13,15 @@ local elapsed = now - last_refill
 local tokens_to_add = elapsed * refill_rate
 tokens = math.min(tokens + tokens_to_add, capacity)
 
-local allow = tokens >= 1
+-- `elapsed` carries the clock's rounding; a token due exactly now may come
+-- out a hair short of 1. The slack scales with the clock, as in the native core.
+local slack = 2.220446049250313e-16 * 64 * math.max(math.abs(now), 1) * refill_rate
+local allow = tokens >= 1 - slack
 if allow then
-  tokens = tokens - 1
+  tokens = math.max(tokens - 1, 0)
   redis.call('HMSET', key, 'tokens', tokens, 'last_refill', now)
   redis.call('EXPIRE', key, ttl)
 end
 
-return { allow and 1 or 0, tokens }
+-- Redis truncates Lua numbers in replies to integers; send the count as text.
+return { allow and 1 or 0, string.format('%.17g', tokens) }

data/lib/throttle_machines/storage/redis.rb

@@ -59,13 +59,13 @@ module ThrottleMachines
 
       # GCRA operations (atomic via Lua)
       def check_gcra_limit(key, emission_interval, delay_tolerance, ttl)
+        now = current_time
         result = run_script(GCRA_SCRIPT, GCRA_SHA,
                             keys: [prefixed(key)],
-                            argv: [emission_interval, delay_tolerance, ttl, current_time])
+                            argv: [emission_interval, delay_tolerance, ttl, now])
 
         allowed = result[0] == 1
-        tat = result[1]
-        now = current_time
+        tat = Float(result[1])
 
         {
           allowed: allowed,
@@ -80,25 +80,18 @@ module ThrottleMachines
                             keys: [prefixed(key)],
                             argv: [capacity, refill_rate, ttl, current_time])
 
-        allowed = result[0] == 1
-        tokens = result[1]
-
-        {
-          allowed: allowed,
-          retry_after: allowed ? 0 : (1 - tokens) / refill_rate,
-          tokens_remaining: tokens.floor
-        }
+        token_bucket_result(result, refill_rate)
       end
 
       # Peek methods for non-consuming checks
       def peek_gcra_limit(key, emission_interval, delay_tolerance)
+        now = current_time
         result = run_script(PEEK_GCRA_SCRIPT, PEEK_GCRA_SHA,
                             keys: [prefixed(key)],
-                            argv: [emission_interval, delay_tolerance, current_time])
+                            argv: [emission_interval, delay_tolerance, now])
 
         allowed = result[0] == 1
-        tat = result[1]
-        now = current_time
+        tat = Float(result[1])
 
         {
           allowed: allowed,
@@ -112,14 +105,7 @@ module ThrottleMachines
                             keys: [prefixed(key)],
                             argv: [capacity, refill_rate, current_time])
 
-        allowed = result[0] == 1
-        tokens_remaining = result[1]
-
-        {
-          allowed: allowed,
-          retry_after: allowed ? 0 : (1 - tokens_remaining) / refill_rate,
-          tokens_remaining: tokens_remaining.floor
-        }
+        token_bucket_result(result, refill_rate)
       end
 
       # No circuit breaker operations: breaker state is owned by BreakerMachines
@@ -147,6 +133,19 @@ module ThrottleMachines
 
       private
 
+      # `result` is [allowed, tokens]: the balance after an admission, or the
+      # fractional balance a refusal waits on. Matches Storage::Memory.
+      def token_bucket_result(result, refill_rate)
+        allowed = result[0] == 1
+        tokens = Float(result[1])
+
+        {
+          allowed: allowed,
+          retry_after: allowed ? 0 : (1 - tokens) / refill_rate,
+          tokens_remaining: allowed ? tokens.floor : 0
+        }
+      end
+
       def prefixed(key)
         "#{@prefix}#{key}"
       end

data/lib/throttle_machines/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: throttle_machines
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
 platform: x86_64-linux
 authors:
 - Abdelkader Boudih
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-10-12 00:00:00.000000000 Z
+date: 2026-10-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport