Sha256: 9f25e534abba6595fcf79dd0bbc5dae90749522790869caa7af349e7f395a247

Contents?: true

Size: 1.44 KB

Versions: 10

Compression:

Stored size: 1.44 KB

Contents

require 'ddtrace/ext/http'
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,
              service: datadog_pin.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

            Datadog::GRPCPropagator
              .inject!(span.context, metadata)
          rescue StandardError => e
            Datadog::Tracer.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

10 entries across 10 versions & 1 rubygems

Version Path
ddtrace-0.19.1 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.19.0 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.18.3 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.18.2 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.18.1 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.18.0 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.17.3 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.17.2 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.17.1 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb
ddtrace-0.17.0 lib/ddtrace/contrib/grpc/datadog_interceptor/client.rb