datadog 2.38.0 → 2.39.0

data/lib/datadog/core/utils/at_fork_monkey_patch.rb

@@ -16,7 +16,7 @@ module Datadog
         def self.apply!
           return false unless supported?
 
-          if RubyVersion.is?('< 3.1')
+          if RubyVersion.is?("< 3.1")
             [
               ::Process.singleton_class, # Process.fork
               ::Kernel.singleton_class,  # Kernel.fork
@@ -39,7 +39,7 @@ module Datadog
 
         def self.at_fork(stage, &block)
           raise(ArgumentError, "Unsupported stage #{stage}") unless stage == :child
-          raise(ArgumentError, 'Missing block argument') unless block
+          raise(ArgumentError, "Missing block argument") unless block
 
           AT_FORK_CHILD_BLOCKS << block
 

data/lib/datadog/core/utils/base64_codec.rb

@@ -7,15 +7,15 @@ module Datadog
       # which is no longer a default gem since Ruby 3.4.
       module Base64Codec
         def self.encode64(bin)
-          [bin].pack('m')
+          [bin].pack("m")
         end
 
         def self.strict_encode64(bin)
-          [bin].pack('m0')
+          [bin].pack("m0")
         end
 
         def self.strict_decode64(str)
-          str.unpack1('m0') #: String # 'm0' format always returns a String
+          str.unpack1("m0") #: String # 'm0' format always returns a String
         end
       end
     end

data/lib/datadog/core/utils/duration.rb

@@ -6,7 +6,7 @@ module Datadog
       # Helper methods for parsing string values into Numeric
       module Duration
         def self.call(value, base: :s)
-          cast = if value.include?('.')
+          cast = if value.include?(".")
             method(:Float)
           else
             method(:Integer)

data/lib/datadog/core/utils/forking.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative '../../ruby_version'
+require_relative "../../ruby_version"
 
 module Datadog
   module Core
@@ -47,7 +47,7 @@ module Datadog
           # object will cause forking to not be detected in the fork when it should have.
           #
           # This wrapper prevents this by initializing the fork PID when the object is created.
-          if RubyVersion.is?('>= 3')
+          if RubyVersion.is?(">= 3")
             def initialize(*args, **kwargs, &block)
               super
               update_fork_pid!

data/lib/datadog/core/utils/lru_cache.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'forwardable'
+require "forwardable"
 
 module Datadog
   module Core
@@ -16,8 +16,8 @@ module Datadog
         def_delegators :@store, :clear, :empty?
 
         def initialize(max_size)
-          raise ArgumentError, 'max_size must be an Integer' unless max_size.is_a?(Integer)
-          raise ArgumentError, 'max_size must be greater than 0' if max_size <= 0
+          raise ArgumentError, "max_size must be an Integer" unless max_size.is_a?(Integer)
+          raise ArgumentError, "max_size must be greater than 0" if max_size <= 0
 
           @max_size = max_size
           @store = {}

data/lib/datadog/core/utils/network.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'ipaddr'
+require "ipaddr"
 
 module Datadog
   module Core
@@ -21,7 +21,7 @@ module Datadog
           cf-connecting-ipv6
         ].freeze
 
-        CGNAT_IP_RANGE = IPAddr.new('100.64.0.0/10')
+        CGNAT_IP_RANGE = IPAddr.new("100.64.0.0/10")
 
         class << self
           # Returns a client IP associated with the request if it was
@@ -75,8 +75,8 @@ module Datadog
 
               next unless value
 
-              ips = value.split(',')
-              ips = process_forwarded_header_values(ips) if name == 'forwarded'
+              ips = value.split(",")
+              ips = process_forwarded_header_values(ips) if name == "forwarded"
 
               ips.each do |ip|
                 parsed_ip = ip_to_ipaddr(ip.strip)
@@ -92,11 +92,11 @@ module Datadog
             values.each_with_object([]) do |value, acc|
               value.downcase!
 
-              value.split(';').each do |tuple_str|
+              value.split(";").each do |tuple_str|
                 tuple_str.strip!
-                next unless tuple_str.start_with?('for=')
+                next unless tuple_str.start_with?("for=")
 
-                tuple_str.delete_prefix!('for=')
+                tuple_str.delete_prefix!("for=")
                 tuple_str.delete!('"')
 
                 acc << tuple_str
@@ -110,14 +110,14 @@ module Datadog
           # The rationale is that in valid IPv6 addresses, colons will always preced dots,
           #   and in valid IPv4 addresses dots will always preced colons.
           def likely_ipv4?(value)
-            dot_index = value.index('.') || value.size
-            colon_index = value.index(':') || value.size
+            dot_index = value.index(".") || value.size
+            colon_index = value.index(":") || value.size
 
             dot_index < colon_index
           end
 
           def strip_zone_specifier(ipv6)
-            ipv6.gsub(/%.*/, '')
+            ipv6.gsub(/%.*/, "")
           end
 
           def strip_ipv6_port(ip)
@@ -129,7 +129,7 @@ module Datadog
           end
 
           def strip_ipv4_port(ip)
-            ip.gsub(/:\d+\z/, '')
+            ip.gsub(/:\d+\z/, "")
           end
 
           def global_ip?(parsed_ip)

data/lib/datadog/core/utils/only_once_successful.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative 'only_once'
+require_relative "only_once"
 
 module Datadog
   module Core

data/lib/datadog/core/utils/safe_dup.rb

@@ -22,7 +22,8 @@ module Datadog
           end
         end
 
-        def self.frozen_dup(v)
+        # Steep: https://github.com/soutaro/steep/issues/2001
+        def self.frozen_dup(v) # steep:ignore MethodBodyTypeMismatch
           # For the case of a String we use the methods -@
           # That method are only for String objects
           # they are faster and chepaer on the memory side.

data/lib/datadog/core/utils/spawn_monkey_patch.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative '../../ruby_version'
+require_relative "../../ruby_version"
 
 module Datadog
   module Core
@@ -21,7 +21,7 @@ module Datadog
         # Prepends `Process.spawn` to merge `env_provider` output into the child's environment hash.
         module ProcessSpawnPatch
           # The One and Only Correct Delegation Pattern
-          if RubyVersion.is?('>= 3')
+          if RubyVersion.is?(">= 3")
             def spawn(*args, **kwargs) # steep:ignore DifferentMethodParameterKind
               super(*SpawnMonkeyPatch.inject_envs(args), **kwargs)
             end

data/lib/datadog/core/utils/url.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'uri'
+require "uri"
 
 module Datadog
   module Core

data/lib/datadog/core/utils.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative 'utils/forking'
+require_relative "utils/forking"
 
 module Datadog
   module Core
@@ -9,7 +9,7 @@ module Datadog
     module Utils
       extend Forking
 
-      EMPTY_STRING = ''.encode(::Encoding::UTF_8).freeze
+      EMPTY_STRING = "".encode(::Encoding::UTF_8).freeze
 
       # Stringifies `value` and ensures the outcome is
       # string is no longer than `size`.
@@ -20,7 +20,7 @@ module Datadog
       # If `omission.size` is larger than `size`, the output
       # will still be `omission.size` in length.
       # @!visibility private
-      def self.truncate(value, size, omission = '...')
+      def self.truncate(value, size, omission = "...")
         string = value.to_s
 
         return string if string.size <= size
@@ -52,7 +52,7 @@ module Datadog
         if binary
           # This option is useful for "gracefully" displaying binary data that
           # often contains text such as marshalled objects
-          str.encode('UTF-8', 'binary', invalid: :replace, undef: :replace, replace: '')
+          str.encode("UTF-8", "binary", invalid: :replace, undef: :replace, replace: "")
         elsif str.encoding == ::Encoding::UTF_8
           str
         elsif str.empty?

data/lib/datadog/core/workers/async.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require_relative '../logger'
+require_relative "../logger"
 
 module Datadog
   module Core

data/lib/datadog/core/workers/polling.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative 'async'
-require_relative 'interval_loop'
+require_relative "async"
+require_relative "interval_loop"
 
 module Datadog
   module Core

data/lib/datadog/core/workers/runtime_metrics.rb

@@ -1,10 +1,10 @@
 # frozen_string_literal: true
 
-require_relative '../runtime/metrics'
+require_relative "../runtime/metrics"
 
-require_relative '../worker'
-require_relative 'async'
-require_relative 'polling'
+require_relative "../worker"
+require_relative "async"
+require_relative "polling"
 
 module Datadog
   module Core

data/lib/datadog/core.rb

@@ -1,11 +1,11 @@
 # frozen_string_literal: true
 
 # Keep this at the top, this is needed at require-time by some files
-require_relative 'ruby_version'
+require_relative "ruby_version"
 
-require_relative 'core/deprecations'
-require_relative 'core/configuration/config_helper'
-require_relative 'core/extensions'
+require_relative "core/deprecations"
+require_relative "core/configuration/config_helper"
+require_relative "core/extensions"
 
 # We must load core extensions to make certain global APIs
 # accessible: both for Datadog features and the core itself.

data/lib/datadog/data_streams/configuration.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative '../core/environment/variable_helpers'
-require_relative 'ext'
+require_relative "../core/environment/variable_helpers"
+require_relative "ext"
 
 module Datadog
   module DataStreams
@@ -38,7 +38,7 @@ module Datadog
               # @!visibility private
               option :interval do |o|
                 o.type :float
-                o.env '_DD_TRACE_STATS_WRITER_INTERVAL'
+                o.env "_DD_TRACE_STATS_WRITER_INTERVAL"
                 o.default 10.0
               end
             end

data/lib/datadog/data_streams/ext.rb

@@ -5,7 +5,7 @@ module Datadog
     # Constants for Data Streams Monitoring configuration
     # @public_api
     module Ext
-      ENV_ENABLED = 'DD_DATA_STREAMS_ENABLED'
+      ENV_ENABLED = "DD_DATA_STREAMS_ENABLED"
     end
   end
 end

data/lib/datadog/data_streams/extensions.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require_relative 'configuration'
-require_relative '../core/configuration'
+require_relative "configuration"
+require_relative "../core/configuration"
 
 module Datadog
   module DataStreams

data/lib/datadog/data_streams/pathway_context.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
-require 'stringio'
-require_relative '../core/utils/base64_codec'
+require "stringio"
+require_relative "../core/utils/base64_codec"
 
 module Datadog
   module DataStreams
@@ -58,7 +58,7 @@ module Datadog
         # - 8 bytes: hash value (little-endian)
         # - VarInt: pathway start time (milliseconds)
         # - VarInt: current edge start time (milliseconds)
-        [@hash].pack('Q') <<
+        [@hash].pack("Q") <<
           encode_var_int_64(time_to_ms(@pathway_start)) <<
           encode_var_int_64(time_to_ms(@current_edge_start))
       end
@@ -73,7 +73,7 @@ module Datadog
         hash_bytes = reader.read(8)
         return nil unless hash_bytes
 
-        hash_value = hash_bytes.unpack1('Q') # : Integer
+        hash_value = hash_bytes.unpack1("Q") # : Integer
 
         # Extract pathway start time (VarInt milliseconds)
         pathway_start_ms = decode_varint(reader)
@@ -116,7 +116,7 @@ module Datadog
           value >>= 7
         end
         bytes << value
-        bytes.pack('C*')
+        bytes.pack("C*")
       end
 
       # Decode an unsigned LEB128 variable-length integer from IO stream.

data/lib/datadog/data_streams/processor.rb

@@ -1,15 +1,15 @@
 # frozen_string_literal: true
 
-require 'zlib'
-require_relative 'pathway_context'
-require_relative 'transport/http'
-require_relative '../version'
-require_relative '../core/worker'
-require_relative '../core/workers/polling'
-require_relative '../core/ddsketch'
-require_relative '../core/buffer/cruby'
-require_relative '../core/utils/time'
-require_relative '../core/utils/fnv'
+require "zlib"
+require_relative "pathway_context"
+require_relative "transport/http"
+require_relative "../version"
+require_relative "../core/worker"
+require_relative "../core/workers/polling"
+require_relative "../core/ddsketch"
+require_relative "../core/buffer/cruby"
+require_relative "../core/utils/time"
+require_relative "../core/utils/fnv"
 
 module Datadog
   module DataStreams
@@ -22,7 +22,7 @@ module Datadog
     class Processor < Core::Worker
       include Core::Workers::Polling
 
-      PROPAGATION_KEY = 'dd-pathway-ctx-base64'
+      PROPAGATION_KEY = "dd-pathway-ctx-base64"
 
       # Default buffer size for lock-free event queue
       # Set to handle high-throughput scenarios (e.g., 10k events/sec for 10s interval)
@@ -41,7 +41,7 @@ module Datadog
       #   (default: DEFAULT_BUFFER_SIZE). Higher values support more throughput but use more memory.
       # @raise [UnsupportedError] if DDSketch is not available on this platform
       def initialize(interval:, logger:, settings:, agent_settings:, agent_info:, buffer_size: DEFAULT_BUFFER_SIZE)
-        raise UnsupportedError, 'DDSketch is not supported' unless Datadog::Core::DDSketch.supported?
+        raise UnsupportedError, "DDSketch is not supported" unless Datadog::Core::DDSketch.supported?
 
         @settings = settings
         @agent_settings = agent_settings
@@ -112,8 +112,8 @@ module Datadog
       # @yield [key, value] Block to inject context into carrier
       # @return [String] Base64 encoded pathway context
       def set_produce_checkpoint(type:, destination:, manual_checkpoint: true, tags: {}, &block)
-        checkpoint_tags = ["type:#{type}", "topic:#{destination}", 'direction:out']
-        checkpoint_tags << 'manual_checkpoint:true' if manual_checkpoint
+        checkpoint_tags = ["type:#{type}", "topic:#{destination}", "direction:out"]
+        checkpoint_tags << "manual_checkpoint:true" if manual_checkpoint
         checkpoint_tags.concat(tags.map { |k, v| "#{k}:#{v}" }) unless tags.empty?
 
         span = Datadog::Tracing.active_span
@@ -140,8 +140,8 @@ module Datadog
           end
         end
 
-        checkpoint_tags = ["type:#{type}", "topic:#{source}", 'direction:in']
-        checkpoint_tags << 'manual_checkpoint:true' if manual_checkpoint
+        checkpoint_tags = ["type:#{type}", "topic:#{source}", "direction:in"]
+        checkpoint_tags << "manual_checkpoint:true" if manual_checkpoint
         checkpoint_tags.concat(tags.map { |k, v| "#{k}:#{v}" }) unless tags.empty?
 
         span = Datadog::Tracing.active_span
@@ -228,7 +228,7 @@ module Datadog
         bucket_time_ns = now_ns - (now_ns % @bucket_size_ns)
         bucket = @buckets[bucket_time_ns] ||= create_bucket
 
-        aggr_key = [event[:tags].join(','), event[:hash], event[:parent_hash]]
+        aggr_key = [event[:tags].join(","), event[:hash], event[:parent_hash]]
         stats = bucket[:pathway_stats][aggr_key] ||= create_pathway_stats
 
         stats[:edge_latency].add(event[:edge_latency_sec])
@@ -247,7 +247,7 @@ module Datadog
 
         direction = nil # : ::String?
         tags.each do |tag|
-          if tag.start_with?('direction:')
+          if tag.start_with?("direction:")
             direction = tag
             break
           end
@@ -272,7 +272,7 @@ module Datadog
         new_hash = compute_pathway_hash(parent_hash, tags)
 
         # Tag the APM span with the pathway hash to link DSM and APM
-        span&.set_tag('pathway.hash', new_hash.to_s)
+        span&.set_tag("pathway.hash", new_hash.to_s)
 
         edge_latency_sec = [now - current_context.current_edge_start, 0.0].max
         full_pathway_latency_sec = [now - current_context.pathway_start, 0.0].max
@@ -316,15 +316,15 @@ module Datadog
         end
 
         payload = {
-          'Env' => @settings.env || 'none',
-          'Service' => @settings.service,
-          'TracerVersion' => Datadog::VERSION::STRING,
-          'Lang' => 'ruby',
-          'Stats' => stats_buckets,
-          'Hostname' => hostname
+          "Env" => @settings.env || "none",
+          "Service" => @settings.service,
+          "TracerVersion" => Datadog::VERSION::STRING,
+          "Lang" => "ruby",
+          "Stats" => stats_buckets,
+          "Hostname" => hostname
         } # : ::Hash[::String, (::String | ::Array[::String])]
 
-        payload['ProcessTags'] = Core::Environment::Process.tags if @settings.experimental_propagate_process_tags_enabled
+        payload["ProcessTags"] = Core::Environment::Process.tags if @settings.experimental_propagate_process_tags_enabled
 
         # Send to agent outside mutex to avoid blocking customer code if agent is slow/hung.
         send_stats_to_agent(payload)
@@ -357,9 +357,9 @@ module Datadog
       end
 
       def decode_and_set_pathway_context(headers)
-        return unless headers && headers['dd-pathway-ctx-base64']
+        return unless headers && headers["dd-pathway-ctx-base64"]
 
-        pathway_ctx = decode_pathway_context(headers['dd-pathway-ctx-base64'])
+        pathway_ctx = decode_pathway_context(headers["dd-pathway-ctx-base64"])
         set_pathway_context(pathway_ctx) if pathway_ctx
       end
 
@@ -369,21 +369,21 @@ module Datadog
       # The hash only needs to be internally consistent:
       # @see Datadog::Core::Environment::AgentInfo#container_tags_checksum
       def compute_pathway_hash(current_hash, tags)
-        service = @settings.service || 'ruby-service'
-        env = @settings.env || 'none'
+        service = @settings.service || "ruby-service"
+        env = @settings.env || "none"
 
         bytes = service.bytes + env.bytes
 
         if @settings.experimental_propagate_process_tags_enabled
           propagation_checksum = @agent_info.propagation_checksum
-          bytes += [propagation_checksum].pack('Q<').bytes if propagation_checksum
+          bytes += [propagation_checksum].pack("Q<").bytes if propagation_checksum
         end
 
         tags.each { |tag| bytes += tag.bytes }
-        byte_string = bytes.pack('C*')
+        byte_string = bytes.pack("C*")
 
         node_hash = Core::Utils::FNV.fnv1_64(byte_string)
-        combined_bytes = [node_hash, current_hash].pack('QQ')
+        combined_bytes = [node_hash, current_hash].pack("QQ")
         Core::Utils::FNV.fnv1_64(combined_bytes)
       end
 
@@ -433,38 +433,38 @@ module Datadog
           bucket_stats = []
           bucket[:pathway_stats].each do |aggr_key, stats|
             edge_tags_str, hash_value, parent_hash = aggr_key
-            edge_tags_array = edge_tags_str.split(',')
+            edge_tags_array = edge_tags_str.split(",")
 
             bucket_stats << {
-              'EdgeTags' => edge_tags_array,
-              'Hash' => hash_value,
-              'ParentHash' => parent_hash,
-              'PathwayLatency' => stats[:full_pathway_latency].encode,
-              'EdgeLatency' => stats[:edge_latency].encode,
+              "EdgeTags" => edge_tags_array,
+              "Hash" => hash_value,
+              "ParentHash" => parent_hash,
+              "PathwayLatency" => stats[:full_pathway_latency].encode,
+              "EdgeLatency" => stats[:edge_latency].encode,
             }
           end
 
           backlogs = []
           bucket[:latest_produce_offsets].each do |key, offset|
-            topic, partition = key.split(':', 2)
+            topic, partition = key.split(":", 2)
             backlogs << {
-              'Tags' => ['type:kafka_produce', "topic:#{topic}", "partition:#{partition}"],
-              'Value' => offset
+              "Tags" => ["type:kafka_produce", "topic:#{topic}", "partition:#{partition}"],
+              "Value" => offset
             }
           end
           bucket[:latest_commit_offsets].each do |key, offset|
-            group, topic, partition = key.split(':', 3)
+            group, topic, partition = key.split(":", 3)
             backlogs << {
-              'Tags' => ['type:kafka_commit', "consumer_group:#{group}", "topic:#{topic}", "partition:#{partition}"],
-              'Value' => offset
+              "Tags" => ["type:kafka_commit", "consumer_group:#{group}", "topic:#{topic}", "partition:#{partition}"],
+              "Value" => offset
             }
           end
 
           serialized_buckets << {
-            'Start' => bucket_time_ns,
-            'Duration' => @bucket_size_ns,
-            'Stats' => bucket_stats,
-            'Backlogs' => backlogs + serialize_consumer_backlogs
+            "Start" => bucket_time_ns,
+            "Duration" => @bucket_size_ns,
+            "Stats" => bucket_stats,
+            "Backlogs" => backlogs + serialize_consumer_backlogs
           }
         end
 
@@ -476,12 +476,12 @@ module Datadog
       def serialize_consumer_backlogs
         @consumer_stats.map do |stat|
           {
-            'Tags' => [
-              'type:kafka_commit',
+            "Tags" => [
+              "type:kafka_commit",
               "topic:#{stat[:topic]}",
               "partition:#{stat[:partition]}"
             ],
-            'Value' => stat[:offset]
+            "Value" => stat[:offset]
           }
         end
       end

data/lib/datadog/data_streams/transport/http/stats.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require_relative '../stats'
-require_relative '../../../core/transport/http/api/endpoint'
-require_relative '../../../core/transport/http/response'
+require_relative "../stats"
+require_relative "../../../core/transport/http/api/endpoint"
+require_relative "../../../core/transport/http/response"
 
 module Datadog
   module DataStreams
@@ -32,10 +32,10 @@ module Datadog
                 env.path = path
                 env.body = env.request.parcel.data
                 if (content_type = env.request.parcel.content_type)
-                  env.headers['content-type'] = content_type
+                  env.headers["content-type"] = content_type
                 end
                 if (content_encoding = env.request.parcel.content_encoding)
-                  env.headers['content-encoding'] = content_encoding
+                  env.headers["content-encoding"] = content_encoding
                 end
 
                 # Send request

data/lib/datadog/data_streams/transport/http.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require_relative '../../core/transport/http'
-require_relative 'http/stats'
-require_relative 'stats'
+require_relative "../../core/transport/http"
+require_relative "http/stats"
+require_relative "stats"
 
 module Datadog
   module DataStreams
@@ -10,7 +10,7 @@ module Datadog
       # HTTP transport for Data Streams Monitoring
       module HTTP
         V01 = Stats::API::Endpoint.new(
-          '/v0.1/pipeline_stats'
+          "/v0.1/pipeline_stats"
         )
 
         module_function
@@ -24,7 +24,7 @@ module Datadog
             agent_settings: agent_settings,
             logger: logger,
           ) do |transport|
-            transport.api 'v0.1', V01, default: true
+            transport.api "v0.1", V01, default: true
 
             # Call block to apply any customization, if provided
             yield(transport) if block_given?