lib/rate_limiter.rb in rate_limiter-0.0.4 vs lib/rate_limiter.rb in rate_limiter-0.0.5

- old
+ new

@@ -1,7 +1,5 @@ -require 'singleton' - require 'rate_limiter/config' require 'rate_limiter/controller' require 'rate_limiter/model' module RateLimiter @@ -12,15 +10,15 @@ def self.enabled? !!RateLimiter.config.enabled end def self.enabled_for_controller=(value) - RateLimiter.store[:request_enabled_for_controller] = value + rate_limiter_store[:request_enabled_for_controller] = value end def self.enabled_for_controller? - !!RateLimiter.store[:request_enabled_for_controller] + !!rate_limiter_store[:request_enabled_for_controller] end def self.timestamp_field=(field_name) RateLimiter.config.timestamp_field = field_name end @@ -28,28 +26,28 @@ def self.timestamp_field RateLimiter.config.timestamp_field end def self.source=(value) - RateLimiter.store[:source] = value + rate_limiter_store[:source] = value end def self.source - RateLimiter.store[:source] + rate_limiter_store[:source] end def self.controller_info=(value) - RateLimiter.store[:controller_info] = value + rate_limiter_store[:controller_info] = value end def self.controller_info - RateLimiter.store[:controller_info] + rate_limiter_store[:controller_info] end private - def self.store - Thread.current[:rate_limiter] || { :request_enabled_for_controller => true } + def self.rate_limiter_store + Thread.current[:rate_limiter] ||= { :request_enabled_for_controller => true } end def self.config @@config ||= RateLimiter::Config.instance end