lib/ahoy/engine.rb in ahoy_matey-1.2.2 vs lib/ahoy/engine.rb in ahoy_matey-1.3.0

@@ -1,7 +1,12 @@
 module Ahoy
   class Engine < ::Rails::Engine
-    initializer "ahoy.middleware", after: "sprockets.environment" do
+    initializer "ahoy.middleware", after: "sprockets.environment" do |app|
       Rails::Rack::Logger.send(:prepend, Ahoy::LogSilencer) if Ahoy.quiet
+
+      if Ahoy.throttle
+        require "ahoy/throttle"
+        app.middleware.use Ahoy::Throttle
+      end
     end
   end
 end