Merge pull request #10 from vipulnsward/mandatory_options

extract mandatory options constant
This commit is contained in:
Aaron Suggs 2013-05-03 07:41:02 -07:00
commit 22f051fa2d

View file

@ -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]