Create separate middleware stack for initializer

2 files changed, 3 insertions, 3 deletions

diff --git a/railties/lib/rails/configuration.rb b/railties/lib/rails/configuration.rb
index f6159b9347..4a70a4800e 100644
--- a/railties/lib/rails/configuration.rb
+++ b/railties/lib/rails/configuration.rb
@@ -108,10 +108,9 @@ module Rails
       defined?(::RAILS_FRAMEWORK_ROOT) ? ::RAILS_FRAMEWORK_ROOT : "#{root_path}/vendor/rails"
     end
 
-    # TODO: Fix this when there is an application object
     def middleware
-      require 'action_controller'
-      ActionController::Dispatcher.middleware
+      require 'action_dispatch'
+      @middleware ||= ActionDispatch::MiddlewareStack.new
     end
 
     # Loads and returns the contents of the #database_configuration_file. The
diff --git a/railties/lib/rails/initializer.rb b/railties/lib/rails/initializer.rb
index d05cb69ffe..399ce26618 100644
--- a/railties/lib/rails/initializer.rb
+++ b/railties/lib/rails/initializer.rb
@@ -608,6 +608,7 @@ Run `rake gems:install` to install the missing gems.
 
   Initializer.default.add :build_application do
     if configuration.frameworks.include?(:action_controller)
+      ActionController::Dispatcher.middleware = configuration.middleware
       Rails.application = Rails::Application.new
     end
   end