Sha256: f78516777c4569d8982a8151aa2596db1a29d8c5799e73bdce9d9782bebf5df3

Contents?: true

Size: 1.64 KB

Versions: 27

Compression:

Stored size: 1.64 KB

Contents

require 'ddtrace/context'
require 'ddtrace/ext/distributed'

module Datadog
  # opentracing.io compliant methods for distributing trace context
  # between two or more distributed services. Note this is very close
  # to the HTTPPropagator; the key difference is the way gRPC handles
  # header information (called "metadata") as it operates over HTTP2
  module GRPCPropagator
    include Ext::DistributedTracing

    def self.inject!(context, metadata)
      metadata[GRPC_METADATA_TRACE_ID] = context.trace_id.to_s
      metadata[GRPC_METADATA_PARENT_ID] = context.span_id.to_s
      metadata[GRPC_METADATA_SAMPLING_PRIORITY] = context.sampling_priority.to_s if context.sampling_priority
    end

    def self.extract(metadata)
      metadata = Carrier.new(metadata)
      return Datadog::Context.new unless metadata.valid?
      Datadog::Context.new(trace_id: metadata.trace_id,
                           span_id: metadata.parent_id,
                           sampling_priority: metadata.sampling_priority)
    end

    # opentracing.io compliant carrier object
    class Carrier
      include Ext::DistributedTracing

      def initialize(metadata = {})
        @metadata = metadata || {}
      end

      def valid?
        trace_id && parent_id
      end

      def trace_id
        value = @metadata[GRPC_METADATA_TRACE_ID].to_i
        value if (1..Span::MAX_ID).cover? value
      end

      def parent_id
        value = @metadata[GRPC_METADATA_PARENT_ID].to_i
        value if (1..Span::MAX_ID).cover? value
      end

      def sampling_priority
        value = @metadata[GRPC_METADATA_SAMPLING_PRIORITY]
        value && value.to_i
      end
    end
  end
end

Version data entries

27 entries across 27 versions & 1 rubygems

Version Path
ddtrace-0.19.1 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.19.0 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.18.3 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.18.2 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.18.1 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.18.0 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.17.3 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.17.2 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.17.1 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.17.0 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.16.1 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.15.0.internaltracinfeature1 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.16.0 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.14.2.disableprotocolversion4 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.15.0 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.14.2.withoutpriorityparsing1 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.14.2 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.14.1 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.15.0.beta1 lib/ddtrace/propagation/grpc_propagator.rb
ddtrace-0.14.0 lib/ddtrace/propagation/grpc_propagator.rb