sjs/rack-attack
1
0
Fork 0
mirror of https://github.com/samsonjs/rack-attack.git synced 2026-04-27 15:07:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: make Throttle.new consistent with Blocklist/Safelist.new

Browse source
This commit is contained in:
Gonzalo Rodriguez 2019-03-01 22:12:12 -03:00
parent 3639afc196
commit 0e8dff4c88
No known key found for this signature in database
GPG key ID: 5DB8B81B049B8AB1
3 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
lib/rack/attack.rb
View file

@ -58,7 +58,7 @@ class Rack::Attack
end end
def throttle(name, options, &block) def throttle(name, options, &block)
throttles[name] = Throttle.new(name, options, block) throttles[name] = Throttle.new(name, options, &block)
end end
def track(name, options = {}, &block) def track(name, options = {}, &block)

2
lib/rack/attack/throttle.rb
View file

@ -6,7 +6,7 @@ module Rack
MANDATORY_OPTIONS = [:limit, :period].freeze MANDATORY_OPTIONS = [:limit, :period].freeze
attr_reader :name, :limit, :period, :block, :type attr_reader :name, :limit, :period, :block, :type
def initialize(name, options, block) def initialize(name, options, &block)
@name, @block = name, block @name, @block = name, block
MANDATORY_OPTIONS.each do |opt| MANDATORY_OPTIONS.each do |opt|
raise ArgumentError, "Must pass #{opt.inspect} option" unless options[opt] raise ArgumentError, "Must pass #{opt.inspect} option" unless options[opt]

2
lib/rack/attack/track.rb
View file

@ -9,7 +9,7 @@ module Rack
options[:type] = :track options[:type] = :track
if options[:limit] && options[:period] if options[:limit] && options[:period]
@filter = Throttle.new(name, options, block) @filter = Throttle.new(name, options, &block)
else else
@filter = Check.new(name, options, &block) @filter = Check.new(name, options, &block)
end end