mirror of
https://github.com/samsonjs/rack-attack.git
synced 2026-04-27 15:07:41 +00:00
Merge pull request #10 from vipulnsward/mandatory_options
extract mandatory options constant
This commit is contained in:
commit
22f051fa2d
1 changed files with 2 additions and 1 deletions
|
|
@ -1,10 +1,11 @@
|
||||||
module Rack
|
module Rack
|
||||||
module Attack
|
module Attack
|
||||||
class Throttle
|
class Throttle
|
||||||
|
MANDATORY_OPTIONS = [:limit, :period]
|
||||||
attr_reader :name, :limit, :period, :block
|
attr_reader :name, :limit, :period, :block
|
||||||
def initialize(name, options, block)
|
def initialize(name, options, block)
|
||||||
@name, @block = name, block
|
@name, @block = name, block
|
||||||
[:limit, :period].each do |opt|
|
MANDATORY_OPTIONS.each do |opt|
|
||||||
raise ArgumentError.new("Must pass #{opt.inspect} option") unless options[opt]
|
raise ArgumentError.new("Must pass #{opt.inspect} option") unless options[opt]
|
||||||
end
|
end
|
||||||
@limit = options[:limit]
|
@limit = options[:limit]
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue