Module: Datadog::RailsActionPatcher

Defined in:
lib/ddtrace/contrib/rails/core_extensions.rb

Overview

RailsActionPatcher contains functions to patch Rails action controller instrumentation

Class Method Summary collapse

Class Method Details

.patch_action_controllerObject



125
126
127
# File 'lib/ddtrace/contrib/rails/core_extensions.rb', line 125

def patch_action_controller
  patch_process_action
end

.patch_process_actionObject



129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
# File 'lib/ddtrace/contrib/rails/core_extensions.rb', line 129

def patch_process_action
  ::ActionController::Instrumentation.class_eval do
    def process_action_with_datadog(*args)
      # mutable payload with a tracing context that is used in two different
      # signals; it propagates the request span so that it can be finished
      # no matter what
      payload = {
        controller: self.class.name,
        action: action_name,
        tracing_context: {}
      }

      begin
        # process and catch request exceptions
        Datadog::Contrib::Rails::ActionController.start_processing(payload)
        result = process_action_without_datadog(*args)
        payload[:status] = response.status
        result
      rescue Exception => e
        payload[:exception] = [e.class.name, e.message]
        payload[:exception_object] = e
        raise e
      end
    ensure
      Datadog::Contrib::Rails::ActionController.finish_processing(payload)
    end

    alias_method :process_action_without_datadog, :process_action
    alias_method :process_action, :process_action_with_datadog
  end
end