ddtrace 1.6.1 → 1.8.0

data/lib/datadog/tracing/trace_digest.rb

@@ -6,6 +6,70 @@ module Datadog
     # Used to propagate context and continue traces across execution boundaries.
     # @public_api
     class TraceDigest
+      # @!attribute [r] span_id
+      #   Datadog id for the currently active span.
+      #   @return [Integer]
+      # @!attribute [r] span_name
+      #   The operation name of the currently active span.
+      #   @return [String]
+      # @!attribute [r] span_resource
+      #   The resource name of the currently active span.
+      #   @return [String]
+      # @!attribute [r] span_service
+      #   The service of the currently active span.
+      #   @return [String]
+      # @!attribute [r] span_type
+      #   The type of the currently active span.
+      #   @return [String]
+      # @!attribute [r] trace_distributed_tags
+      #   Datadog-specific tags that support richer distributed tracing association.
+      #   @return [Hash<String,String>]
+      # @!attribute [r] trace_hostname
+      #   The hostname of the currently active trace. Use to attribute traces to hosts.
+      #   @return [String]
+      # @!attribute [r] trace_id
+      #   Datadog id for the currently active trace.
+      #   @return [Integer]
+      # @!attribute [r] trace_name
+      #   Operation name for the currently active trace.
+      #   @return [Integer]
+      # @!attribute [r] trace_origin
+      #   Datadog-specific attribution of this trace's creation.
+      #   @return [String]
+      # @!attribute [r] trace_process_id
+      #   The OS-specific process id.
+      #   @return [Integer]
+      # @!attribute [r] trace_resource
+      #   The resource name of the currently active trace.
+      #   @return [String]
+      # @!attribute [r] trace_runtime_id
+      #   Unique id to this Ruby process. Used to differentiate traces coming from
+      #   child processes forked from same parent process.
+      #   @return [String]
+      # @!attribute [r] trace_sampling_priority
+      #   Datadog-specific sampling decision for the currently active trace.
+      #   @return [Integer]
+      # @!attribute [r] trace_service
+      #   The service of the currently active trace.
+      #   @return [String]
+      # @!attribute [r] trace_distributed_id
+      #   The trace id extracted from a distributed context, if different from `trace_id`.
+      #
+      #   The current use case is when the distributed context has a trace id integer larger than 64-bit:
+      #   This attribute will preserve the original id, while `trace_id` will only contain the lower 64 bits.
+      #   @return [Integer]
+      #   @see https://www.w3.org/TR/trace-context/#trace-id
+      # @!attribute [r] trace_tracestate
+      #   The W3C "tracestate" extracted from a distributed context.
+      #   This field is a string representing vendor-specific distribution data.
+      #   @return [String]
+      #   @see https://www.w3.org/TR/trace-context/#tracestate-header
+      # @!attribute [r] trace_flags
+      #   The W3C "trace-flags" extracted from a distributed context. This field is an 8-bit unsigned integer.
+      #   @return [Integer]
+      #   @see https://www.w3.org/TR/trace-context/#trace-flags
+      # TODO: The documentation for the last attribute above won't be rendered.
+      # TODO: This might be a YARD bug as adding an attribute, making it now second-last attribute, renders correctly.
       attr_reader \
         :span_id,
         :span_name,
@@ -21,7 +85,10 @@ module Datadog
         :trace_resource,
         :trace_runtime_id,
         :trace_sampling_priority,
-        :trace_service
+        :trace_service,
+        :trace_distributed_id,
+        :trace_flags,
+        :trace_state
 
       def initialize(
         span_id: nil,
@@ -38,7 +105,10 @@ module Datadog
         trace_resource: nil,
         trace_runtime_id: nil,
         trace_sampling_priority: nil,
-        trace_service: nil
+        trace_service: nil,
+        trace_distributed_id: nil,
+        trace_flags: nil,
+        trace_state: nil
       )
         @span_id = span_id
         @span_name = span_name && span_name.dup.freeze
@@ -55,6 +125,9 @@ module Datadog
         @trace_runtime_id = trace_runtime_id && trace_runtime_id.dup.freeze
         @trace_sampling_priority = trace_sampling_priority
         @trace_service = trace_service && trace_service.dup.freeze
+        @trace_distributed_id = trace_distributed_id
+        @trace_flags = trace_flags
+        @trace_state = trace_state && trace_state.dup.freeze
 
         freeze
       end

data/lib/datadog/tracing/trace_operation.rb

@@ -4,12 +4,13 @@ require_relative '../core'
 require_relative '../core/environment/identity'
 require_relative '../core/utils'
 
-require_relative 'sampling/ext'
 require_relative 'event'
+require_relative 'metadata/tagging'
+require_relative 'sampling/ext'
 require_relative 'span_operation'
-require_relative 'trace_segment'
 require_relative 'trace_digest'
-require_relative 'metadata/tagging'
+require_relative 'trace_segment'
+require_relative 'utils'
 
 module Datadog
   module Tracing
@@ -70,7 +71,7 @@ module Datadog
         metrics: nil
       )
         # Attributes
-        @id = id || Core::Utils.next_id
+        @id = id || Tracing::Utils.next_id
         @max_length = max_length || DEFAULT_MAX_LENGTH
         @parent_span_id = parent_span_id
         @sampled = sampled.nil? ? true : sampled

data/lib/datadog/tracing/trace_segment.rb

@@ -173,7 +173,7 @@ module Datadog
       end
 
       def process_id_tag
-        meta[Core::Runtime::Ext::TAG_PID]
+        meta[Core::Runtime::Ext::TAG_PROCESS_ID]
       end
 
       def rate_limiter_rate_tag

data/lib/datadog/tracing/utils.rb

@@ -0,0 +1,50 @@
+# typed: true
+
+require_relative '../core/utils/forking'
+
+module Datadog
+  module Tracing
+    # Utils contains low-level tracing utility functions.
+    # @public_api
+    module Utils
+      extend Datadog::Core::Utils::Forking
+
+      # The max value for a {Datadog::Tracing::Span} identifier.
+      # Span and trace identifiers should be strictly positive and strictly inferior to this limit.
+      #
+      # Limited to +2<<62-1+ positive integers, as Ruby is able to represent such numbers "inline",
+      # inside a +VALUE+ scalar, thus not requiring memory allocation.
+      #
+      # The range of IDs also has to consider portability across different languages and platforms.
+      RUBY_MAX_ID = (1 << 62) - 1
+
+      # Excludes zero from possible values
+      RUBY_ID_RANGE = (1..RUBY_MAX_ID).freeze
+
+      # While we only generate 63-bit integers due to limitations in other languages, we support
+      # parsing 64-bit integers for distributed tracing since an upstream system may generate one
+      EXTERNAL_MAX_ID = 1 << 64
+
+      # We use a custom random number generator because we want no interference
+      # with the default one. Using the default prng, we could break code that
+      # would rely on srand/rand sequences.
+
+      # Return a randomly generated integer, valid as a Span ID or Trace ID.
+      # This method is thread-safe and fork-safe.
+      def self.next_id
+        after_fork! { reset! }
+        id_rng.rand(RUBY_ID_RANGE)
+      end
+
+      def self.id_rng
+        @id_rng ||= Random.new
+      end
+
+      def self.reset!
+        @id_rng = Random.new
+      end
+
+      private_class_method :id_rng, :reset!
+    end
+  end
+end

data/lib/ddtrace/transport/trace_formatter.rb

@@ -94,7 +94,7 @@ module Datadog
       end
 
       def tag_lang!
-        return if trace.lang.nil? || root_span.get_tag(Tracing::Metadata::Ext::TAG_PEER_SERVICE)
+        return if trace.lang.nil?
 
         root_span.set_tag(
           Core::Runtime::Ext::TAG_LANG,
@@ -114,10 +114,7 @@ module Datadog
       def tag_process_id!
         return unless trace.process_id
 
-        root_span.set_tag(
-          Core::Runtime::Ext::TAG_PID,
-          trace.process_id
-        )
+        root_span.set_tag(Core::Runtime::Ext::TAG_PROCESS_ID, trace.process_id)
       end
 
       def tag_rate_limiter_rate!

data/lib/ddtrace/version.rb

@@ -3,8 +3,8 @@
 module DDTrace
   module VERSION
     MAJOR = 1
-    MINOR = 6
-    PATCH = 1
+    MINOR = 8
+    PATCH = 0
     PRE = nil
 
     STRING = [MAJOR, MINOR, PATCH, PRE].compact.join('.')

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ddtrace
 version: !ruby/object:Gem::Version
-  version: 1.6.1
+  version: 1.8.0
 platform: ruby
 authors:
 - Datadog, Inc.
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-11-16 00:00:00.000000000 Z
+date: 2022-12-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: msgpack
@@ -33,7 +33,7 @@ dependencies:
         version: 0.10.16
     - - "<="
       - !ruby/object:Gem::Version
-        version: 0.10.17
+        version: 0.10.18
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -43,7 +43,7 @@ dependencies:
         version: 0.10.16
     - - "<="
       - !ruby/object:Gem::Version
-        version: 0.10.17
+        version: 0.10.18
 - !ruby/object:Gem::Dependency
   name: libddwaf
   requirement: !ruby/object:Gem::Requirement
@@ -102,6 +102,10 @@ files:
 - ext/ddtrace_profiling_native_extension/collectors_cpu_and_wall_time.c
 - ext/ddtrace_profiling_native_extension/collectors_cpu_and_wall_time.h
 - ext/ddtrace_profiling_native_extension/collectors_cpu_and_wall_time_worker.c
+- ext/ddtrace_profiling_native_extension/collectors_dynamic_sampling_rate.c
+- ext/ddtrace_profiling_native_extension/collectors_dynamic_sampling_rate.h
+- ext/ddtrace_profiling_native_extension/collectors_idle_sampling_helper.c
+- ext/ddtrace_profiling_native_extension/collectors_idle_sampling_helper.h
 - ext/ddtrace_profiling_native_extension/collectors_stack.c
 - ext/ddtrace_profiling_native_extension/collectors_stack.h
 - ext/ddtrace_profiling_native_extension/extconf.rb
@@ -114,11 +118,17 @@ files:
 - ext/ddtrace_profiling_native_extension/profiling.c
 - ext/ddtrace_profiling_native_extension/ruby_helpers.c
 - ext/ddtrace_profiling_native_extension/ruby_helpers.h
+- ext/ddtrace_profiling_native_extension/setup_signal_handler.c
+- ext/ddtrace_profiling_native_extension/setup_signal_handler.h
 - ext/ddtrace_profiling_native_extension/stack_recorder.c
 - ext/ddtrace_profiling_native_extension/stack_recorder.h
+- ext/ddtrace_profiling_native_extension/time_helpers.c
+- ext/ddtrace_profiling_native_extension/time_helpers.h
 - lib/datadog/appsec.rb
 - lib/datadog/appsec/assets.rb
 - lib/datadog/appsec/assets/blocked.html
+- lib/datadog/appsec/assets/blocked.json
+- lib/datadog/appsec/assets/blocked.text
 - lib/datadog/appsec/assets/waf_rules/README.md
 - lib/datadog/appsec/assets/waf_rules/recommended.json
 - lib/datadog/appsec/assets/waf_rules/risky.json
@@ -168,6 +178,7 @@ files:
 - lib/datadog/appsec/reactive/engine.rb
 - lib/datadog/appsec/reactive/operation.rb
 - lib/datadog/appsec/reactive/subscriber.rb
+- lib/datadog/appsec/response.rb
 - lib/datadog/ci.rb
 - lib/datadog/ci/configuration/components.rb
 - lib/datadog/ci/configuration/settings.rb
@@ -199,6 +210,7 @@ files:
 - lib/datadog/core/configuration/base.rb
 - lib/datadog/core/configuration/components.rb
 - lib/datadog/core/configuration/dependency_resolver.rb
+- lib/datadog/core/configuration/ext.rb
 - lib/datadog/core/configuration/option.rb
 - lib/datadog/core/configuration/option_definition.rb
 - lib/datadog/core/configuration/option_definition_set.rb
@@ -206,7 +218,6 @@ files:
 - lib/datadog/core/configuration/options.rb
 - lib/datadog/core/configuration/settings.rb
 - lib/datadog/core/diagnostics/environment_logger.rb
-- lib/datadog/core/diagnostics/ext.rb
 - lib/datadog/core/diagnostics/health.rb
 - lib/datadog/core/encoding.rb
 - lib/datadog/core/environment/cgroup.rb
@@ -302,6 +313,8 @@ files:
 - lib/datadog/profiling/collectors/code_provenance.rb
 - lib/datadog/profiling/collectors/cpu_and_wall_time.rb
 - lib/datadog/profiling/collectors/cpu_and_wall_time_worker.rb
+- lib/datadog/profiling/collectors/dynamic_sampling_rate.rb
+- lib/datadog/profiling/collectors/idle_sampling_helper.rb
 - lib/datadog/profiling/collectors/old_stack.rb
 - lib/datadog/profiling/collectors/stack.rb
 - lib/datadog/profiling/encoding/profile.rb
@@ -339,6 +352,7 @@ files:
 - lib/datadog/tracing/buffer.rb
 - lib/datadog/tracing/client_ip.rb
 - lib/datadog/tracing/configuration/ext.rb
+- lib/datadog/tracing/configuration/settings.rb
 - lib/datadog/tracing/context.rb
 - lib/datadog/tracing/context_provider.rb
 - lib/datadog/tracing/contrib.rb
@@ -492,6 +506,8 @@ files:
 - lib/datadog/tracing/contrib/grpc/datadog_interceptor.rb
 - lib/datadog/tracing/contrib/grpc/datadog_interceptor/client.rb
 - lib/datadog/tracing/contrib/grpc/datadog_interceptor/server.rb
+- lib/datadog/tracing/contrib/grpc/distributed/fetcher.rb
+- lib/datadog/tracing/contrib/grpc/distributed/propagation.rb
 - lib/datadog/tracing/contrib/grpc/ext.rb
 - lib/datadog/tracing/contrib/grpc/integration.rb
 - lib/datadog/tracing/contrib/grpc/intercept_with_datadog.rb
@@ -506,6 +522,8 @@ files:
 - lib/datadog/tracing/contrib/hanami/router_tracing.rb
 - lib/datadog/tracing/contrib/http/circuit_breaker.rb
 - lib/datadog/tracing/contrib/http/configuration/settings.rb
+- lib/datadog/tracing/contrib/http/distributed/fetcher.rb
+- lib/datadog/tracing/contrib/http/distributed/propagation.rb
 - lib/datadog/tracing/contrib/http/ext.rb
 - lib/datadog/tracing/contrib/http/instrumentation.rb
 - lib/datadog/tracing/contrib/http/integration.rb
@@ -622,6 +640,7 @@ files:
 - lib/datadog/tracing/contrib/redis/patcher.rb
 - lib/datadog/tracing/contrib/redis/quantize.rb
 - lib/datadog/tracing/contrib/redis/tags.rb
+- lib/datadog/tracing/contrib/redis/trace_middleware.rb
 - lib/datadog/tracing/contrib/redis/vendor/LICENSE
 - lib/datadog/tracing/contrib/redis/vendor/resolver.rb
 - lib/datadog/tracing/contrib/registerable.rb
@@ -689,17 +708,18 @@ files:
 - lib/datadog/tracing/contrib/utils/quantization/hash.rb
 - lib/datadog/tracing/contrib/utils/quantization/http.rb
 - lib/datadog/tracing/correlation.rb
+- lib/datadog/tracing/diagnostics/ext.rb
+- lib/datadog/tracing/diagnostics/health.rb
+- lib/datadog/tracing/distributed/b3_multi.rb
+- lib/datadog/tracing/distributed/b3_single.rb
+- lib/datadog/tracing/distributed/datadog.rb
 - lib/datadog/tracing/distributed/datadog_tags_codec.rb
-- lib/datadog/tracing/distributed/headers/b3.rb
-- lib/datadog/tracing/distributed/headers/b3_single.rb
-- lib/datadog/tracing/distributed/headers/datadog.rb
+- lib/datadog/tracing/distributed/fetcher.rb
 - lib/datadog/tracing/distributed/headers/ext.rb
-- lib/datadog/tracing/distributed/headers/parser.rb
 - lib/datadog/tracing/distributed/helpers.rb
-- lib/datadog/tracing/distributed/metadata/b3.rb
-- lib/datadog/tracing/distributed/metadata/b3_single.rb
-- lib/datadog/tracing/distributed/metadata/datadog.rb
-- lib/datadog/tracing/distributed/metadata/parser.rb
+- lib/datadog/tracing/distributed/none.rb
+- lib/datadog/tracing/distributed/propagation.rb
+- lib/datadog/tracing/distributed/trace_context.rb
 - lib/datadog/tracing/event.rb
 - lib/datadog/tracing/flush.rb
 - lib/datadog/tracing/metadata.rb
@@ -710,7 +730,6 @@ files:
 - lib/datadog/tracing/pipeline.rb
 - lib/datadog/tracing/pipeline/span_filter.rb
 - lib/datadog/tracing/pipeline/span_processor.rb
-- lib/datadog/tracing/propagation/grpc.rb
 - lib/datadog/tracing/propagation/http.rb
 - lib/datadog/tracing/runtime/metrics.rb
 - lib/datadog/tracing/sampling/all_sampler.rb
@@ -736,6 +755,7 @@ files:
 - lib/datadog/tracing/trace_operation.rb
 - lib/datadog/tracing/trace_segment.rb
 - lib/datadog/tracing/tracer.rb
+- lib/datadog/tracing/utils.rb
 - lib/datadog/tracing/workers.rb
 - lib/datadog/tracing/workers/trace_writer.rb
 - lib/datadog/tracing/writer.rb
@@ -796,7 +816,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: 2.0.0
 requirements: []
-rubygems_version: 3.3.7
+rubygems_version: 3.3.26
 signing_key:
 specification_version: 4
 summary: Datadog tracing code for your Ruby applications

data/lib/datadog/tracing/distributed/headers/b3.rb

@@ -1,55 +0,0 @@
-# typed: true
-
-require_relative 'parser'
-require_relative '../helpers'
-require_relative 'ext'
-require_relative '../../trace_digest'
-
-module Datadog
-  module Tracing
-    module Distributed
-      module Headers
-        # B3 provides helpers to inject or extract headers for B3 style headers
-        module B3
-          include Ext
-
-          def self.inject!(digest, env)
-            return if digest.nil?
-
-            # DEV: We need these to be hex encoded
-            env[B3_HEADER_TRACE_ID] = digest.trace_id.to_s(16)
-            env[B3_HEADER_SPAN_ID] = digest.span_id.to_s(16)
-
-            if digest.trace_sampling_priority
-              sampling_priority = Helpers.clamp_sampling_priority(
-                digest.trace_sampling_priority
-              )
-              env[B3_HEADER_SAMPLED] = sampling_priority.to_s
-            end
-
-            env
-          end
-
-          def self.extract(env)
-            # Extract values from headers
-            # DEV: B3 doesn't have "origin"
-            headers = Parser.new(env)
-            trace_id = headers.id(B3_HEADER_TRACE_ID, 16)
-            span_id = headers.id(B3_HEADER_SPAN_ID, 16)
-            # We don't need to try and convert sampled since B3 supports 0/1 (AUTO_REJECT/AUTO_KEEP)
-            sampling_priority = headers.number(B3_HEADER_SAMPLED)
-
-            # Return early if this propagation is not valid
-            return unless trace_id && span_id
-
-            TraceDigest.new(
-              trace_id: trace_id,
-              span_id: span_id,
-              trace_sampling_priority: sampling_priority
-            )
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/tracing/distributed/headers/b3_single.rb

@@ -1,67 +0,0 @@
-# typed: true
-
-require_relative 'parser'
-require_relative '../helpers'
-require_relative 'ext'
-require_relative '../../trace_digest'
-
-module Datadog
-  module Tracing
-    module Distributed
-      module Headers
-        # B3Single provides helpers to inject or extract headers for B3 single header style headers
-        module B3Single
-          include Ext
-
-          def self.inject!(digest, env)
-            return if digest.nil?
-
-            # Header format:
-            #   b3: {TraceId}-{SpanId}-{SamplingState}-{ParentSpanId}
-            # https://github.com/apache/incubator-zipkin-b3-propagation/tree/7c6e9f14d6627832bd80baa87ac7dabee7be23cf#single-header
-            # DEV: `{SamplingState}` and `{ParentSpanId`}` are optional
-
-            # DEV: We need these to be hex encoded
-            header = "#{digest.trace_id.to_s(16)}-#{digest.span_id.to_s(16)}"
-
-            if digest.trace_sampling_priority
-              sampling_priority = Helpers.clamp_sampling_priority(
-                digest.trace_sampling_priority
-              )
-              header += "-#{sampling_priority}"
-            end
-
-            env[B3_HEADER_SINGLE] = header
-
-            env
-          end
-
-          def self.extract(env)
-            # Header format:
-            #   b3: {TraceId}-{SpanId}-{SamplingState}-{ParentSpanId}
-            # https://github.com/apache/incubator-zipkin-b3-propagation/tree/7c6e9f14d6627832bd80baa87ac7dabee7be23cf#single-header
-            # DEV: `{SamplingState}` and `{ParentSpanId`}` are optional
-
-            headers = Parser.new(env)
-            value = headers.header(B3_HEADER_SINGLE)
-            return if value.nil?
-
-            parts = value.split('-')
-            trace_id = Helpers.value_to_id(parts[0], 16) unless parts.empty?
-            span_id = Helpers.value_to_id(parts[1], 16) if parts.length > 1
-            sampling_priority = Helpers.value_to_number(parts[2]) if parts.length > 2
-
-            # Return early if this propagation is not valid
-            return unless trace_id && span_id
-
-            TraceDigest.new(
-              span_id: span_id,
-              trace_id: trace_id,
-              trace_sampling_priority: sampling_priority
-            )
-          end
-        end
-      end
-    end
-  end
-end

data/lib/datadog/tracing/distributed/headers/datadog.rb

@@ -1,144 +0,0 @@
-# frozen_string_literal: true
-# typed: true
-
-require_relative 'parser'
-require_relative 'ext'
-require_relative '../../metadata/ext'
-require_relative '../../trace_digest'
-require_relative '../datadog_tags_codec'
-
-module Datadog
-  module Tracing
-    module Distributed
-      module Headers
-        # Datadog provides helpers to inject or extract headers for Datadog style headers
-        module Datadog
-          class << self
-            include Ext
-
-            def inject!(digest, env)
-              return if digest.nil?
-
-              env[HTTP_HEADER_TRACE_ID] = digest.trace_id.to_s
-              env[HTTP_HEADER_PARENT_ID] = digest.span_id.to_s
-              env[HTTP_HEADER_SAMPLING_PRIORITY] = digest.trace_sampling_priority.to_s if digest.trace_sampling_priority
-              env[HTTP_HEADER_ORIGIN] = digest.trace_origin.to_s unless digest.trace_origin.nil?
-
-              inject_tags(digest, env)
-
-              env
-            end
-
-            def extract(env)
-              # Extract values from headers
-              headers = Parser.new(env)
-              trace_id = headers.id(HTTP_HEADER_TRACE_ID)
-              parent_id = headers.id(HTTP_HEADER_PARENT_ID)
-              origin = headers.header(HTTP_HEADER_ORIGIN)
-              sampling_priority = headers.number(HTTP_HEADER_SAMPLING_PRIORITY)
-
-              # Return early if this propagation is not valid
-              # DEV: To be valid we need to have a trace id and a parent id
-              #      or when it is a synthetics trace, just the trace id.
-              # DEV: `Parser#id` will not return 0
-              return unless (trace_id && parent_id) || (origin && trace_id)
-
-              trace_distributed_tags = extract_tags(headers)
-
-              # Return new trace headers
-              TraceDigest.new(
-                span_id: parent_id,
-                trace_id: trace_id,
-                trace_origin: origin,
-                trace_sampling_priority: sampling_priority,
-                trace_distributed_tags: trace_distributed_tags,
-              )
-            end
-
-            private
-
-            # Export trace distributed tags through the `x-datadog-tags` header.
-            #
-            # DEV: This method accesses global state (the active trace) to record its error state as a trace tag.
-            # DEV: This means errors cannot be reported if there's not active span.
-            # DEV: Ideally, we'd have a dedicated error reporting stream for all of ddtrace.
-            # DEV: The same comment applies to the {.extract_tags}.
-            def inject_tags(digest, env)
-              return if digest.trace_distributed_tags.nil? || digest.trace_distributed_tags.empty?
-
-              if ::Datadog.configuration.tracing.x_datadog_tags_max_length <= 0
-                active_trace = Tracing.active_trace
-                active_trace.set_tag('_dd.propagation_error', 'disabled') if active_trace
-                return
-              end
-
-              encoded_tags = DatadogTagsCodec.encode(digest.trace_distributed_tags)
-
-              if encoded_tags.size > ::Datadog.configuration.tracing.x_datadog_tags_max_length
-                active_trace = Tracing.active_trace
-                active_trace.set_tag('_dd.propagation_error', 'inject_max_size') if active_trace
-
-                ::Datadog.logger.warn(
-                  "Failed to inject x-datadog-tags: tags are too large (size:#{encoded_tags.size} " \
-                  "limit:#{::Datadog.configuration.tracing.x_datadog_tags_max_length}). This limit can be configured " \
-                  'through the DD_TRACE_X_DATADOG_TAGS_MAX_LENGTH environment variable.'
-                )
-                return
-              end
-
-              env[HTTP_HEADER_TAGS] = encoded_tags
-            rescue => e
-              active_trace = Tracing.active_trace
-              active_trace.set_tag('_dd.propagation_error', 'encoding_error') if active_trace
-              ::Datadog.logger.warn(
-                "Failed to inject x-datadog-tags: #{e.class.name} #{e.message} at #{Array(e.backtrace).first}"
-              )
-            end
-
-            # Import `x-datadog-tags` header tags as trace distributed tags.
-            # Only tags that have the `_dd.p.` prefix are processed.
-            def extract_tags(headers)
-              tags_header = headers.header(HTTP_HEADER_TAGS)
-              return unless tags_header
-
-              if ::Datadog.configuration.tracing.x_datadog_tags_max_length <= 0
-                active_trace = Tracing.active_trace
-                active_trace.set_tag('_dd.propagation_error', 'disabled') if active_trace
-                return
-              end
-
-              if tags_header.size > ::Datadog.configuration.tracing.x_datadog_tags_max_length
-                active_trace = Tracing.active_trace
-                active_trace.set_tag('_dd.propagation_error', 'extract_max_size') if active_trace
-
-                ::Datadog.logger.warn(
-                  "Failed to extract x-datadog-tags: tags are too large (size:#{tags_header.size} " \
-                  "limit:#{::Datadog.configuration.tracing.x_datadog_tags_max_length}). This limit can be configured " \
-                  'through the DD_TRACE_X_DATADOG_TAGS_MAX_LENGTH environment variable.'
-                )
-                return
-              end
-
-              tags = DatadogTagsCodec.decode(tags_header)
-              # Only extract keys with the expected Datadog prefix
-              tags.select! do |key, _|
-                key.start_with?(Tracing::Metadata::Ext::Distributed::TAGS_PREFIX) && key != EXCLUDED_TAG
-              end
-              tags
-            rescue => e
-              active_trace = Tracing.active_trace
-              active_trace.set_tag('_dd.propagation_error', 'decoding_error') if active_trace
-              ::Datadog.logger.warn(
-                "Failed to extract x-datadog-tags: #{e.class.name} #{e.message} at #{Array(e.backtrace).first}"
-              )
-            end
-
-            # We want to exclude tags that we don't want to propagate downstream.
-            EXCLUDED_TAG = '_dd.p.upstream_services'
-            private_constant :EXCLUDED_TAG
-          end
-        end
-      end
-    end
-  end
-end