sjs/rack-attack
1
0
Fork 0
mirror of https://github.com/samsonjs/rack-attack.git synced 2026-03-25 09:25:49 +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
def throttle(name, options, &block)
throttles[name] = Throttle.new(name, options, block)
throttles[name] = Throttle.new(name, options, &block)
end
def track(name, options = {}, &block)

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

@ -6,7 +6,7 @@ module Rack
MANDATORY_OPTIONS = [:limit, :period].freeze
attr_reader :name, :limit, :period, :block, :type
def initialize(name, options, block)
def initialize(name, options, &block)
@name, @block = name, block
MANDATORY_OPTIONS.each do |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
if options[:limit] && options[:period]
@filter = Throttle.new(name, options, block)
@filter = Throttle.new(name, options, &block)
else
@filter = Check.new(name, options, &block)
end