datadog 2.38.0 → 2.39.0

data/lib/datadog/tracing/remote.rb

@@ -1,14 +1,14 @@
 # frozen_string_literal: true
 
-require_relative '../core/remote/dispatcher'
-require_relative 'configuration/dynamic'
+require_relative "../core/remote/dispatcher"
+require_relative "configuration/dynamic"
 
 module Datadog
   module Tracing
     # Remote configuration declaration
     module Remote
       class << self
-        PRODUCT = 'APM_TRACING'
+        PRODUCT = "APM_TRACING"
 
         CAPABILITIES = [
           1 << 12, # APM_TRACING_SAMPLE_RATE: Dynamic trace sampling rate configuration
@@ -29,7 +29,7 @@ module Datadog
         end
 
         def process_config(config, content, repository = nil)
-          lib_config = config['lib_config']
+          lib_config = config["lib_config"]
 
           env_vars = Datadog::Tracing::Configuration::Dynamic::OPTIONS.map do |name, env_var, option|
             value = lib_config[name]
@@ -42,25 +42,32 @@ module Datadog
             [env_var, value]
           end
 
-          if (di_enabled = lib_config['dynamic_instrumentation_enabled']) != nil # rubocop:disable Style/NonNilCheck
+          if (di_enabled = lib_config["dynamic_instrumentation_enabled"]) != nil # rubocop:disable Style/NonNilCheck
             # repository is forwarded so that an enable signal can reconcile DI
             # against probes delivered in an earlier poll while DI was stopped
             # (see Datadog::DI::Remote.handle_rc_enablement).
             Datadog::DI::Remote.handle_rc_enablement(di_enabled, repository)
 
+            components = Datadog.send(:components, allow_initialization: false)
+            di_products = Datadog::DI::Remote.products +
+              Datadog::SymbolDatabase::Remote.deferred_products(Datadog.configuration)
+
             if di_enabled
-              # DI was just (implicitly) enabled. A Symbol Database upload signal
-              # received in an earlier poll while DI was inactive was deferred by
-              # the component's DI-active gate; re-attempt it now. Mirrors the DI
-              # probe replay in handle_rc_enablement above. allow_initialization:
-              # false because this runs on the remote-config worker thread.
-              Datadog.send(:components, allow_initialization: false)&.symbol_database&.resume_pending_upload
+              components&.symbol_database&.resume_pending_upload
+              # Advertise the DI products only if the component actually started.
+              # handle_rc_enablement above no-ops when DI cannot run — the
+              # component is nil on an unsupported runtime, or the enable signal
+              # is blocked by DD_DYNAMIC_INSTRUMENTATION_ENABLED=false. Advertising
+              # then would report DI as in use when it is not and invite probe
+              # configs the tracer must refuse; withdraw the products otherwise.
+              if components&.dynamic_instrumentation&.started?
+                components&.remote&.add_products(*di_products)
+              else
+                components&.remote&.remove_products(*di_products)
+              end
             else
-              # DI was disabled via remote configuration. In the nil-default
-              # (follows-DI) case, stop Symbol Database too so its TracePoint and
-              # scheduler don't keep uploading while DI is off. An explicit
-              # symbol_database.enabled = true is independent and keeps running.
-              Datadog.send(:components, allow_initialization: false)&.symbol_database&.stop_for_di_disable
+              components&.symbol_database&.stop_for_di_disable
+              components&.remote&.remove_products(*di_products)
             end
           end
 

data/lib/datadog/tracing/sampling/all_sampler.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative 'sampler'
+require_relative "sampler"
 
 module Datadog
   module Tracing

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

@@ -37,22 +37,22 @@ module Datadog
         # this tag is currently formatted as `"-sampling_mechanism"`.
         module Decision
           # Used before the tracer receives any rates from agent and there are no rules configured.
-          DEFAULT = '-0'
+          DEFAULT = "-0"
           # The sampling rate received in the agent's http response.
-          AGENT_RATE = '-1'
+          AGENT_RATE = "-1"
           # Locally configured rule.
-          TRACE_SAMPLING_RULE = '-3'
+          TRACE_SAMPLING_RULE = "-3"
           # User directly sets sampling priority via {Tracing.reject!} or {Tracing.keep!},
           # or by a custom sampler implementation.
-          MANUAL = '-4'
+          MANUAL = "-4"
           # Formerly AppSec.
-          ASM = '-5'
+          ASM = "-5"
           # AI Guard.
-          AI_GUARD = '-13'
+          AI_GUARD = "-13"
           # Dynamically configured rule, explicitly created by the user.
-          REMOTE_USER_RULE = '-11'
+          REMOTE_USER_RULE = "-11"
           # Dynamically configured  rule, automatically generated by Datadog.
-          REMOTE_DYNAMIC_RULE = '-12'
+          REMOTE_DYNAMIC_RULE = "-12"
         end
       end
     end

data/lib/datadog/tracing/sampling/matcher.rb

@@ -7,7 +7,7 @@ module Datadog
       # @abstract
       class Matcher
         # Pattern that matches any string
-        MATCH_ALL_PATTERN = '*'
+        MATCH_ALL_PATTERN = "*"
 
         # Returns `true` if the trace should conforms to this rule, `false` otherwise
         #
@@ -35,8 +35,8 @@ module Datadog
 
           # Our valid special characters, `?` and `*`, were just escaped
           # by `Regexp.quote` above. We need to unescape them:
-          glob.gsub!('\?', '.') # Any single character
-          glob.gsub!('\*', '.*') # Any substring
+          glob.gsub!('\?', ".") # Any single character
+          glob.gsub!('\*', ".*") # Any substring
 
           # Patterns have to match the whole input string
           glob = "\\A#{glob}\\z"
@@ -108,7 +108,7 @@ module Datadog
             # Format metrics as strings, to allow for partial number matching (/4.*/ matching '400', '404', etc.).
             # Because metrics are floats, we use the '%g' format specifier to avoid trailing zeros, which
             # can affect exact string matching (e.g. '400' matching '400.0').
-            tag = format('%g', tag) if tag.is_a?(Numeric)
+            tag = format("%g", tag) if tag.is_a?(Numeric)
 
             matcher.match?(tag)
           end

data/lib/datadog/tracing/sampling/priority_sampler.rb

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
 
-require_relative 'ext'
-require_relative 'all_sampler'
-require_relative 'rate_sampler'
-require_relative 'rate_by_service_sampler'
+require_relative "ext"
+require_relative "all_sampler"
+require_relative "rate_sampler"
+require_relative "rate_by_service_sampler"
 
 module Datadog
   module Tracing

data/lib/datadog/tracing/sampling/rate_by_key_sampler.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative 'sampler'
-require_relative 'rate_sampler'
+require_relative "sampler"
+require_relative "rate_sampler"
 
 module Datadog
   module Tracing
@@ -14,7 +14,7 @@ module Datadog
         def initialize(default_key, default_rate = 1.0, decision: nil, &block)
           super()
 
-          raise ArgumentError, 'No resolver given!' unless block
+          raise ArgumentError, "No resolver given!" unless block
 
           @default_key = default_key
           @resolver = block

data/lib/datadog/tracing/sampling/rate_by_service_sampler.rb

@@ -1,13 +1,13 @@
 # frozen_string_literal: true
 
-require_relative 'rate_by_key_sampler'
+require_relative "rate_by_key_sampler"
 
 module Datadog
   module Tracing
     module Sampling
       # {Datadog::Tracing::Sampling::RateByServiceSampler} samples different services at different rates
       class RateByServiceSampler < RateByKeySampler
-        DEFAULT_KEY = 'service:,env:'
+        DEFAULT_KEY = "service:,env:"
 
         def initialize(default_rate = 1.0, env: nil, decision: Datadog::Tracing::Sampling::Ext::Decision::DEFAULT)
           super(

data/lib/datadog/tracing/sampling/rate_sampler.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative 'sampler'
-require_relative '../../core/knuth_sampler'
+require_relative "sampler"
+require_relative "../../core/knuth_sampler"
 
 module Datadog
   module Tracing

data/lib/datadog/tracing/sampling/rule.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require_relative 'matcher'
-require_relative 'rate_sampler'
-require_relative '../../core/telemetry/logger'
+require_relative "matcher"
+require_relative "rate_sampler"
+require_relative "../../core/telemetry/logger"
 
 module Datadog
   module Tracing
@@ -36,7 +36,7 @@ module Datadog
           Datadog.logger.error(
             "Matcher failed. Cause: #{e.class}: #{e.message} Source: #{Array(e.backtrace).first}"
           )
-          Datadog::Core::Telemetry::Logger.report(e, description: 'Matcher failed')
+          Datadog::Core::Telemetry::Logger.report(e, description: "Matcher failed")
           nil
         end
 

data/lib/datadog/tracing/sampling/rule_sampler.rb

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
 
-require_relative 'ext'
-require_relative '../../core/rate_limiter'
-require_relative 'rule'
-require_relative '../../core/telemetry/logger'
+require_relative "ext"
+require_relative "../../core/rate_limiter"
+require_relative "rule"
+require_relative "../../core/telemetry/logger"
 
 module Datadog
   module Tracing
@@ -48,14 +48,14 @@ module Datadog
             nil
           else
             # TODO: Simplify .tags access, as `Tracer#tags` can't be arbitrarily changed anymore
-            RateByServiceSampler.new(1.0, env: -> { Tracing.send(:tracer).tags['env'] })
+            RateByServiceSampler.new(1.0, env: -> { Tracing.send(:tracer).tags["env"] })
           end
           @reconsider_sample_resource_enabled = @rules.any? { |rule| resource_rule?(rule) }
         end
 
         def self.parse(rules, rate_limit, default_sample_rate)
           parsed_rules = JSON.parse(rules).map do |rule|
-            sample_rate = rule['sample_rate']
+            sample_rate = rule["sample_rate"]
 
             begin
               sample_rate = Float(sample_rate)
@@ -64,12 +64,12 @@ module Datadog
             end
 
             kwargs = {
-              name: rule['name'],
-              service: rule['service'],
-              resource: rule['resource'],
-              tags: rule['tags'],
+              name: rule["name"],
+              service: rule["service"],
+              resource: rule["resource"],
+              tags: rule["tags"],
               sample_rate: sample_rate,
-              provenance: if (provenance = rule['provenance'])
+              provenance: if (provenance = rule["provenance"])
                             # `Rule::PROVENANCE_*` values are symbols, so convert strings to match
                             provenance.to_sym
                           else
@@ -135,7 +135,7 @@ module Datadog
           Datadog.logger.error(
             "Rule sampling failed. Cause: #{e.class}: #{e.message} Source: #{Array(e.backtrace).first}"
           )
-          Datadog::Core::Telemetry::Logger.report(e, description: 'Rule sampling failed')
+          Datadog::Core::Telemetry::Logger.report(e, description: "Rule sampling failed")
 
           yield(trace)
         end

data/lib/datadog/tracing/sampling/sampler.rb

@@ -15,7 +15,7 @@ module Datadog
         # @param [Datadog::Tracing::TraceOperation] trace
         # @return [Boolean] should this trace be kept?
         def sample!(trace)
-          raise NotImplementedError, 'Samplers must implement the #sample! method'
+          raise NotImplementedError, "Samplers must implement the #sample! method"
         end
 
         # The sampling rate, if this sampler has such concept.
@@ -24,7 +24,7 @@ module Datadog
         # @param [Datadog::Tracing::TraceOperation] trace
         # @return [Float,nil] sampling ratio between 0.0 and 1.0 (inclusive), or `nil` if not applicable
         def sample_rate(trace)
-          raise NotImplementedError, 'Samplers must implement the #sample_rate method'
+          raise NotImplementedError, "Samplers must implement the #sample_rate method"
         end
       end
     end

data/lib/datadog/tracing/sampling/span/ext.rb

@@ -13,11 +13,11 @@ module Datadog
           DEFAULT_MAX_PER_SECOND = -1
 
           # Sampling decision method used to come to the sampling decision for this span
-          TAG_MECHANISM = '_dd.span_sampling.mechanism'
+          TAG_MECHANISM = "_dd.span_sampling.mechanism"
           # Sampling rate applied to this span, if a rule applies
-          TAG_RULE_RATE = '_dd.span_sampling.rule_rate'
+          TAG_RULE_RATE = "_dd.span_sampling.rule_rate"
           # Rate limit configured for this span, if a rule applies
-          TAG_MAX_PER_SECOND = '_dd.span_sampling.max_per_second'
+          TAG_MAX_PER_SECOND = "_dd.span_sampling.max_per_second"
         end
       end
     end

data/lib/datadog/tracing/sampling/span/matcher.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative '../matcher'
+require_relative "../matcher"
 
 module Datadog
   module Tracing
@@ -11,7 +11,7 @@ module Datadog
           attr_reader :name, :service
 
           # Pattern that matches any string
-          MATCH_ALL_PATTERN = '*'
+          MATCH_ALL_PATTERN = "*"
 
           # Matches span name and service to their respective patterns provided.
           #

data/lib/datadog/tracing/sampling/span/rule.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative 'ext'
+require_relative "ext"
 
 module Datadog
   module Tracing

data/lib/datadog/tracing/sampling/span/rule_parser.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require 'json'
+require "json"
 
-require_relative 'ext'
-require_relative 'matcher'
-require_relative 'rule'
+require_relative "ext"
+require_relative "matcher"
+require_relative "rule"
 
 module Datadog
   module Tracing
@@ -75,22 +75,22 @@ module Datadog
 
             def parse_rule(hash)
               matcher_options = {}
-              if (name_pattern = hash['name'])
+              if (name_pattern = hash["name"])
                 matcher_options[:name_pattern] = name_pattern
               end
 
-              if (service_pattern = hash['service'])
+              if (service_pattern = hash["service"])
                 matcher_options[:service_pattern] = service_pattern
               end
 
               matcher = Matcher.new(**matcher_options)
 
               rule_options = {}
-              if (sample_rate = hash['sample_rate'])
+              if (sample_rate = hash["sample_rate"])
                 rule_options[:sample_rate] = sample_rate
               end
 
-              if (max_per_second = hash['max_per_second'])
+              if (max_per_second = hash["max_per_second"])
                 rule_options[:rate_limit] = max_per_second
               end
 

data/lib/datadog/tracing/span.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require_relative '../core/utils/safe_dup'
-require_relative 'utils'
+require_relative "../core/utils/safe_dup"
+require_relative "utils"
 
-require_relative 'metadata/ext'
-require_relative 'metadata'
+require_relative "metadata/ext"
+require_relative "metadata"
 
 module Datadog
   module Tracing
@@ -143,7 +143,7 @@ module Datadog
       # TODO: Change this to reflect attributes when serialization
       # isn't handled by this method.
       def to_hash
-        @meta['events'] = @events.map(&:to_hash).to_json unless @events.empty?
+        @meta["events"] = @events.map(&:to_hash).to_json unless @events.empty?
 
         h = {
           error: @status,
@@ -185,19 +185,19 @@ module Datadog
           q.text "Start: #{start_time}\n"
           q.text "End: #{end_time}\n"
           q.text "Duration: #{duration.to_f}\n"
-          q.group(2, 'Tags: [', "]\n") do
+          q.group(2, "Tags: [", "]\n") do
             q.breakable
             q.seplist @meta.each do |key, value|
               q.text "#{key} => #{value}"
             end
           end
-          q.group(2, 'Metrics: [', "]\n") do
+          q.group(2, "Metrics: [", "]\n") do
             q.breakable
             q.seplist @metrics.each do |key, value|
               q.text "#{key} => #{value}"
             end
           end
-          q.group(2, 'Metastruct: [', ']') do
+          q.group(2, "Metastruct: [", "]") do
             metastruct.pretty_print(q)
           end
         end

data/lib/datadog/tracing/span_event.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'time'
+require "time"
 
 module Datadog
   module Tracing
@@ -41,8 +41,8 @@ module Datadog
       # of being limiting span events to the size limit of a span tag.
       # All Datadog agents support this format.
       def to_hash
-        h = {'name' => @name, 'time_unix_nano' => @time_unix_nano}
-        h['attributes'] = @attributes unless @attributes.empty?
+        h = {"name" => @name, "time_unix_nano" => @time_unix_nano}
+        h["attributes"] = @attributes unless @attributes.empty?
         h
       end
 
@@ -51,7 +51,7 @@ module Datadog
       # This serialization format removes the serialization limitations of the `span.set_tag('events)` approach,
       # but is only supported by newer version of the Datadog agent.
       def to_native_format
-        h = {'name' => @name, 'time_unix_nano' => @time_unix_nano}
+        h = {"name" => @name, "time_unix_nano" => @time_unix_nano}
 
         attr = {}
         @attributes.each do |key, value|
@@ -62,7 +62,7 @@ module Datadog
           end
         end
 
-        h['attributes'] = attr unless @attributes.empty?
+        h["attributes"] = attr unless @attributes.empty?
 
         h
       end

data/lib/datadog/tracing/span_operation.rb

@@ -1,19 +1,19 @@
 # frozen_string_literal: true
 
-require 'time'
+require "time"
 
-require_relative '../core/environment/identity'
-require_relative '../core/utils'
-require_relative '../core/utils/time'
-require_relative '../core/utils/safe_dup'
+require_relative "../core/environment/identity"
+require_relative "../core/utils"
+require_relative "../core/utils/time"
+require_relative "../core/utils/safe_dup"
 
-require_relative 'event'
-require_relative 'metadata'
-require_relative 'metadata/ext'
-require_relative 'span'
-require_relative 'span_event'
-require_relative 'span_link'
-require_relative 'utils'
+require_relative "event"
+require_relative "metadata"
+require_relative "metadata/ext"
+require_relative "span"
+require_relative "span_event"
+require_relative "span_link"
+require_relative "utils"
 
 module Datadog
   module Tracing
@@ -146,7 +146,7 @@ module Datadog
       end
 
       def measure
-        raise ArgumentError, 'Must provide block to measure!' unless block_given?
+        raise ArgumentError, "Must provide block to measure!" unless block_given?
         # TODO: Should we just invoke the block and skip tracing instead?
         raise AlreadyStartedError if started?
 
@@ -318,14 +318,14 @@ module Datadog
         exc = Core::Error.build_from(exception)
 
         event_attributes = {
-          'exception.type' => exc.type,
-          'exception.message' => exc.message,
-          'exception.stacktrace' => exc.backtrace,
+          "exception.type" => exc.type,
+          "exception.message" => exc.message,
+          "exception.stacktrace" => exc.backtrace,
         }
 
         # Steep: Caused by wrong declaration, should be the same parameters as `merge`
         # https://github.com/ruby/rbs/blob/3d0fb3a7fdde60af7120e875fe3bd7237b5b6a88/core/hash.rbs#L1468
-        @span_events << SpanEvent.new('exception', attributes: event_attributes.merge!(attributes)) # steep:ignore ArgumentTypeMismatch
+        @span_events << SpanEvent.new("exception", attributes: event_attributes.merge!(attributes)) # steep:ignore ArgumentTypeMismatch
       end
 
       # Return a string representation of the span.
@@ -374,19 +374,19 @@ module Datadog
           q.text "Start: #{start_time}\n"
           q.text "End: #{end_time}\n"
           q.text "Duration: #{duration.to_f if stopped?}\n"
-          q.group(2, 'Tags: [', "]\n") do
+          q.group(2, "Tags: [", "]\n") do
             q.breakable
             q.seplist meta.each do |key, value|
               q.text "#{key} => #{value}"
             end
           end
-          q.group(2, 'Metrics: [', "]\n") do
+          q.group(2, "Metrics: [", "]\n") do
             q.breakable
             q.seplist metrics.each do |key, value|
               q.text "#{key} => #{value}"
             end
           end
-          q.group(2, 'Metastruct: [', ']') do
+          q.group(2, "Metastruct: [", "]") do
             metastruct.pretty_print(q)
           end
         end
@@ -496,7 +496,7 @@ module Datadog
       # Error when the span attempts to start again after being started
       class AlreadyStartedError < StandardError
         def message
-          'Cannot measure an already started span!'
+          "Cannot measure an already started span!"
         end
       end
 

data/lib/datadog/tracing/sync_writer.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require_relative 'pipeline'
-require_relative 'runtime/metrics'
-require_relative 'writer'
+require_relative "pipeline"
+require_relative "runtime/metrics"
+require_relative "writer"
 
-require_relative 'transport/http'
+require_relative "transport/http"
 
 module Datadog
   module Tracing

data/lib/datadog/tracing/trace_operation.rb

@@ -1,15 +1,15 @@
 # frozen_string_literal: true
 
-require_relative '../core/environment/identity'
-require_relative '../core/utils'
-require_relative 'event'
-require_relative 'metadata/tagging'
-require_relative 'sampling/ext'
-require_relative 'span_operation'
-require_relative 'trace_digest'
-require_relative 'correlation'
-require_relative 'trace_segment'
-require_relative 'utils'
+require_relative "../core/environment/identity"
+require_relative "../core/utils"
+require_relative "event"
+require_relative "metadata/tagging"
+require_relative "sampling/ext"
+require_relative "span_operation"
+require_relative "trace_digest"
+require_relative "correlation"
+require_relative "trace_segment"
+require_relative "utils"
 
 module Datadog
   module Tracing
@@ -206,7 +206,7 @@ module Datadog
       def set_distributed_source(product_bit)
         source = get_tag(Metadata::Ext::Distributed::TAG_TRACE_SOURCE)&.to_i(16) || 0
         source |= product_bit
-        set_tag(Metadata::Ext::Distributed::TAG_TRACE_SOURCE, format('%02X', source))
+        set_tag(Metadata::Ext::Distributed::TAG_TRACE_SOURCE, format("%02X", source))
       end
 
       def tags