lib/logstasher/rails_ext/action_controller/metal/instrumentation.rb in logstasher-0.4.9 vs lib/logstasher/rails_ext/action_controller/metal/instrumentation.rb in logstasher-0.5.0

@@ -9,25 +9,27 @@
           :method     => request.method,
           :path       => (request.fullpath rescue "unknown")
       }
 
       LogStasher.add_default_fields_to_payload(raw_payload, request)
-      if self.respond_to?(:logtasher_add_custom_fields_to_payload)
-        before_keys = raw_payload.keys.clone
-        logtasher_add_custom_fields_to_payload(raw_payload)
-        after_keys = raw_payload.keys
-        # Store all extra keys added to payload hash in payload itself. This is a thread safe way
-        LogStasher.custom_fields += after_keys - before_keys
-      end
 
       ActiveSupport::Notifications.instrument("start_processing.action_controller", raw_payload.dup)
 
       ActiveSupport::Notifications.instrument("process_action.action_controller", raw_payload) do |payload|
         result = super
+        
+        if self.respond_to?(:logtasher_add_custom_fields_to_payload)
+          before_keys = raw_payload.keys.clone
+          logtasher_add_custom_fields_to_payload(raw_payload)
+          after_keys = raw_payload.keys
+          # Store all extra keys added to payload hash in payload itself. This is a thread safe way
+          LogStasher.custom_fields += after_keys - before_keys
+        end
+        
         payload[:status] = response.status
         append_info_to_payload(payload)
         result
       end
     end
 
   end
-end
\ No newline at end of file
+end