Sha256: 36dc46ac6da0eaa72a804d08c8fdbbc74217a14849192b2044c2c49747fd9880
Contents?: true
Size: 1.63 KB
Versions: 13
Compression:
Stored size: 1.63 KB
Contents
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
# frozen_string_literal: true
module NewRelic
module Agent
module Instrumentation
module NetHTTP
INSTRUMENTATION_NAME = NewRelic::Agent.base_name(name)
def request_with_tracing(request)
NewRelic::Agent.record_instrumentation_invocation(INSTRUMENTATION_NAME)
wrapped_request = NewRelic::Agent::HTTPClients::NetHTTPRequest.new(self, 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)
# RUBY-1244 Disable further tracing in request to avoid double
# counting if connection wasn't started (which calls request again).
NewRelic::Agent.disable_all_tracing do
response = NewRelic::Agent::Tracer.capture_segment_error(segment) do
yield
end
end
wrapped_response = NewRelic::Agent::HTTPClients::NetHTTPResponse.new(response)
if NewRelic::Agent::LLM.openai_parent?(segment)
NewRelic::Agent::LLM.populate_openai_response_headers(wrapped_response, segment.parent)
end
segment.process_response_headers(wrapped_response)
response
ensure
segment&.finish
end
end
end
end
end
end
Version data entries
13 entries across 13 versions & 1 rubygems