ddtrace 0.54.0 → 1.1.0

data/lib/ddtrace/contrib/httpclient/configuration/settings.rb

@@ -1,33 +0,0 @@
-# typed: false
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/httpclient/ext'
-
-module Datadog
-  module Contrib
-    module Httpclient
-      module Configuration
-        # Custom settings for the Httpclient integration
-        class Settings < Contrib::Configuration::Settings
-          option :enabled do |o|
-            o.default { env_to_bool(Ext::ENV_ENABLED, true) }
-            o.lazy
-          end
-
-          option :analytics_enabled do |o|
-            o.default { env_to_bool([Ext::ENV_ANALYTICS_ENABLED, Ext::ENV_ANALYTICS_ENABLED_OLD], false) }
-            o.lazy
-          end
-
-          option :analytics_sample_rate do |o|
-            o.default { env_to_float([Ext::ENV_ANALYTICS_SAMPLE_RATE, Ext::ENV_ANALYTICS_SAMPLE_RATE_OLD], 1.0) }
-            o.lazy
-          end
-
-          option :distributed_tracing, default: true
-          option :service_name, default: Ext::SERVICE_NAME
-          option :split_by_domain, default: false
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httpclient/ext.rb

@@ -1,18 +0,0 @@
-# typed: true
-module Datadog
-  module Contrib
-    module Httpclient
-      # Httpclient integration constants
-      module Ext
-        APP = 'httpclient'.freeze
-        ENV_ENABLED = 'DD_TRACE_HTTPCLIENT_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED = 'DD_TRACE_HTTPCLIENT_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED_OLD = 'DD_HTTPCLIENT_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE = 'DD_TRACE_HTTPCLIENT_ANALYTICS_SAMPLE_RATE'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE_OLD = 'DD_HTTPCLIENT_ANALYTICS_SAMPLE_RATE'.freeze
-        SERVICE_NAME = 'httpclient'.freeze
-        SPAN_REQUEST = 'httpclient.request'.freeze
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httpclient/instrumentation.rb

@@ -1,148 +0,0 @@
-# typed: false
-require 'ddtrace/ext/app_types'
-require 'ddtrace/ext/http'
-require 'ddtrace/ext/net'
-require 'ddtrace/ext/distributed'
-require 'ddtrace/contrib/analytics'
-require 'ddtrace/propagation/http_propagator'
-require 'ddtrace/contrib/http_annotation_helper'
-
-module Datadog
-  module Contrib
-    module Httpclient
-      # Instrumentation for Httpclient
-      module Instrumentation
-        def self.included(base)
-          base.prepend(InstanceMethods)
-        end
-
-        # Instance methods for configuration
-        module InstanceMethods
-          include Datadog::Contrib::HttpAnnotationHelper
-
-          def do_get_block(req, proxy, conn, &block)
-            host = req.header.request_uri.host
-            request_options = datadog_configuration(host)
-            pin = datadog_pin(request_options)
-
-            return super unless pin && pin.tracer
-
-            pin.tracer.trace(Ext::SPAN_REQUEST, on_error: method(:annotate_span_with_error!)) do |span|
-              begin
-                request_options[:service_name] = pin.service_name
-                span.service = service_name(host, request_options)
-                span.span_type = Datadog::Ext::HTTP::TYPE_OUTBOUND
-
-                if pin.tracer.enabled && !should_skip_distributed_tracing?(pin)
-                  Datadog::HTTPPropagator.inject!(span.context, req.header)
-                end
-
-                # Add additional request specific tags to the span.
-                annotate_span_with_request!(span, req, request_options)
-              rescue StandardError => e
-                logger.error("error preparing span for httpclient request: #{e}, Source: #{e.backtrace}")
-              ensure
-                res = super
-              end
-
-              # Add additional response specific tags to the span.
-              annotate_span_with_response!(span, res)
-
-              res
-            end
-          end
-
-          private
-
-          def annotate_span_with_request!(span, req, req_options)
-            http_method = req.header.request_method.upcase
-            uri = req.header.request_uri
-
-            span.resource = http_method
-            span.set_tag(Datadog::Ext::HTTP::METHOD, http_method)
-            span.set_tag(Datadog::Ext::HTTP::URL, uri.path)
-            span.set_tag(Datadog::Ext::NET::TARGET_HOST, uri.host)
-            span.set_tag(Datadog::Ext::NET::TARGET_PORT, uri.port)
-
-            # Tag as an external peer service
-            span.set_tag(Datadog::Ext::Integration::TAG_PEER_SERVICE, span.service)
-
-            set_analytics_sample_rate(span, req_options)
-          end
-
-          def annotate_span_with_response!(span, response)
-            return unless response && response.status
-
-            span.set_tag(Datadog::Ext::HTTP::STATUS_CODE, response.status)
-
-            case response.status.to_i
-            when 400...599
-              span.set_error(["Error #{response.status}", response.body])
-            end
-          end
-
-          def annotate_span_with_error!(span, error)
-            span.set_error(error)
-          end
-
-          def datadog_pin(config = Datadog.configuration[:httprb])
-            service = config[:service_name]
-            tracer = config[:tracer]
-
-            @datadog_pin ||= Datadog::Pin.new(
-              service,
-              app: Ext::APP,
-              app_type: Datadog::Ext::HTTP::TYPE_OUTBOUND,
-              tracer: -> { config[:tracer] }
-            )
-
-            if @datadog_pin.service_name == default_datadog_pin.service_name && @datadog_pin.service_name != service
-              @datadog_pin.service = service
-            end
-            if @datadog_pin.tracer == default_datadog_pin.tracer && @datadog_pin.tracer != tracer
-              @datadog_pin.tracer = tracer
-            end
-
-            @datadog_pin
-          end
-
-          def default_datadog_pin
-            config = Datadog.configuration[:httpclient]
-            service = config[:service_name]
-
-            @default_datadog_pin ||= Datadog::Pin.new(
-              service,
-              app: Ext::APP,
-              app_type: Datadog::Ext::HTTP::TYPE_OUTBOUND,
-              tracer: -> { config[:tracer] }
-            )
-          end
-
-          def datadog_configuration(host = :default)
-            Datadog.configuration[:httpclient, host]
-          end
-
-          def analytics_enabled?(request_options)
-            Contrib::Analytics.enabled?(request_options[:analytics_enabled])
-          end
-
-          def logger
-            Datadog.logger
-          end
-
-          def should_skip_distributed_tracing?(pin)
-            return !pin.config[:distributed_tracing] if pin.config && pin.config.key?(:distributed_tracing)
-
-            !Datadog.configuration[:httpclient][:distributed_tracing]
-          end
-
-          def set_analytics_sample_rate(span, request_options)
-            return unless analytics_enabled?(request_options)
-
-            Contrib::Analytics.set_sample_rate(span, request_options[:analytics_sample_rate])
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httpclient/integration.rb

@@ -1,44 +0,0 @@
-# typed: false
-require 'ddtrace/contrib/integration'
-require 'ddtrace/contrib/httpclient/configuration/settings'
-require 'ddtrace/contrib/configuration/resolvers/pattern_resolver'
-require 'ddtrace/contrib/httpclient/patcher'
-
-module Datadog
-  module Contrib
-    module Httpclient
-      # Description of Httpclient integration
-      class Integration
-        include Contrib::Integration
-
-        MINIMUM_VERSION = Gem::Version.new('2.2.0')
-
-        register_as :httpclient
-
-        def self.version
-          Gem.loaded_specs['httpclient'] && Gem.loaded_specs['httpclient'].version
-        end
-
-        def self.loaded?
-          !defined?(::HTTPClient).nil?
-        end
-
-        def self.compatible?
-          super && version >= MINIMUM_VERSION
-        end
-
-        def default_configuration
-          Configuration::Settings.new
-        end
-
-        def patcher
-          Patcher
-        end
-
-        def resolver
-          @resolver ||= Contrib::Configuration::Resolvers::PatternResolver.new
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httpclient/patcher.rb

@@ -1,39 +0,0 @@
-# typed: true
-require 'ddtrace/contrib/patcher'
-require 'ddtrace/contrib/httpclient/instrumentation'
-require 'ddtrace/utils/only_once'
-
-module Datadog
-  module Contrib
-    # Datadog Httpclient integration.
-    module Httpclient
-      # Patcher enables patching of 'httpclient' module.
-      module Patcher
-        include Contrib::Patcher
-
-        PATCH_ONLY_ONCE = Datadog::Utils::OnlyOnce.new
-
-        module_function
-
-        def patched?
-          PATCH_ONLY_ONCE.ran?
-        end
-
-        def target_version
-          Integration.version
-        end
-
-        # patch applies our patch
-        def patch
-          PATCH_ONLY_ONCE.run do
-            begin
-              ::HTTPClient.include(Instrumentation)
-            rescue StandardError => e
-              Datadog.logger.error("Unable to apply httpclient integration: #{e}")
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httprb/configuration/settings.rb

@@ -1,33 +0,0 @@
-# typed: false
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/httprb/ext'
-
-module Datadog
-  module Contrib
-    module Httprb
-      module Configuration
-        # Custom settings for the Httprb integration
-        class Settings < Contrib::Configuration::Settings
-          option :enabled do |o|
-            o.default { env_to_bool(Ext::ENV_ENABLED, true) }
-            o.lazy
-          end
-
-          option :analytics_enabled do |o|
-            o.default { env_to_bool([Ext::ENV_ANALYTICS_ENABLED, Ext::ENV_ANALYTICS_ENABLED_OLD], false) }
-            o.lazy
-          end
-
-          option :analytics_sample_rate do |o|
-            o.default { env_to_float([Ext::ENV_ANALYTICS_SAMPLE_RATE, Ext::ENV_ANALYTICS_SAMPLE_RATE_OLD], 1.0) }
-            o.lazy
-          end
-
-          option :distributed_tracing, default: true
-          option :service_name, default: Ext::SERVICE_NAME
-          option :split_by_domain, default: false
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httprb/ext.rb

@@ -1,18 +0,0 @@
-# typed: true
-module Datadog
-  module Contrib
-    module Httprb
-      # Httprb integration constants
-      module Ext
-        APP = 'httprb'.freeze
-        ENV_ENABLED = 'DD_TRACE_HTTPRB_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED = 'DD_TRACE_HTTPRB_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED_OLD = 'DD_HTTPRB_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE = 'DD_TRACE_EHTTPRB_ANALYTICS_SAMPLE_RATE'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE_OLD = 'DD_HTTPRB_ANALYTICS_SAMPLE_RATE'.freeze
-        SERVICE_NAME = 'httprb'.freeze
-        SPAN_REQUEST = 'httprb.request'.freeze
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httprb/instrumentation.rb

@@ -1,158 +0,0 @@
-# typed: false
-require 'ddtrace/ext/app_types'
-require 'ddtrace/ext/http'
-require 'ddtrace/ext/net'
-require 'ddtrace/ext/distributed'
-require 'ddtrace/contrib/analytics'
-require 'ddtrace/propagation/http_propagator'
-require 'ddtrace/contrib/http_annotation_helper'
-
-module Datadog
-  module Contrib
-    module Httprb
-      # Instrumentation for Httprb
-      module Instrumentation
-        def self.included(base)
-          base.prepend(InstanceMethods)
-        end
-
-        # Instance methods for configuration
-        module InstanceMethods
-          include Datadog::Contrib::HttpAnnotationHelper
-
-          def perform(req, options)
-            host = req.uri.host if req.respond_to?(:uri) && req.uri
-            request_options = datadog_configuration(host)
-            pin = datadog_pin(request_options)
-
-            return super(req, options) unless pin && pin.tracer
-
-            pin.tracer.trace(Ext::SPAN_REQUEST, on_error: method(:annotate_span_with_error!)) do |span|
-              begin
-                request_options[:service_name] = pin.service_name
-                span.service = service_name(host, request_options)
-                span.span_type = Datadog::Ext::HTTP::TYPE_OUTBOUND
-
-                if pin.tracer.enabled && !should_skip_distributed_tracing?(pin)
-                  Datadog::HTTPPropagator.inject!(span.context, req)
-                end
-
-                # Add additional request specific tags to the span.
-                annotate_span_with_request!(span, req, request_options)
-              rescue StandardError => e
-                logger.error("error preparing span for http.rb request: #{e}, Source: #{e.backtrace}")
-              ensure
-                res = super(req, options)
-              end
-
-              # Add additional response specific tags to the span.
-              annotate_span_with_response!(span, res)
-
-              res
-            end
-          end
-
-          private
-
-          def annotate_span_with_request!(span, req, req_options)
-            if req.verb && req.verb.is_a?(String) || req.verb.is_a?(Symbol)
-              http_method = req.verb.to_s.upcase
-              span.resource = http_method
-              span.set_tag(Datadog::Ext::HTTP::METHOD, http_method)
-            else
-              logger.debug("service #{req_options[:service_name]} span #{Ext::SPAN_REQUEST} missing request verb")
-            end
-
-            if req.uri
-              uri = req.uri
-              span.set_tag(Datadog::Ext::HTTP::URL, uri.path)
-              span.set_tag(Datadog::Ext::NET::TARGET_HOST, uri.host)
-              span.set_tag(Datadog::Ext::NET::TARGET_PORT, uri.port)
-            else
-              logger.debug("service #{req_options[:service_name]} span #{Ext::SPAN_REQUEST} missing uri")
-            end
-
-            # Tag as an external peer service
-            span.set_tag(Datadog::Ext::Integration::TAG_PEER_SERVICE, span.service)
-
-            set_analytics_sample_rate(span, req_options)
-          end
-
-          def annotate_span_with_response!(span, response)
-            return unless response && response.code
-
-            span.set_tag(Datadog::Ext::HTTP::STATUS_CODE, response.code)
-
-            case response.code.to_i
-            when 400...599
-              # https://github.com/DataDog/dd-trace-rb/issues/1116
-              # parsing the response body message will alter downstream application behavior
-              span.set_error(["Error #{response.code}", 'Error'])
-            end
-          end
-
-          def annotate_span_with_error!(span, error)
-            span.set_error(error)
-          end
-
-          def datadog_pin(config = Datadog.configuration[:httprb])
-            service = config[:service_name]
-            tracer = config[:tracer]
-
-            @datadog_pin ||= Datadog::Pin.new(
-              service,
-              app: Ext::APP,
-              app_type: Datadog::Ext::HTTP::TYPE_OUTBOUND,
-              tracer: -> { config[:tracer] }
-            )
-
-            if @datadog_pin.service_name == default_datadog_pin.service_name && @datadog_pin.service_name != service
-              @datadog_pin.service = service
-            end
-            if @datadog_pin.tracer == default_datadog_pin.tracer && @datadog_pin.tracer != tracer
-              @datadog_pin.tracer = tracer
-            end
-
-            @datadog_pin
-          end
-
-          def default_datadog_pin
-            config = Datadog.configuration[:httprb]
-            service = config[:service_name]
-
-            @default_datadog_pin ||= Datadog::Pin.new(
-              service,
-              app: Ext::APP,
-              app_type: Datadog::Ext::HTTP::TYPE_OUTBOUND,
-              tracer: -> { config[:tracer] }
-            )
-          end
-
-          def datadog_configuration(host = :default)
-            Datadog.configuration[:httprb, host]
-          end
-
-          def analytics_enabled?(request_options)
-            Contrib::Analytics.enabled?(request_options[:analytics_enabled])
-          end
-
-          def logger
-            Datadog.logger
-          end
-
-          def should_skip_distributed_tracing?(pin)
-            return !pin.config[:distributed_tracing] if pin.config && pin.config.key?(:distributed_tracing)
-
-            !Datadog.configuration[:httprb][:distributed_tracing]
-          end
-
-          def set_analytics_sample_rate(span, request_options)
-            return unless analytics_enabled?(request_options)
-
-            Contrib::Analytics.set_sample_rate(span, request_options[:analytics_sample_rate])
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httprb/integration.rb

@@ -1,44 +0,0 @@
-# typed: false
-require 'ddtrace/contrib/integration'
-require 'ddtrace/contrib/httprb/configuration/settings'
-require 'ddtrace/contrib/configuration/resolvers/pattern_resolver'
-require 'ddtrace/contrib/httprb/patcher'
-
-module Datadog
-  module Contrib
-    module Httprb
-      # Description of Httprb integration
-      class Integration
-        include Contrib::Integration
-
-        MINIMUM_VERSION = Gem::Version.new('2.0.0')
-
-        register_as :httprb
-
-        def self.version
-          Gem.loaded_specs['http'] && Gem.loaded_specs['http'].version
-        end
-
-        def self.loaded?
-          !defined?(::HTTP).nil? && !defined?(::HTTP::Client).nil?
-        end
-
-        def self.compatible?
-          super && version >= MINIMUM_VERSION
-        end
-
-        def default_configuration
-          Configuration::Settings.new
-        end
-
-        def patcher
-          Patcher
-        end
-
-        def resolver
-          @resolver ||= Contrib::Configuration::Resolvers::PatternResolver.new
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/httprb/patcher.rb

@@ -1,39 +0,0 @@
-# typed: true
-require 'ddtrace/contrib/patcher'
-require 'ddtrace/contrib/httprb/instrumentation'
-require 'ddtrace/utils/only_once'
-
-module Datadog
-  module Contrib
-    # Datadog Httprb integration.
-    module Httprb
-      # Patcher enables patching of 'httprb' module.
-      module Patcher
-        include Contrib::Patcher
-
-        PATCH_ONLY_ONCE = Datadog::Utils::OnlyOnce.new
-
-        module_function
-
-        def patched?
-          PATCH_ONLY_ONCE.ran?
-        end
-
-        def target_version
-          Integration.version
-        end
-
-        # patch applies our patch
-        def patch
-          PATCH_ONLY_ONCE.run do
-            begin
-              ::HTTP::Client.include(Instrumentation)
-            rescue StandardError => e
-              Datadog.logger.error("Unable to apply httprb integration: #{e}")
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/integration.rb

@@ -1,17 +0,0 @@
-# typed: true
-require 'ddtrace/contrib/configurable'
-require 'ddtrace/contrib/patchable'
-require 'ddtrace/contrib/registerable'
-
-module Datadog
-  module Contrib
-    # Base provides features that are shared across all integrations
-    module Integration
-      def self.included(base)
-        base.include(Configurable)
-        base.include(Patchable)
-        base.include(Registerable)
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/kafka/configuration/settings.rb

@@ -1,31 +0,0 @@
-# typed: false
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/kafka/ext'
-
-module Datadog
-  module Contrib
-    module Kafka
-      module Configuration
-        # Custom settings for the Kafka integration
-        class Settings < Contrib::Configuration::Settings
-          option :enabled do |o|
-            o.default { env_to_bool(Ext::ENV_ENABLED, true) }
-            o.lazy
-          end
-
-          option :analytics_enabled do |o|
-            o.default { env_to_bool([Ext::ENV_ANALYTICS_ENABLED, Ext::ENV_ANALYTICS_ENABLED_OLD], false) }
-            o.lazy
-          end
-
-          option :analytics_sample_rate do |o|
-            o.default { env_to_float([Ext::ENV_ANALYTICS_SAMPLE_RATE, Ext::ENV_ANALYTICS_SAMPLE_RATE_OLD], 1.0) }
-            o.lazy
-          end
-
-          option :service_name, default: Ext::SERVICE_NAME
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/kafka/consumer_event.rb

@@ -1,15 +0,0 @@
-# typed: true
-module Datadog
-  module Contrib
-    module Kafka
-      # Defines basic behaviors for an event for a consumer.
-      module ConsumerEvent
-        def process(span, _event, _id, payload)
-          super
-
-          span.set_tag(Ext::TAG_GROUP, payload[:group_id])
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/kafka/consumer_group_event.rb

@@ -1,15 +0,0 @@
-# typed: true
-module Datadog
-  module Contrib
-    module Kafka
-      # Defines basic behaviors for an event for a consumer group.
-      module ConsumerGroupEvent
-        def process(span, _event, _id, payload)
-          super
-
-          span.resource = payload[:group_id]
-        end
-      end
-    end
-  end
-end