Module: ActionController::Instrumentation

Defined in:
lib/govuk_app_config/govuk_json_logging/rails_ext/action_controller/metal/instrumentation.rb

Instance Method Summary collapse

Instance Method Details

#orig_process_action ⇒ Object 



7
 
# File 'lib/govuk_app_config/govuk_json_logging/rails_ext/action_controller/metal/instrumentation.rb', line 7

alias_method "orig_process_action", "process_action"

#process_action(*args) ⇒ Object Also known as: logstasher_process_action 



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/govuk_app_config/govuk_json_logging/rails_ext/action_controller/metal/instrumentation.rb', line 9

def process_action(*args)
  # The raw payload has been updated to reflect the payload structure used
  # in Rails 7.1, primarily the addition of the `headers`, `request` keys
  # and using `request.filtered_path` instead of `request.fullpath`.
  # https://github.com/rails/rails/blame/d39db5d1891f7509cde2efc425c9d69bbb77e670/actionpack/lib/action_controller/metal/instrumentation.rb#L60
  raw_payload = {
    controller: self.class.name,
    action: action_name,
    request:,
    params: request.filtered_parameters,
    headers: request.headers,
    format: request.format.ref,
    method: request.request_method,
    path: begin
      request.filtered_path
    rescue StandardError
      "unknown"
    end,
  }

  LogStasher.add_default_fields_to_payload(raw_payload, request)

  LogStasher.clear_request_context
  LogStasher.add_default_fields_to_request_context(request)

  ActiveSupport::Notifications.instrument("start_processing.action_controller", raw_payload.dup)

  ActiveSupport::Notifications.instrument("process_action.action_controller", raw_payload) do |payload|
    if respond_to?(:logstasher_add_custom_fields_to_request_context)
      logstasher_add_custom_fields_to_request_context(LogStasher.request_context)
    end

    if respond_to?(:logstasher_add_custom_fields_to_payload)
      before_keys = raw_payload.keys.clone
      logstasher_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::CustomFields.add(*(after_keys - before_keys))
    end

    result = super

    payload[:status] = response.status
    append_info_to_payload(payload)
    LogStasher.store.each do |key, value|
      payload[key] = value
    end

    LogStasher.request_context.each do |key, value|
      payload[key] = value
    end
    result
  end
end