Sha256: c87f57cfe4c8a28f9212489b8d3d7e225b257bab261577919b175cd566f2fe05
Contents?: true
Size: 1.61 KB
Versions: 8
Compression:
Stored size: 1.61 KB
Contents
require 'ddtrace/ext/http'
require 'ddtrace/contrib/analytics'
require 'ddtrace/contrib/grpc/ext'
module Datadog
module Contrib
module GRPC
module DatadogInterceptor
# The DatadogInterceptor::Client implements the tracing strategy
# for gRPC client-side endpoitns. This middleware compoent will
# inject trace context information into gRPC metadata prior to
# sending the request to the server.
class Client < Base
def trace(keywords)
keywords[:metadata] ||= {}
options = {
span_type: Datadog::Ext::HTTP::TYPE_OUTBOUND,
service: service_name,
resource: format_resource(keywords[:method])
}
tracer.trace(Ext::SPAN_CLIENT, options) do |span|
annotate!(span, keywords[:metadata])
yield
end
end
private
def annotate!(span, metadata)
metadata.each do |header, value|
span.set_tag(header, value)
end
# Set analytics sample rate
Contrib::Analytics.set_sample_rate(span, analytics_sample_rate) if analytics_enabled?
Datadog::GRPCPropagator
.inject!(span.context, metadata)
rescue StandardError => e
Datadog::Logger.log.debug("GRPC client trace failed: #{e}")
end
def format_resource(proto_method)
proto_method.downcase
.split('/')
.reject(&:empty?)
.join('.')
end
end
end
end
end
end
Version data entries
8 entries across 8 versions & 1 rubygems