Module: NewRelic::Agent::Instrumentation::HTTPClient::Instrumentation

Included in:
Prepend
Defined in:
lib/new_relic/agent/instrumentation/httpclient/instrumentation.rb

Constant Summary collapse

INSTRUMENTATION_NAME = 
'HTTPClient'

Instance Method Summary collapse

Instance Method Details

#with_tracing(request, connection) ⇒ Object 



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
 
# File 'lib/new_relic/agent/instrumentation/httpclient/instrumentation.rb', line 10

def with_tracing(request, connection)
  NewRelic::Agent.record_instrumentation_invocation(INSTRUMENTATION_NAME)

  wrapped_request = NewRelic::Agent::HTTPClients::HTTPClientRequest.new(request)
  segment = NewRelic::Agent::Tracer.start_external_request_segment(
    library: wrapped_request.type,
    uri: wrapped_request.uri,
    procedure: wrapped_request.method
  )

  begin
    response = nil
    segment.add_request_headers(wrapped_request)

    NewRelic::Agent::Tracer.capture_segment_error(segment) do
      yield
    end

    response = connection.pop
    connection.push(response)

    wrapped_response = ::NewRelic::Agent::HTTPClients::HTTPClientResponse.new(response)
    segment.process_response_headers(wrapped_response)

    response
  ensure
    ::NewRelic::Agent::Transaction::Segment.finish(segment)
  end
end