Add ability to use multiple instances of middleware

rack · Oct 17, 2019 · 5cd776a · 5cd776a
1 parent 6731e23
commit 5cd776a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/lib/rack/attack.rb b/lib/rack/attack.rb
@@ -84,9 +84,15 @@ def clear!
 
     attr_reader :configuration
 
-    def initialize(app)
+    def initialize(app, &block)
       @app = app
-      @configuration = self.class.configuration
+      @configuration =
+        if block_given?
+          configuration = Configuration.new
+          configuration.instance_exec(&block)
+        else
+          self.class.configuration
+        end
     end
 
     def call(env)