throttle_machines 0.1.1 → 0.2.4

data/lib/throttle_machines.rb

@@ -3,7 +3,7 @@
 require 'json'
 require 'timeout'
 require 'zeitwerk'
-require 'active_support/configurable'
+require 'active_support/core_ext/class/attribute'
 
 # Ecosystem dependencies
 require 'chrono_machines'
@@ -15,27 +15,53 @@ loader.ignore("#{__dir__}/throttle_machines/engine.rb") unless defined?(Rails::E
 loader.setup
 
 module ThrottleMachines
-  include ActiveSupport::Configurable
+  class Configuration
+    attr_accessor :default_limit, :default_period, :default_storage, :clock,
+                  :default_algorithm, :instrumentation_enabled, :instrumentation_backend,
+                  :_storage_instance
+
+    def initialize
+      @default_limit = 100
+      @default_period = 60 # 1 minute
+      @default_storage = :memory
+      @default_algorithm = :fixed_window
+      @clock = nil
+      @instrumentation_enabled = true
+      @instrumentation_backend = nil
+      @fiber_safe = false
+      @_storage_instance = nil
+    end
+
+    def fiber_safe
+      @fiber_safe
+    end
 
-  # Define configuration options with defaults
-  config_accessor :default_limit, default: 100
-  config_accessor :default_period, default: 60 # 1 minute
-  config_accessor :default_storage, default: :memory
-  config_accessor :clock, default: nil
-  config_accessor :instrumentation_enabled, default: true
-  config_accessor :instrumentation_backend
-  config_accessor :_storage_instance
+    def fiber_safe=(value)
+      @fiber_safe = !!value
+      ThrottleMachines.load_async_runtime! if @fiber_safe && ThrottleMachines.respond_to?(:load_async_runtime!)
+    end
+
+    alias async_enabled fiber_safe
+    alias async_enabled= fiber_safe=
+  end
+
+  @config = Configuration.new
 
   class << self
     # Delegate monotonic time to BreakerMachines for consistency
     delegate :monotonic_time, to: :BreakerMachines
 
+    def config
+      @config
+    end
+
     def configure
       yield(config) if block_given?
 
       # Apply instrumentation settings
       Instrumentation.enabled = config.instrumentation_enabled
       Instrumentation.backend = config.instrumentation_backend if config.instrumentation_backend
+      load_async_runtime! if config.fiber_safe
     end
 
     def storage
@@ -65,32 +91,52 @@ module ThrottleMachines
       control
     end
 
-    def limit(key, limit:, period:, algorithm: :fixed_window, &)
+    def limit(key, limit:, period:, algorithm: nil, &block)
       limiter = limiter(key, limit: limit, period: period, algorithm: algorithm)
-      limiter.throttle!(&)
+      limiter.throttle!(&block)
     end
 
-    def break_circuit(key, failures:, timeout:, &)
-      # Delegate to breaker_machines
+    def break_circuit(key, failures:, timeout:, &block)
+      # Delegate to BreakerMachines; use reset_timeout for open duration
       breaker = BreakerMachines::Circuit.new(
-        key: key,
+        key,
         failure_threshold: failures,
-        timeout: timeout
+        reset_timeout: timeout
       )
-      breaker.call(&)
+      breaker.call(&block)
     end
 
-    def retry_with(max_attempts: 3, backoff: :exponential, &)
+    def retry_with(max_attempts: 3, backoff: :exponential, &block)
       # Delegate to chrono_machines
       policy_options = {
         max_attempts: max_attempts,
         jitter_factor: backoff == :exponential ? 1.0 : 0.0
       }
-      ChronoMachines.retry(policy_options, &)
+      ChronoMachines.retry(policy_options, &block)
     end
 
-    def limiter(key, limit:, period:, algorithm: :fixed_window)
-      Limiter.new(key, limit: limit, period: period, algorithm: algorithm, storage: storage)
+    def limiter(key, limit:, period:, algorithm: nil)
+      Limiter.new(
+        key,
+        limit: limit,
+        period: period,
+        algorithm: algorithm || config.default_algorithm,
+        storage: storage
+      )
+    end
+
+    def async_breaker(...)
+      ThrottleMachines::AsyncBreaker.new(...)
+    end
+
+    def load_async_runtime!
+      require 'async'
+      require 'async/task'
+      true
+    rescue LoadError => e
+      raise unless e.message.include?('async')
+
+      raise LoadError, "The 'async' gem is required for fiber_safe mode. Add `gem 'async'` to your Gemfile."
     end
 
     private
@@ -131,4 +177,24 @@ module ThrottleMachines
 
   CircuitOpenError = BreakerMachines::CircuitOpenError
   RetryExhaustedError = ChronoMachines::MaxRetriesExceededError
+
+  # Back-compat wrapper: use BreakerMachines as the circuit implementation
+  Breaker = BreakerMachines::Circuit
+end
+
+# Optional speedups (order matters: most specific first)
+# These are loaded after Zeitwerk setup to allow graceful fallback.
+
+# Native Rust extension for performance-critical algorithms
+begin
+  require_relative "throttle_machines/native_speedup"
+rescue LoadError
+  # Native extension not available, pure Ruby will be used
+end
+
+# Async/Fiber support when Async gem is loaded
+begin
+  require_relative "throttle_machines/async_support"
+rescue LoadError
+  # Async gem not available
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: throttle_machines
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.4
 platform: ruby
 authors:
 - Abdelkader Boudih
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: 8.0.4
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '7.0'
+        version: 8.0.4
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
   requirement: !ruby/object:Gem::Requirement
@@ -71,28 +71,84 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.4'
+        version: '0.19'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.4'
+        version: '0.19'
 - !ruby/object:Gem::Dependency
   name: chrono_machines
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.2'
+        version: '0.9'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '5.16'
+- !ruby/object:Gem::Dependency
+  name: rb_sys
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.2'
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: rake-compiler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
 description: ThrottleMachines provides ultra-thin, elegant rate limiting with temporal
   precision for distributed systems.
 email:
@@ -103,9 +159,10 @@ extra_rdoc_files: []
 files:
 - LICENSE
 - README.md
-- Rakefile
 - lib/throttle_machines.rb
+- lib/throttle_machines/async_breaker.rb
 - lib/throttle_machines/async_limiter.rb
+- lib/throttle_machines/async_support.rb
 - lib/throttle_machines/control.rb
 - lib/throttle_machines/controller_helpers.rb
 - lib/throttle_machines/dependency_error.rb
@@ -115,11 +172,14 @@ files:
 - lib/throttle_machines/instrumentation.rb
 - lib/throttle_machines/limiter.rb
 - lib/throttle_machines/middleware.rb
+- lib/throttle_machines/native_speedup.rb
 - lib/throttle_machines/rack_middleware.rb
 - lib/throttle_machines/rack_middleware/allow2_ban.rb
+- lib/throttle_machines/rack_middleware/ban_filter.rb
 - lib/throttle_machines/rack_middleware/blocklist.rb
 - lib/throttle_machines/rack_middleware/configuration.rb
 - lib/throttle_machines/rack_middleware/fail2_ban.rb
+- lib/throttle_machines/rack_middleware/list_filter.rb
 - lib/throttle_machines/rack_middleware/request.rb
 - lib/throttle_machines/rack_middleware/safelist.rb
 - lib/throttle_machines/rack_middleware/throttle.rb
@@ -129,12 +189,9 @@ files:
 - lib/throttle_machines/storage/null.rb
 - lib/throttle_machines/storage/redis.rb
 - lib/throttle_machines/storage/redis/gcra.lua
-- lib/throttle_machines/storage/redis/get_breaker_state.lua
 - lib/throttle_machines/storage/redis/increment_counter.lua
 - lib/throttle_machines/storage/redis/peek_gcra.lua
 - lib/throttle_machines/storage/redis/peek_token_bucket.lua
-- lib/throttle_machines/storage/redis/record_breaker_failure.lua
-- lib/throttle_machines/storage/redis/record_breaker_success.lua
 - lib/throttle_machines/storage/redis/token_bucket.lua
 - lib/throttle_machines/throttled_error.rb
 - lib/throttle_machines/version.rb
@@ -146,6 +203,8 @@ metadata:
   source_code_uri: https://github.com/seuros/throttle_machines
   changelog_uri: https://github.com/seuros/throttle_machines/blob/main/CHANGELOG.md
   rubygems_mfa_required: 'true'
+  cargo_crate_name: throttle_machines_native
+  cargo_manifest_path: ext/throttle_machines_native/ffi/Cargo.toml
 rdoc_options: []
 require_paths:
 - lib
@@ -153,14 +212,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.3.0
+      version: 4.0.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 4.0.20
 specification_version: 4
 summary: Advanced Rate limiting for Ruby applications
 test_files: []

data/Rakefile

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-require 'bundler/gem_tasks'
-require 'rake/testtask'
-
-Rake::TestTask.new(:test) do |t|
-  t.libs << 'test'
-  t.test_files = FileList['test/**/*_test.rb'].exclude('test/dummy/**/*')
-  t.verbose = false
-end
-
-task default: :test

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

@@ -1,23 +0,0 @@
-local data = redis.call('HGETALL', KEYS[1])
-if #data == 0 then
-  return {}
-end
-
-local state = {}
-for i = 1, #data, 2 do
-  state[data[i]] = data[i + 1]
-end
-
--- Auto-transition from open to half-open if timeout passed
-if state['state'] == 'open' and state['opens_at'] then
-  local now = tonumber(ARGV[1])
-  local opens_at = tonumber(state['opens_at'])
-  
-  if now >= opens_at then
-    redis.call('HSET', KEYS[1], 'state', 'half_open', 'half_open_attempts', '0')
-    state['state'] = 'half_open'
-    state['half_open_attempts'] = '0'
-  end
-end
-
-return state

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

@@ -1,24 +0,0 @@
-local state = redis.call('HGET', KEYS[1], 'state') or 'closed'
-local now = ARGV[3]
-local timeout = tonumber(ARGV[2])
-
-if state == 'half_open' then
-  -- Failure in half-open state, just re-open the circuit
-  redis.call('HMSET', KEYS[1],
-    'state', 'open',
-    'opens_at', tonumber(now) + timeout,
-    'last_failure', now
-  )
-else -- state is 'closed' or nil
-  local failures = redis.call('HINCRBY', KEYS[1], 'failures', 1)
-  redis.call('HSET', KEYS[1], 'last_failure', now)
-  
-  if failures >= tonumber(ARGV[1]) then
-    redis.call('HMSET', KEYS[1],
-      'state', 'open',
-      'opens_at', tonumber(now) + timeout
-    )
-  end
-end
-
-redis.call('EXPIRE', KEYS[1], timeout * 2)

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

@@ -1,16 +0,0 @@
-local state = redis.call('HGET', KEYS[1], 'state')
-
-if state == 'half_open' then
-  -- Increment half-open attempts and potentially close the circuit
-  local attempts = redis.call('HINCRBY', KEYS[1], 'half_open_attempts', 1)
-  
-  if attempts >= tonumber(ARGV[1]) then
-    redis.call('DEL', KEYS[1])
-  end
-elseif state == 'closed' then
-  -- Reset failure count on success in closed state
-  local failures = redis.call('HGET', KEYS[1], 'failures')
-  if failures and tonumber(failures) > 0 then
-    redis.call('HSET', KEYS[1], 'failures', 0)
-  end
-end