Overwrite middleware to handle errors in logstash compatible format.

Author: Marc Grimme

lib/logstasher.rb

@@ -4,6 +4,7 @@
 require 'logstasher/active_record/log_subscriber'
 require 'logstasher/action_view/log_subscriber'
 require 'logstasher/rails_ext/action_controller/base'
+require 'logstasher/rails_ext/rack/debug_exceptions'
 require 'request_store'
 require 'active_support/core_ext/module/attribute_accessors'
 require 'active_support/core_ext/string/inflections'
@@ -137,6 +138,16 @@ def configured_to_suppress_app_logs?(config)
     !!(config.suppress_app_log.nil? ? config.supress_app_log : config.suppress_app_log)
   end
 
+  def modify_middleware(app)
+    if enabled
+      if configured_to_suppress_app_logs? app.config.logstasher
+        app.middleware.swap ::ActionDispatch::DebugExceptions, ::LogStasher::ActionDispatch::DebugExceptions
+      else
+        app.middleware.swap ::ActionDispatch::DebugExceptions, ::LogStasher::ActionDispatch::TwoWayDebugExceptions
+      end
+    end
+  end
+
   def custom_fields
     Thread.current[:logstasher_custom_fields] ||= []
   end

lib/logstasher/rails_ext/rack/debug_exceptions.rb

@@ -0,0 +1,35 @@
+module LogStasher
+  module ActionDispatch
+    def build_exception_hash(wrapper)
+      exception = wrapper.exception
+      trace = wrapper.application_trace
+      trace = wrapper.framework_trace if trace.empty?
+
+      { error:
+        ({ exception: exception.class, message: exception.message, trace: trace}.
+         merge!( exception.respond_to?(:annotated_source_code) && { annotated_source_code: exception.annoted_source_code } || {} ))
+      }
+    end
+
+    class DebugExceptions < ::ActionDispatch::DebugExceptions
+      include ::LogStasher::ActionDispatch
+
+      private
+      def log_error(env, wrapper)
+        LogStasher.logger << LogStasher.build_logstash_event(build_exception_hash(wrapper), ["exception"]).to_json + "\n"
+      end
+
+    end
+
+    class TwoWayDebugExceptions < ::ActionDispatch::DebugExceptions
+      include ::LogStasher::ActionDispatch
+
+      private
+      def log_error(env, wrapper)
+        LogStasher.logger << LogStasher.build_logstash_event(build_exception_hash(wrapper), ["exception"]).to_json + "\n"
+
+        super(env, wrapper)
+      end
+    end
+  end
+end

lib/logstasher/railtie.rb

@@ -21,7 +21,7 @@ class Railtie < Rails::Railtie
     # Load and ERB templating of YAML files
     LOGSTASHER = File.exists?(config_file) ? YAML.load(ERB.new(File.read(config_file)).result).symbolize_keys : nil
 
-    initializer :logstasher, :before => :load_config_initializers do |app|
+    initializer :logstasher, before: :load_config_initializers do |app|
       if LOGSTASHER.present?
         # process common configs
         LogStasher.process_config(app.config.logstasher, LOGSTASHER)
@@ -38,6 +38,10 @@ class Railtie < Rails::Railtie
         LogStasher.setup(config.logstasher) if config.logstasher.enabled
       end
     end
+
+    initializer 'logstasher.insert_middleware', after: :load_config_initializers do |app|
+      LogStasher.modify_middleware app
+    end
   end
 
   def process_config(config, yml_config)