mirror of
https://github.com/samsonjs/rack-attack.git
synced 2026-03-25 09:25:49 +00:00
87f628d0c1
For throttling, when the redis client throws an exception, the request ends up getting rate limited. Modify this to be similar to how ActiveSupport.MemCacheStore functions (the read, write and increment methods do not raise exceptions)
57 lines
1.5 KiB
Ruby
57 lines
1.5 KiB
Ruby
require 'delegate'
|
|
|
|
module Rack
|
|
module Attack
|
|
class StoreProxy
|
|
def self.build(store)
|
|
# RedisStore#increment needs different behavior, so detect that
|
|
# (method has an arity of 2; must call #expire separately
|
|
if defined?(::ActiveSupport::Cache::RedisStore) && store.is_a?(::ActiveSupport::Cache::RedisStore)
|
|
# ActiveSupport::Cache::RedisStore doesn't expose any way to set an expiry,
|
|
# so use the raw Redis::Store instead
|
|
store = store.instance_variable_get(:@data)
|
|
end
|
|
|
|
if defined?(::Redis::Store) && store.is_a?(::Redis::Store)
|
|
RedisStoreProxy.new(store)
|
|
else
|
|
store
|
|
end
|
|
end
|
|
|
|
class RedisStoreProxy < SimpleDelegator
|
|
def initialize(store)
|
|
super(store)
|
|
end
|
|
|
|
def read(key)
|
|
self.get(key)
|
|
rescue Redis::BaseError
|
|
nil
|
|
end
|
|
|
|
def write(key, value, options={})
|
|
if (expires_in = options[:expires_in])
|
|
self.setex(key, expires_in, value)
|
|
else
|
|
self.set(key, value)
|
|
end
|
|
rescue Redis::BaseError
|
|
nil
|
|
end
|
|
|
|
def increment(key, amount, options={})
|
|
count = nil
|
|
self.pipelined do
|
|
count = self.incrby(key, amount)
|
|
self.expire(key, options[:expires_in]) if options[:expires_in]
|
|
end
|
|
count.value if count
|
|
rescue Redis::BaseError
|
|
nil
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|