insert airbrake rails middleware after better errors middleware if be…

…tter errors middleware defined
airbrake · Jun 12, 2015 · 5d66e27 · 5d66e27
1 parent 6b444d9
commit 5d66e27
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/lib/airbrake/railtie.rb b/lib/airbrake/railtie.rb
@@ -12,13 +12,16 @@ class Railtie < ::Rails::Railtie
 
     initializer "airbrake.middleware" do |app|
 
-      middleware = if defined?(ActionDispatch::DebugExceptions)
-        # Rails >= 3.2.0
-        "ActionDispatch::DebugExceptions"
-      else
-        # Rails < 3.2.0
-        "ActionDispatch::ShowExceptions"
-      end
+      middleware =
+        if !::Rails.env.production? && defined?(::BetterErrors::Middleware)
+          "BetterErrors::Middleware"
+        elsif defined?(::ActionDispatch::DebugExceptions)
+          # Rails >= 3.2.0
+          "ActionDispatch::DebugExceptions"
+        else
+          # Rails < 3.2.0
+          "ActionDispatch::ShowExceptions"
+        end
 
       app.config.middleware.insert_after middleware,
         "Airbrake::Rails::Middleware"