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

@@ -14,20 +14,23 @@
 
       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)
+        LogStasher.store.each do |key, value|
+          payload[key] = value
+        end
         result
       end
     end
 
   end