ddtrace 1.4.2 → 1.23.3

data/lib/ddtrace/transport/http/builder.rb

@@ -1,178 +0,0 @@
-# typed: true
-
-require_relative '../../../datadog/core/configuration/agent_settings_resolver'
-require_relative 'adapters/registry'
-require_relative 'api/map'
-require_relative 'api/instance'
-require_relative 'client'
-
-module Datadog
-  module Transport
-    module HTTP
-      # Builds new instances of Transport::HTTP::Client
-      class Builder
-        REGISTRY = Adapters::Registry.new
-
-        attr_reader \
-          :apis,
-          :api_options,
-          :default_adapter,
-          :default_api,
-          :default_headers
-
-        def initialize
-          # Global settings
-          @default_adapter = nil
-          @default_headers = {}
-
-          # Client settings
-          @apis = API::Map.new
-          @default_api = nil
-
-          # API settings
-          @api_options = {}
-
-          yield(self) if block_given?
-        end
-
-        def adapter(config, *args, **kwargs)
-          @default_adapter = case config
-                             when Core::Configuration::AgentSettingsResolver::AgentSettings
-                               registry_klass = REGISTRY.get(config.adapter)
-                               raise UnknownAdapterError, config.adapter if registry_klass.nil?
-
-                               registry_klass.build(config)
-                             when Symbol
-                               registry_klass = REGISTRY.get(config)
-                               raise UnknownAdapterError, config if registry_klass.nil?
-
-                               registry_klass.new(*args, **kwargs)
-                             else
-                               config
-                             end
-        end
-
-        def headers(values = {})
-          @default_headers.merge!(values)
-        end
-
-        # Adds a new API to the client
-        # Valid options:
-        #  - :adapter
-        #  - :default
-        #  - :fallback
-        #  - :headers
-        def api(key, spec, options = {})
-          options = options.dup
-
-          # Copy spec into API map
-          @apis[key] = spec
-
-          # Apply as default API, if specified to do so.
-          @default_api = key if options.delete(:default) || @default_api.nil?
-
-          # Save all other settings for initialization
-          (@api_options[key] ||= {}).merge!(options)
-        end
-
-        def default_api=(key)
-          raise UnknownApiError, key unless @apis.key?(key)
-
-          @default_api = key
-        end
-
-        def to_transport
-          raise NoDefaultApiError if @default_api.nil?
-
-          # DEV: Should not be specific to traces
-          Transport::Traces::Transport.new(to_api_instances, @default_api)
-        end
-
-        def to_api_instances
-          raise NoApisError if @apis.empty?
-
-          @apis.inject(API::Map.new) do |instances, (key, spec)|
-            instances.tap do
-              api_options = @api_options[key].dup
-
-              # Resolve the adapter to use for this API
-              adapter = api_options.delete(:adapter) || @default_adapter
-              raise NoAdapterForApiError, key if adapter.nil?
-
-              # Resolve fallback and merge headers
-              fallback = api_options.delete(:fallback)
-              api_options[:headers] = @default_headers.merge((api_options[:headers] || {}))
-
-              # Add API::Instance with all settings
-              instances[key] = api_instance_class.new(
-                spec,
-                adapter,
-                api_options
-              )
-
-              # Configure fallback, if provided.
-              instances.with_fallbacks(key => fallback) unless fallback.nil?
-            end
-          end
-        end
-
-        def api_instance_class
-          API::Instance
-        end
-
-        # Raised when the API key does not match known APIs.
-        class UnknownApiError < StandardError
-          attr_reader :key
-
-          def initialize(key)
-            @key = key
-          end
-
-          def message
-            "Unknown transport API '#{key}'!"
-          end
-        end
-
-        # Raised when the identifier cannot be matched to an adapter.
-        class UnknownAdapterError < StandardError
-          attr_reader :type
-
-          def initialize(type)
-            @type = type
-          end
-
-          def message
-            "Unknown transport adapter '#{type}'!"
-          end
-        end
-
-        # Raised when an adapter cannot be resolved for an API instance.
-        class NoAdapterForApiError < StandardError
-          attr_reader :key
-
-          def initialize(key)
-            @key = key
-          end
-
-          def message
-            "No adapter resolved for transport API '#{key}'!"
-          end
-        end
-
-        # Raised when built without defining APIs.
-        class NoApisError < StandardError
-          def message
-            'No APIs configured for transport!'
-          end
-        end
-
-        # Raised when client built without defining a default API.
-        class NoDefaultApiError < StandardError
-          def message
-            'No default API configured for transport!'
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/transport/http/client.rb

@@ -1,54 +0,0 @@
-# typed: true
-
-require_relative 'statistics'
-require_relative 'env'
-
-module Datadog
-  module Transport
-    module HTTP
-      # Routes, encodes, and sends tracer data to the trace agent via HTTP.
-      class Client
-        include Transport::HTTP::Statistics
-
-        attr_reader :api
-
-        def initialize(api)
-          @api = api
-        end
-
-        def send_request(request, &block)
-          # Build request into env
-          env = build_env(request)
-
-          # Get responses from API
-          response = yield(api, env)
-
-          # Update statistics
-          update_stats_from_response!(response)
-
-          response
-        rescue StandardError => e
-          message =
-            "Internal error during #{self.class.name} request. Cause: #{e.class.name} #{e.message} " \
-            "Location: #{Array(e.backtrace).first}"
-
-          # Log error
-          if stats.consecutive_errors > 0
-            Datadog.logger.debug(message)
-          else
-            Datadog.logger.error(message)
-          end
-
-          # Update statistics
-          update_stats_from_exception!(e)
-
-          InternalErrorResponse.new(e)
-        end
-
-        def build_env(request)
-          Env.new(request)
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/transport/http/env.rb

@@ -1,58 +0,0 @@
-# typed: true
-
-module Datadog
-  module Transport
-    module HTTP
-      # Data structure for an HTTP request
-      class Env < Hash
-        attr_reader \
-          :request
-
-        def initialize(request, options = nil)
-          @request = request
-          merge!(options) unless options.nil?
-        end
-
-        def verb
-          self[:verb]
-        end
-
-        def verb=(value)
-          self[:verb] = value
-        end
-
-        def path
-          self[:path]
-        end
-
-        def path=(value)
-          self[:path] = value
-        end
-
-        def body
-          self[:body]
-        end
-
-        def body=(value)
-          self[:body] = value
-        end
-
-        def headers
-          self[:headers] ||= {}
-        end
-
-        def headers=(value)
-          self[:headers] = value
-        end
-
-        def form
-          self[:form] ||= {}
-        end
-
-        def form=(value)
-          self[:form] = value
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/transport/http/response.rb

@@ -1,58 +0,0 @@
-# typed: false
-
-require_relative '../response'
-
-module Datadog
-  module Transport
-    module HTTP
-      # Wraps an HTTP response from an adapter.
-      #
-      # Used by endpoints to wrap responses from adapters with
-      # fields or behavior that's specific to that endpoint.
-      module Response
-        def initialize(http_response)
-          @http_response = http_response
-        end
-
-        # (see Datadog::Transport::Response#payload)
-        def payload
-          @http_response.payload
-        end
-
-        # (see Datadog::Transport::Response#internal_error?)
-        def internal_error?
-          @http_response.internal_error?
-        end
-
-        # (see Datadog::Transport::Response#unsupported?)
-        def unsupported?
-          @http_response.unsupported?
-        end
-
-        # (see Datadog::Transport::Response#ok?)
-        def ok?
-          @http_response.ok?
-        end
-
-        # (see Datadog::Transport::Response#not_found?)
-        def not_found?
-          @http_response.not_found?
-        end
-
-        # (see Datadog::Transport::Response#client_error?)
-        def client_error?
-          @http_response.client_error?
-        end
-
-        # (see Datadog::Transport::Response#server_error?)
-        def server_error?
-          @http_response.server_error?
-        end
-
-        def code
-          @http_response.respond_to?(:code) ? @http_response.code : nil
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/transport/http/statistics.rb

@@ -1,45 +0,0 @@
-# typed: true
-
-require_relative '../statistics'
-
-module Datadog
-  module Transport
-    module HTTP
-      # Tracks statistics for HTTP transports
-      module Statistics
-        def self.included(base)
-          base.include(Datadog::Transport::Statistics)
-          base.include(InstanceMethods)
-        end
-
-        # Instance methods for HTTP statistics
-        module InstanceMethods
-          # Decorate metrics for HTTP responses
-          def metrics_for_response(response)
-            super.tap do |metrics|
-              # Add status code tag to api.responses metric
-              if metrics.key?(:api_responses)
-                (metrics[:api_responses].options[:tags] ||= []).tap do |tags|
-                  tags << metrics_tag_value(response.code)
-                end
-              end
-            end
-          end
-
-          private
-
-          # The most common status code on a healthy tracer
-          STATUS_CODE_200 = 'status_code:200'.freeze
-
-          def metrics_tag_value(status_code)
-            if status_code == 200
-              STATUS_CODE_200 # DEV Saves string concatenation/creation for common case
-            else
-              "status_code:#{status_code}"
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/transport/http/traces.rb

@@ -1,146 +0,0 @@
-# typed: false
-
-require 'json'
-
-require_relative '../traces'
-require_relative 'client'
-require_relative 'response'
-require_relative 'api/endpoint'
-require_relative 'api/instance'
-
-module Datadog
-  module Transport
-    module HTTP
-      # HTTP transport behavior for traces
-      module Traces
-        # Response from HTTP transport for traces
-        class Response
-          include HTTP::Response
-          include Datadog::Transport::Traces::Response
-
-          def initialize(http_response, options = {})
-            super(http_response)
-            @service_rates = options.fetch(:service_rates, nil)
-            @trace_count = options.fetch(:trace_count, 0)
-          end
-        end
-
-        # Extensions for HTTP client
-        module Client
-          def send_payload(request)
-            send_request(request) do |api, env|
-              api.send_traces(env)
-            end
-          end
-        end
-
-        module API
-          # Extensions for HTTP API Spec
-          module Spec
-            attr_reader :traces
-
-            def traces=(endpoint)
-              @traces = endpoint
-            end
-
-            def send_traces(env, &block)
-              raise NoTraceEndpointDefinedError, self if traces.nil?
-
-              traces.call(env, &block)
-            end
-
-            def encoder
-              traces.encoder
-            end
-
-            # Raised when traces sent but no traces endpoint is defined
-            class NoTraceEndpointDefinedError < StandardError
-              attr_reader :spec
-
-              def initialize(spec)
-                @spec = spec
-              end
-
-              def message
-                'No trace endpoint is defined for API specification!'
-              end
-            end
-          end
-
-          # Extensions for HTTP API Instance
-          module Instance
-            def send_traces(env)
-              raise TracesNotSupportedError, spec unless spec.is_a?(Traces::API::Spec)
-
-              spec.send_traces(env) do |request_env|
-                call(request_env)
-              end
-            end
-
-            # Raised when traces sent to API that does not support traces
-            class TracesNotSupportedError < StandardError
-              attr_reader :spec
-
-              def initialize(spec)
-                @spec = spec
-              end
-
-              def message
-                'Traces not supported for this API!'
-              end
-            end
-          end
-
-          # Endpoint for submitting trace data
-          class Endpoint < HTTP::API::Endpoint
-            HEADER_CONTENT_TYPE = 'Content-Type'.freeze
-            HEADER_TRACE_COUNT = 'X-Datadog-Trace-Count'.freeze
-            SERVICE_RATE_KEY = 'rate_by_service'.freeze
-
-            attr_reader \
-              :encoder
-
-            def initialize(path, encoder, options = {})
-              super(:post, path)
-              @encoder = encoder
-              @service_rates = options.fetch(:service_rates, false)
-            end
-
-            def service_rates?
-              @service_rates == true
-            end
-
-            def call(env, &block)
-              # Add trace count header
-              env.headers[HEADER_TRACE_COUNT] = env.request.parcel.trace_count.to_s
-
-              # Encode body & type
-              env.headers[HEADER_CONTENT_TYPE] = encoder.content_type
-              env.body = env.request.parcel.data
-
-              # Query for response
-              http_response = super(env, &block)
-
-              # Process the response
-              response_options = { trace_count: env.request.parcel.trace_count }.tap do |options|
-                # Parse service rates, if configured to do so.
-                if service_rates? && !http_response.payload.to_s.empty?
-                  body = JSON.parse(http_response.payload)
-                  options[:service_rates] = body[SERVICE_RATE_KEY] if body.is_a?(Hash) && body.key?(SERVICE_RATE_KEY)
-                end
-              end
-
-              # Build and return a trace response
-              Traces::Response.new(http_response, response_options)
-            end
-          end
-        end
-
-        # Add traces behavior to transport components
-        HTTP::Client.include(Traces::Client)
-        HTTP::API::Spec.include(Traces::API::Spec)
-        HTTP::API::Instance.include(Traces::API::Instance)
-      end
-    end
-  end
-end

data/lib/ddtrace/transport/http.rb

@@ -1,121 +0,0 @@
-# frozen_string_literal: true
-
-# typed: true
-
-require 'uri'
-
-require_relative '../../datadog/core/environment/container'
-require_relative '../../datadog/core/environment/ext'
-require_relative 'ext'
-require_relative 'http/adapters/net'
-require_relative 'http/adapters/test'
-require_relative 'http/adapters/unix_socket'
-require_relative 'http/api'
-require_relative 'http/builder'
-require_relative '../version'
-
-module Datadog
-  module Transport
-    # Namespace for HTTP transport components
-    module HTTP
-      include Kernel # Ensure that kernel methods are always available (https://sorbet.org/docs/error-reference#7003)
-
-      # NOTE: Due to... legacy reasons... This class likes having a default `AgentSettings` instance to fall back to.
-      # Because we generate this instance with an empty instance of `Settings`, the resulting `AgentSettings` below
-      # represents only settings specified via environment variables + the usual defaults.
-      #
-      # DO NOT USE THIS IN NEW CODE, as it ignores any settings specified by users via `Datadog.configure`.
-      DO_NOT_USE_ENVIRONMENT_AGENT_SETTINGS = Datadog::Core::Configuration::AgentSettingsResolver.call(
-        Datadog::Core::Configuration::Settings.new,
-        logger: nil,
-      )
-
-      module_function
-
-      # Builds a new Transport::HTTP::Client
-      def new(&block)
-        Builder.new(&block).to_transport
-      end
-
-      # Builds a new Transport::HTTP::Client with default settings
-      # Pass a block to override any settings.
-      def default(
-        agent_settings: DO_NOT_USE_ENVIRONMENT_AGENT_SETTINGS,
-        **options
-      )
-        new do |transport|
-          transport.adapter(agent_settings)
-          transport.headers default_headers
-
-          apis = API.defaults
-
-          transport.api API::V4, apis[API::V4], fallback: API::V3, default: true
-          transport.api API::V3, apis[API::V3]
-
-          # Apply any settings given by options
-          unless options.empty?
-            transport.default_api = options[:api_version] if options.key?(:api_version)
-            transport.headers options[:headers] if options.key?(:headers)
-          end
-
-          if agent_settings.deprecated_for_removal_transport_configuration_proc
-            agent_settings.deprecated_for_removal_transport_configuration_proc.call(transport)
-          end
-
-          # Call block to apply any customization, if provided
-          yield(transport) if block_given?
-        end
-      end
-
-      def default_headers
-        {
-          Datadog::Transport::Ext::HTTP::HEADER_CLIENT_COMPUTED_TOP_LEVEL => '1',
-          Datadog::Transport::Ext::HTTP::HEADER_META_LANG => Datadog::Core::Environment::Ext::LANG,
-          Datadog::Transport::Ext::HTTP::HEADER_META_LANG_VERSION => Datadog::Core::Environment::Ext::LANG_VERSION,
-          Datadog::Transport::Ext::HTTP::HEADER_META_LANG_INTERPRETER => Datadog::Core::Environment::Ext::LANG_INTERPRETER,
-          Datadog::Transport::Ext::HTTP::HEADER_META_TRACER_VERSION => Datadog::Core::Environment::Ext::TRACER_VERSION
-        }.tap do |headers|
-          # Add container ID, if present.
-          container_id = Datadog::Core::Environment::Container.container_id
-          headers[Datadog::Transport::Ext::HTTP::HEADER_CONTAINER_ID] = container_id unless container_id.nil?
-        end
-      end
-
-      def default_adapter
-        Transport::Ext::HTTP::ADAPTER
-      end
-
-      def default_hostname(logger: Datadog.logger)
-        logger.warn(
-          'Deprecated for removal: Using #default_hostname for configuration is deprecated and will ' \
-          'be removed on a future ddtrace release.'
-        )
-
-        DO_NOT_USE_ENVIRONMENT_AGENT_SETTINGS.hostname
-      end
-
-      def default_port(logger: Datadog.logger)
-        logger.warn(
-          'Deprecated for removal: Using #default_hostname for configuration is deprecated and will ' \
-          'be removed on a future ddtrace release.'
-        )
-
-        DO_NOT_USE_ENVIRONMENT_AGENT_SETTINGS.port
-      end
-
-      def default_url(logger: Datadog.logger)
-        logger.warn(
-          'Deprecated for removal: Using #default_url for configuration is deprecated and will ' \
-          'be removed on a future ddtrace release.'
-        )
-
-        nil
-      end
-
-      # Add adapters to registry
-      Builder::REGISTRY.set(Adapters::Net, Transport::Ext::HTTP::ADAPTER)
-      Builder::REGISTRY.set(Adapters::Test, Transport::Ext::Test::ADAPTER)
-      Builder::REGISTRY.set(Adapters::UnixSocket, Transport::Ext::UnixSocket::ADAPTER)
-    end
-  end
-end