ddtrace 1.9.0 → 1.10.1

data/lib/datadog/tracing/contrib/shoryuken/tracer.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require_relative '../analytics'
 
 module Datadog
@@ -20,6 +18,8 @@ module Datadog
               span_type: Tracing::Metadata::Ext::AppTypes::TYPE_WORKER,
               on_error: @error_handler
             ) do |span|
+              span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_MESSAGING_SYSTEM)
+
               span.set_tag(Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
               span.set_tag(Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_JOB)
 

data/lib/datadog/tracing/contrib/sidekiq/client_tracer.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require_relative '../../metadata/ext'
 require_relative '../analytics'
 require_relative 'ext'
@@ -25,6 +23,8 @@ module Datadog
             Datadog::Tracing.trace(Ext::SPAN_PUSH, service: @sidekiq_service) do |span|
               span.resource = resource
 
+              span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_COMPONENT)
+
               span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
               span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_PUSH)
 

data/lib/datadog/tracing/contrib/sidekiq/configuration/settings.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../../../span_operation'
 require_relative '../../configuration/settings'
 require_relative '../ext'

data/lib/datadog/tracing/contrib/sidekiq/ext.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib

data/lib/datadog/tracing/contrib/sidekiq/integration.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../integration'
 require_relative 'configuration/settings'
 require_relative 'patcher'

data/lib/datadog/tracing/contrib/sidekiq/patcher.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../patcher'
 
 module Datadog

data/lib/datadog/tracing/contrib/sidekiq/server_internal_tracer/heartbeat.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib
@@ -13,6 +11,8 @@ module Datadog
               Datadog::Tracing.trace(Ext::SPAN_STOP, service: configuration[:service_name]) do |span|
                 span.span_type = Datadog::Tracing::Metadata::Ext::AppTypes::TYPE_WORKER
 
+                span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_COMPONENT)
+
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_STOP)
 
@@ -33,6 +33,8 @@ module Datadog
               Datadog::Tracing.trace(Ext::SPAN_HEARTBEAT, service: configuration[:service_name]) do |span|
                 span.span_type = Datadog::Tracing::Metadata::Ext::AppTypes::TYPE_WORKER
 
+                span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_COMPONENT)
+
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_HEARTBEAT)
 

data/lib/datadog/tracing/contrib/sidekiq/server_internal_tracer/job_fetch.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib
@@ -15,6 +13,8 @@ module Datadog
               Datadog::Tracing.trace(Ext::SPAN_JOB_FETCH, service: configuration[:service_name]) do |span|
                 span.span_type = Datadog::Tracing::Metadata::Ext::AppTypes::TYPE_WORKER
 
+                span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_COMPONENT)
+
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_JOB_FETCH)
 

data/lib/datadog/tracing/contrib/sidekiq/server_internal_tracer/redis_info.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib
@@ -13,6 +11,8 @@ module Datadog
               Datadog::Tracing.trace(Ext::SPAN_REDIS_INFO, service: configuration[:service_name]) do |span|
                 span.span_type = Datadog::Tracing::Metadata::Ext::AppTypes::TYPE_WORKER
 
+                span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_COMPONENT)
+
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_REDIS_INFO)
 

data/lib/datadog/tracing/contrib/sidekiq/server_internal_tracer/scheduled_poller.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib
@@ -14,6 +12,8 @@ module Datadog
               Datadog::Tracing.trace(Ext::SPAN_SCHEDULED_PUSH, service: configuration[:service_name]) do |span|
                 span.span_type = Datadog::Tracing::Metadata::Ext::AppTypes::TYPE_WORKER
 
+                span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_COMPONENT)
+
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_SCHEDULED_PUSH)
 
@@ -34,6 +34,8 @@ module Datadog
               Datadog::Tracing.trace(Ext::SPAN_SCHEDULED_WAIT, service: configuration[:service_name]) do |span|
                 span.span_type = Datadog::Tracing::Metadata::Ext::AppTypes::TYPE_WORKER
 
+                span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_COMPONENT)
+
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
                 span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_SCHEDULED_WAIT)
 

data/lib/datadog/tracing/contrib/sidekiq/server_tracer.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require_relative '../../metadata/ext'
 require_relative '../analytics'
 require_relative 'ext'
@@ -37,6 +35,8 @@ module Datadog
             ) do |span|
               span.resource = resource
 
+              span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_COMPONENT)
+
               span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
               span.set_tag(Datadog::Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_JOB)
 

data/lib/datadog/tracing/contrib/sidekiq/tracing.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require 'yaml'
 
 require_relative '../../../core'

data/lib/datadog/tracing/contrib/sinatra/configuration/settings.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../../configuration/settings'
 require_relative '../ext'
 

data/lib/datadog/tracing/contrib/sinatra/env.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require 'time'
 
 require_relative '../../metadata/ext'

data/lib/datadog/tracing/contrib/sinatra/ext.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib

data/lib/datadog/tracing/contrib/sinatra/framework.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 module Datadog
   module Tracing
     module Contrib

data/lib/datadog/tracing/contrib/sinatra/headers.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require_relative '../../metadata/ext'
 
 module Datadog

data/lib/datadog/tracing/contrib/sinatra/integration.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../integration'
 require_relative 'configuration/settings'
 require_relative 'patcher'

data/lib/datadog/tracing/contrib/sinatra/patcher.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../../../core/utils/only_once'
 require_relative '../patcher'
 require_relative '../rack/middlewares'
@@ -40,7 +38,6 @@ module Datadog
 
         # Patcher enables patching of 'sinatra' module.
         module Patcher
-          include Kernel # Ensure that kernel methods are always available (https://sorbet.org/docs/error-reference#7003)
           include Contrib::Patcher
 
           module_function

data/lib/datadog/tracing/contrib/sinatra/tracer.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require 'sinatra/base'
 
 require_relative '../../../core/utils/only_once'

data/lib/datadog/tracing/contrib/sinatra/tracer_middleware.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../../metadata/ext'
 require_relative '../analytics'
 require_relative '../rack/ext'

data/lib/datadog/tracing/contrib/sneakers/configuration/settings.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-# typed: false
-
 require_relative '../../../span_operation'
 require_relative '../../configuration/settings'
 

data/lib/datadog/tracing/contrib/sneakers/ext.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-# typed: strict
-
 module Datadog
   module Tracing
     module Contrib
@@ -19,6 +17,7 @@ module Datadog
           TAG_JOB_BODY = 'sneakers.body'
           TAG_COMPONENT = 'sneakers'
           TAG_OPERATION_JOB = 'job'
+          TAG_MESSAGING_SYSTEM = 'rabbitmq'
         end
       end
     end

data/lib/datadog/tracing/contrib/sneakers/integration.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-# typed: false
-
 require_relative '../integration'
 require_relative 'ext'
 require_relative 'configuration/settings'

data/lib/datadog/tracing/contrib/sneakers/patcher.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-# typed: false
-
 require_relative '../patcher'
 require_relative 'tracer'
 

data/lib/datadog/tracing/contrib/sneakers/tracer.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-# typed: true
-
 require_relative '../../metadata/ext'
 require_relative '../analytics'
 
@@ -24,6 +22,8 @@ module Datadog
             }
 
             Tracing.trace(Ext::SPAN_JOB, **trace_options) do |request_span|
+              request_span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_MESSAGING_SYSTEM)
+
               request_span.set_tag(Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
               request_span.set_tag(Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_JOB)
 
@@ -37,6 +37,7 @@ module Datadog
 
               request_span.resource = @app.to_proc.binding.eval('self.class').to_s
               request_span.set_tag(Ext::TAG_JOB_ROUTING_KEY, delivery_info.routing_key)
+              request_span.set_tag(Contrib::Ext::Messaging::TAG_RABBITMQ_ROUTING_KEY, delivery_info.routing_key)
               request_span.set_tag(Ext::TAG_JOB_QUEUE, delivery_info.consumer.queue.name)
 
               request_span.set_tag(Ext::TAG_JOB_BODY, deserialized_msg) if configuration[:tag_body]

data/lib/datadog/tracing/contrib/status_code_matcher.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require 'set'
 
 require_relative '../../core'

data/lib/datadog/tracing/contrib/stripe/patcher.rb

@@ -1,5 +1,4 @@
 # frozen_string_literal: true
-# typed: false
 
 require_relative '../patcher'
 require_relative 'request'

data/lib/datadog/tracing/contrib/sucker_punch/configuration/settings.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../../configuration/settings'
 require_relative '../ext'
 

data/lib/datadog/tracing/contrib/sucker_punch/exception_handler.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require 'sucker_punch'
 
 module Datadog

data/lib/datadog/tracing/contrib/sucker_punch/ext.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib

data/lib/datadog/tracing/contrib/sucker_punch/instrumentation.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require 'sucker_punch'
 
 require_relative '../../metadata/ext'

data/lib/datadog/tracing/contrib/sucker_punch/integration.rb

@@ -1,5 +1,3 @@
-# typed: false
-
 require_relative '../integration'
 require_relative 'configuration/settings'
 require_relative 'patcher'

data/lib/datadog/tracing/contrib/sucker_punch/patcher.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require_relative '../patcher'
 require_relative 'ext'
 
@@ -9,7 +7,6 @@ module Datadog
       module SuckerPunch
         # Patcher enables patching of 'sucker_punch' module.
         module Patcher
-          include Kernel # Ensure that kernel methods are always available (https://sorbet.org/docs/error-reference#7003)
           include Contrib::Patcher
 
           module_function

data/lib/datadog/tracing/contrib/utils/database.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib

data/lib/datadog/tracing/contrib/utils/quantization/hash.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Contrib

data/lib/datadog/tracing/contrib/utils/quantization/http.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require 'uri'
 require 'set'
 
@@ -10,8 +8,6 @@ module Datadog
         module Quantization
           # Quantization for HTTP resources
           module HTTP
-            include Kernel # Ensure that kernel methods are always available (https://sorbet.org/docs/error-reference#7003)
-
             PLACEHOLDER = '?'.freeze
 
             # taken from Ruby https://github.com/ruby/uri/blob/ffbab83de6d8748c9454414e02db5317609166eb/lib/uri/rfc3986_parser.rb

data/lib/datadog/tracing/contrib.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 require_relative '../tracing'
 require_relative 'contrib/registry'
 require_relative 'contrib/extensions'

data/lib/datadog/tracing/correlation.rb

@@ -1,6 +1,6 @@
-# typed: true
-
 require_relative '../core'
+require_relative 'utils'
+require_relative 'metadata/ext'
 
 module Datadog
   module Tracing
@@ -66,11 +66,23 @@ module Datadog
             attributes << "#{LOG_ATTR_ENV}=#{env}" unless env.nil?
             attributes << "#{LOG_ATTR_SERVICE}=#{service}"
             attributes << "#{LOG_ATTR_VERSION}=#{version}" unless version.nil?
-            attributes << "#{LOG_ATTR_TRACE_ID}=#{trace_id}"
+            attributes << "#{LOG_ATTR_TRACE_ID}=#{logging_trace_id}"
             attributes << "#{LOG_ATTR_SPAN_ID}=#{span_id}"
             attributes.join(' ')
           end
         end
+
+        private
+
+        def logging_trace_id
+          @logging_trace_id ||=
+            if Datadog.configuration.tracing.trace_id_128_bit_logging_enabled &&
+                !Tracing::Utils::TraceId.to_high_order(@trace_id).zero?
+              Kernel.format('%032x', trace_id)
+            else
+              Tracing::Utils::TraceId.to_low_order(@trace_id)
+            end
+        end
       end
 
       module_function

data/lib/datadog/tracing/diagnostics/ext.rb

@@ -1,5 +1,3 @@
-# typed: true
-
 module Datadog
   module Tracing
     module Diagnostics

data/lib/datadog/tracing/diagnostics/health.rb

@@ -1,5 +1,3 @@
-# typed: ignore
-
 require_relative 'ext'
 
 module Datadog

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

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
-# typed: true
 
 require_relative 'helpers'
 require_relative '../trace_digest'
+require_relative '../utils'
 
 module Datadog
   module Tracing
@@ -46,13 +46,19 @@ module Datadog
         def extract(data)
           # DEV: B3 doesn't have "origin"
           fetcher = @fetcher.new(data)
-          trace_id = fetcher.id(@trace_id_key, base: 16)
-          span_id = fetcher.id(@span_id_key, base: 16)
-          # We don't need to try and convert sampled since B3 supports 0/1 (AUTO_REJECT/AUTO_KEEP)
-          sampling_priority = fetcher.number(@sampled_key)
+
+          trace_id = Helpers.parse_hex_id(fetcher[@trace_id_key])
 
           # Return early if this propagation is not valid
-          return unless trace_id && span_id
+          return if trace_id.nil? || trace_id <= 0 || trace_id > Tracing::Utils::TraceId::MAX
+
+          span_id = Helpers.parse_hex_id(fetcher[@span_id_key])
+
+          # Return early if this propagation is not valid
+          return if span_id.nil? || span_id <= 0 || span_id >= Tracing::Utils::EXTERNAL_MAX_ID
+
+          # We don't need to try and convert sampled since B3 supports 0/1 (AUTO_REJECT/AUTO_KEEP)
+          sampling_priority = Helpers.parse_decimal_id(fetcher[@sampled_key])
 
           TraceDigest.new(
             trace_id: trace_id,

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

@@ -1,5 +1,4 @@
 # frozen_string_literal: true
-# typed: true
 
 require_relative 'helpers'
 require_relative '../trace_digest'
@@ -47,12 +46,15 @@ module Datadog
           return unless value
 
           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
+          trace_id = Helpers.parse_hex_id(parts[0]) unless parts.empty?
+          # Return early if this propagation is not valid
+          return if trace_id.nil? || trace_id <= 0 || trace_id > Tracing::Utils::TraceId::MAX
 
-          # Return if this propagation is not valid
-          return unless trace_id && span_id
+          span_id = Helpers.parse_hex_id(parts[1]) if parts.length > 1
+          # Return early if this propagation is not valid
+          return if span_id.nil? || span_id <= 0 || span_id >= Tracing::Utils::EXTERNAL_MAX_ID
+
+          sampling_priority = Helpers.parse_decimal_id(parts[2]) if parts.length > 2
 
           TraceDigest.new(
             span_id: span_id,

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

@@ -1,9 +1,10 @@
 # frozen_string_literal: true
-# typed: true
 
 require_relative '../metadata/ext'
 require_relative '../trace_digest'
 require_relative 'datadog_tags_codec'
+require_relative '../utils'
+require_relative 'helpers'
 
 module Datadog
   module Tracing
@@ -39,21 +40,26 @@ module Datadog
         def inject!(digest, data)
           return if digest.nil?
 
-          data[@trace_id_key] = digest.trace_id.to_s
+          data[@trace_id_key] = Tracing::Utils::TraceId.to_low_order(digest.trace_id).to_s
+
           data[@parent_id_key] = digest.span_id.to_s
           data[@sampling_priority_key] = digest.trace_sampling_priority.to_s if digest.trace_sampling_priority
           data[@origin_key] = digest.trace_origin.to_s if digest.trace_origin
 
-          inject_tags(digest, data)
+          build_tags(digest).tap do |tags|
+            inject_tags!(tags, data) unless tags.empty?
+          end
 
           data
         end
 
         def extract(data)
           fetcher = @fetcher.new(data)
-          trace_id = fetcher.id(@trace_id_key)
-          parent_id = fetcher.id(@parent_id_key)
-          sampling_priority = fetcher.number(@sampling_priority_key)
+
+          trace_id  = parse_trace_id(fetcher)
+          parent_id = parse_parent_id(fetcher)
+
+          sampling_priority = Helpers.parse_decimal_id(fetcher[@sampling_priority_key])
           origin = fetcher[@origin_key]
 
           # Return early if this propagation is not valid
@@ -64,6 +70,10 @@ module Datadog
 
           trace_distributed_tags = extract_tags(fetcher)
 
+          # If trace id is 128 bits long,
+          # Concatentated high order 64 bit hex-encoded `tid` tag
+          trace_id = extract_trace_id!(trace_id, trace_distributed_tags)
+
           TraceDigest.new(
             span_id: parent_id,
             trace_id: trace_id,
@@ -75,20 +85,57 @@ module Datadog
 
         private
 
+        def parse_trace_id(fetcher_object)
+          trace_id = Helpers.parse_decimal_id(fetcher_object[@trace_id_key])
+
+          return unless trace_id
+          return if trace_id <= 0 || trace_id >= Tracing::Utils::EXTERNAL_MAX_ID
+
+          trace_id
+        end
+
+        def parse_parent_id(fetcher_object)
+          parent_id = Helpers.parse_decimal_id(fetcher_object[@parent_id_key])
+
+          return unless parent_id
+          return if parent_id <= 0 || parent_id >= Tracing::Utils::EXTERNAL_MAX_ID
+
+          parent_id
+        end
+
+        def build_tags(digest)
+          high_order = Tracing::Utils::TraceId.to_high_order(digest.trace_id)
+          tags = digest.trace_distributed_tags || {}
+
+          return tags if high_order == 0
+
+          tags.merge(Tracing::Metadata::Ext::Distributed::TAG_TID => high_order.to_s(16))
+        end
+
+        # Side effect: Remove high order 64 bit hex-encoded `tid` tag from distributed tags
+        def extract_trace_id!(trace_id, tags)
+          return trace_id unless tags
+          return trace_id unless (high_order = tags.delete(Tracing::Metadata::Ext::Distributed::TAG_TID))
+
+          Tracing::Utils::TraceId.concatenate(high_order.to_i(16), trace_id)
+        end
+
         # Export trace distributed tags through the `x-datadog-tags` key.
         #
         # 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.
-        def inject_tags(digest, data)
-          return if digest.trace_distributed_tags.nil? || digest.trace_distributed_tags.empty?
+        def inject_tags!(tags, data)
           return set_tags_propagation_error(reason: 'disabled') if tags_disabled?
 
-          tags = DatadogTagsCodec.encode(digest.trace_distributed_tags)
+          encoded_tags = DatadogTagsCodec.encode(tags)
 
-          return set_tags_propagation_error(reason: 'inject_max_size') if tags_too_large?(tags.size, scenario: 'inject')
+          return set_tags_propagation_error(reason: 'inject_max_size') if tags_too_large?(
+            encoded_tags.size,
+            scenario: 'inject'
+          )
 
-          data[@tags_key] = tags
+          data[@tags_key] = encoded_tags
         rescue => e
           set_tags_propagation_error(reason: 'encoding_error')
           ::Datadog.logger.warn(

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

@@ -1,5 +1,4 @@
 # frozen_string_literal: true
-# typed: false
 
 module Datadog
   module Tracing