Class: RosettAi::DBus::RateLimiter

Inherits:
Object
  • Object
show all
Defined in:
lib/rosett_ai/dbus/rate_limiter.rb

Overview

Token bucket rate limiter for D-Bus method calls.

Provides application-level rate limiting to prevent resource exhaustion from rapid D-Bus method invocations. Each method type can have its own rate limit.

Thread-safe: uses mutex for bucket state.

Examples:

limiter = RateLimiter.new(rate: 10, burst: 20)
if limiter.allow?
  # process request
else
  raise RateLimitExceeded
end

Defined Under Namespace

Classes: RateLimitExceeded

Constant Summary collapse

DEFAULT_RATE =

Default limits (requests per second / burst capacity)

10
DEFAULT_BURST =

Returns Default burst limit for rate limiting.

Returns:

  • (Integer) —

    Default burst limit for rate limiting.

20

Instance Method Summary collapse

Constructor Details

#initialize(rate: DEFAULT_RATE, burst: DEFAULT_BURST) ⇒ RateLimiter

Returns a new instance of RateLimiter.

Parameters:

  • rate (Numeric) (defaults to: DEFAULT_RATE) —

    tokens added per second

  • burst (Integer) (defaults to: DEFAULT_BURST) —

    maximum bucket capacity



39
40
41
42
43
44
45
 
# File 'lib/rosett_ai/dbus/rate_limiter.rb', line 39

def initialize(rate: DEFAULT_RATE, burst: DEFAULT_BURST)
  @rate = rate.to_f
  @burst = burst
  @tokens = burst.to_f
  @last_update = Process.clock_gettime(Process::CLOCK_MONOTONIC)
  @mutex = Mutex.new
end

Instance Method Details

#allow? ⇒ Boolean

Check if a request is allowed and consume a token if so.

Returns:

  • (Boolean) —

    true if request is allowed



50
51
52
53
54
55
56
57
58
59
60
 
# File 'lib/rosett_ai/dbus/rate_limiter.rb', line 50

def allow?
  @mutex.synchronize do
    refill_tokens
    if @tokens >= 1.0
      @tokens -= 1.0
      true
    else
      false
    end
  end
end

#available_tokens ⇒ Float

Current available tokens (for monitoring/debugging).

Returns:

  • (Float) —

    current token count



73
74
75
76
77
78
 
# File 'lib/rosett_ai/dbus/rate_limiter.rb', line 73

def available_tokens
  @mutex.synchronize do
    refill_tokens
    @tokens
  end
end

#consume!(method_name = nil) ⇒ Object

Consume a token or raise RateLimitExceeded.

Parameters:

  • method_name (String, nil) (defaults to: nil) —

    optional method name for error message

Raises:



66
67
68
 
# File 'lib/rosett_ai/dbus/rate_limiter.rb', line 66

def consume!(method_name = nil)
  raise RateLimitExceeded, method_name unless allow?
end