ddtrace 0.38.0

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

@@ -0,0 +1,26 @@
+require 'ddtrace/contrib/configuration/settings'
+require 'ddtrace/contrib/rest_client/ext'
+
+module Datadog
+  module Contrib
+    module RestClient
+      module Configuration
+        # Custom settings for the RestClient integration
+        class Settings < Contrib::Configuration::Settings
+          option :analytics_enabled do |o|
+            o.default { env_to_bool(Ext::ENV_ANALYTICS_ENABLED, false) }
+            o.lazy
+          end
+
+          option :analytics_sample_rate do |o|
+            o.default { env_to_float(Ext::ENV_ANALYTICS_SAMPLE_RATE, 1.0) }
+            o.lazy
+          end
+
+          option :distributed_tracing, default: true
+          option :service_name, default: Ext::SERVICE_NAME
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,14 @@
+module Datadog
+  module Contrib
+    module RestClient
+      # RestClient integration constants
+      module Ext
+        APP = 'rest_client'.freeze
+        ENV_ANALYTICS_ENABLED = 'DD_REST_CLIENT_ANALYTICS_ENABLED'.freeze
+        ENV_ANALYTICS_SAMPLE_RATE = 'DD_REST_CLIENT_ANALYTICS_SAMPLE_RATE'.freeze
+        SERVICE_NAME = 'rest_client'.freeze
+        SPAN_REQUEST = 'rest_client.request'.freeze
+      end
+    end
+  end
+end

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

@@ -0,0 +1,38 @@
+require 'ddtrace/contrib/integration'
+require 'ddtrace/contrib/rest_client/configuration/settings'
+require 'ddtrace/contrib/rest_client/patcher'
+
+module Datadog
+  module Contrib
+    module RestClient
+      # Description of RestClient integration
+      class Integration
+        include Contrib::Integration
+
+        MINIMUM_VERSION = Gem::Version.new('1.8')
+
+        register_as :rest_client
+
+        def self.version
+          Gem.loaded_specs['rest-client'] && Gem.loaded_specs['rest-client'].version
+        end
+
+        def self.loaded?
+          !defined?(::RestClient::Request).nil?
+        end
+
+        def self.compatible?
+          super && version >= MINIMUM_VERSION
+        end
+
+        def default_configuration
+          Configuration::Settings.new
+        end
+
+        def patcher
+          Patcher
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+module Datadog
+  module Contrib
+    module RestClient
+      # Patcher enables patching of 'rest_client' module.
+      module Patcher
+        include Contrib::Patcher
+
+        module_function
+
+        def target_version
+          Integration.version
+        end
+
+        def patch
+          require 'ddtrace/ext/app_types'
+          require 'ddtrace/contrib/rest_client/request_patch'
+
+          ::RestClient::Request.send(:include, RequestPatch)
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/rest_client/request_patch.rb

@@ -0,0 +1,89 @@
+require 'ddtrace/ext/net'
+require 'ddtrace/ext/distributed'
+require 'ddtrace/propagation/http_propagator'
+require 'ddtrace/contrib/rest_client/ext'
+
+module Datadog
+  module Contrib
+    module RestClient
+      # RestClient RequestPatch
+      module RequestPatch
+        def self.included(base)
+          base.send(:prepend, InstanceMethods)
+        end
+
+        # InstanceMethods - implementing instrumentation
+        module InstanceMethods
+          def execute(&block)
+            uri = URI.parse(url)
+
+            return super(&block) unless datadog_configuration[:tracer].enabled
+
+            datadog_trace_request(uri) do |span|
+              if datadog_configuration[:distributed_tracing]
+                Datadog::HTTPPropagator.inject!(span.context, processed_headers)
+              end
+
+              super(&block)
+            end
+          end
+
+          def datadog_tag_request(uri, span)
+            span.resource = method.to_s.upcase
+
+            # Set analytics sample rate
+            Contrib::Analytics.set_sample_rate(span, analytics_sample_rate) if analytics_enabled?
+
+            span.set_tag(Datadog::Ext::HTTP::URL, uri.path)
+            span.set_tag(Datadog::Ext::HTTP::METHOD, method.to_s.upcase)
+            span.set_tag(Datadog::Ext::NET::TARGET_HOST, uri.host)
+            span.set_tag(Datadog::Ext::NET::TARGET_PORT, uri.port)
+          end
+
+          def datadog_trace_request(uri)
+            span = datadog_configuration[:tracer].trace(Ext::SPAN_REQUEST,
+                                                        service: datadog_configuration[:service_name],
+                                                        span_type: Datadog::Ext::HTTP::TYPE_OUTBOUND)
+
+            datadog_tag_request(uri, span)
+
+            yield(span).tap do |response|
+              # Verify return value is a response
+              # If so, add additional tags.
+              if response.is_a?(::RestClient::Response)
+                span.set_tag(Datadog::Ext::HTTP::STATUS_CODE, response.code)
+              end
+            end
+          rescue ::RestClient::ExceptionWithResponse => e
+            span.set_error(e) if Datadog::Ext::HTTP::ERROR_RANGE.cover?(e.http_code)
+            span.set_tag(Datadog::Ext::HTTP::STATUS_CODE, e.http_code)
+
+            raise e
+            # rubocop:disable Lint/RescueException
+          rescue Exception => e
+            # rubocop:enable Lint/RescueException
+            span.set_error(e) if span
+
+            raise e
+          ensure
+            span.finish if span
+          end
+
+          private
+
+          def datadog_configuration
+            Datadog.configuration[:rest_client]
+          end
+
+          def analytics_enabled?
+            Contrib::Analytics.enabled?(datadog_configuration[:analytics_enabled])
+          end
+
+          def analytics_sample_rate
+            datadog_configuration[:analytics_sample_rate]
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+require 'ddtrace/contrib/configuration/settings'
+require 'ddtrace/contrib/sequel/ext'
+
+module Datadog
+  module Contrib
+    module Sequel
+      module Configuration
+        # Custom settings for the Sequel integration
+        class Settings < Contrib::Configuration::Settings
+          option :analytics_enabled do |o|
+            o.default { env_to_bool(Ext::ENV_ANALYTICS_ENABLED, false) }
+            o.lazy
+          end
+
+          option :analytics_sample_rate do |o|
+            o.default { env_to_float(Ext::ENV_ANALYTICS_SAMPLE_RATE, 1.0) }
+            o.lazy
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/sequel/database.rb

@@ -0,0 +1,61 @@
+require 'ddtrace/ext/sql'
+require 'ddtrace/ext/app_types'
+require 'ddtrace/contrib/analytics'
+require 'ddtrace/contrib/sequel/ext'
+require 'ddtrace/contrib/sequel/utils'
+
+module Datadog
+  module Contrib
+    module Sequel
+      # Adds instrumentation to Sequel::Database
+      module Database
+        def self.included(base)
+          base.send(:prepend, InstanceMethods)
+        end
+
+        # Instance methods for instrumenting Sequel::Database
+        module InstanceMethods
+          def run(sql, options = ::Sequel::OPTS)
+            opts = parse_opts(sql, options)
+
+            response = nil
+
+            datadog_pin.tracer.trace(Ext::SPAN_QUERY) do |span|
+              span.service = datadog_pin.service
+              span.resource = opts[:query]
+              span.span_type = Datadog::Ext::SQL::TYPE
+              Utils.set_analytics_sample_rate(span)
+              span.set_tag(Ext::TAG_DB_VENDOR, adapter_name)
+              response = super(sql, options)
+            end
+            response
+          end
+
+          def datadog_pin
+            @pin ||= Datadog::Pin.new(
+              Datadog.configuration[:sequel][:service_name] || adapter_name,
+              app: Ext::APP,
+              app_type: Datadog::Ext::AppTypes::DB,
+              tracer: -> { Datadog.configuration[:sequel][:tracer] }
+            )
+          end
+
+          private
+
+          def adapter_name
+            Utils.adapter_name(self)
+          end
+
+          def parse_opts(sql, opts)
+            db_opts = if ::Sequel::VERSION < '3.41.0' && self.class.to_s !~ /Dataset$/
+                        @opts
+                      elsif instance_variable_defined?(:@pool) && @pool
+                        @pool.db.opts
+                      end
+            Utils.parse_opts(sql, opts, db_opts)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/sequel/dataset.rb

@@ -0,0 +1,62 @@
+require 'ddtrace/ext/sql'
+require 'ddtrace/ext/app_types'
+require 'ddtrace/contrib/analytics'
+require 'ddtrace/contrib/sequel/ext'
+require 'ddtrace/contrib/sequel/utils'
+
+module Datadog
+  module Contrib
+    module Sequel
+      # Adds instrumentation to Sequel::Dataset
+      module Dataset
+        def self.included(base)
+          base.send(:prepend, InstanceMethods)
+        end
+
+        # Instance methods for instrumenting Sequel::Dataset
+        module InstanceMethods
+          def execute(sql, options = ::Sequel::OPTS, &block)
+            trace_execute(proc { super(sql, options, &block) }, sql, options, &block)
+          end
+
+          def execute_ddl(sql, options = ::Sequel::OPTS, &block)
+            trace_execute(proc { super(sql, options, &block) }, sql, options, &block)
+          end
+
+          def execute_dui(sql, options = ::Sequel::OPTS, &block)
+            trace_execute(proc { super(sql, options, &block) }, sql, options, &block)
+          end
+
+          def execute_insert(sql, options = ::Sequel::OPTS, &block)
+            trace_execute(proc { super(sql, options, &block) }, sql, options, &block)
+          end
+
+          def datadog_pin
+            Datadog::Pin.get_from(db)
+          end
+
+          private
+
+          def trace_execute(super_method, sql, options, &block)
+            opts = Utils.parse_opts(sql, options, db.opts)
+            response = nil
+
+            datadog_pin.tracer.trace(Ext::SPAN_QUERY) do |span|
+              span.service = datadog_pin.service
+              span.resource = opts[:query]
+              span.span_type = Datadog::Ext::SQL::TYPE
+              Utils.set_analytics_sample_rate(span)
+              span.set_tag(Ext::TAG_DB_VENDOR, adapter_name)
+              response = super_method.call(sql, options, &block)
+            end
+            response
+          end
+
+          def adapter_name
+            Utils.adapter_name(db)
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,15 @@
+module Datadog
+  module Contrib
+    module Sequel
+      # Sequel integration constants
+      module Ext
+        APP = 'sequel'.freeze
+        ENV_ANALYTICS_ENABLED = 'DD_SEQUEL_ANALYTICS_ENABLED'.freeze
+        ENV_ANALYTICS_SAMPLE_RATE = 'DD_SEQUEL_ANALYTICS_SAMPLE_RATE'.freeze
+        SERVICE_NAME = 'sequel'.freeze
+        SPAN_QUERY = 'sequel.query'.freeze
+        TAG_DB_VENDOR = 'sequel.db.vendor'.freeze
+      end
+    end
+  end
+end

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

@@ -0,0 +1,38 @@
+require 'ddtrace/contrib/integration'
+require 'ddtrace/contrib/sequel/configuration/settings'
+require 'ddtrace/contrib/sequel/patcher'
+
+module Datadog
+  module Contrib
+    module Sequel
+      # Description of Sequel integration
+      class Integration
+        include Contrib::Integration
+
+        MINIMUM_VERSION = Gem::Version.new('3.41')
+
+        register_as :sequel, auto_patch: false
+
+        def self.version
+          Gem.loaded_specs['sequel'] && Gem.loaded_specs['sequel'].version
+        end
+
+        def self.loaded?
+          !defined?(::Sequel).nil?
+        end
+
+        def self.compatible?
+          super && version >= MINIMUM_VERSION
+        end
+
+        def default_configuration
+          Configuration::Settings.new
+        end
+
+        def patcher
+          Patcher
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,33 @@
+require 'ddtrace/contrib/patcher'
+require 'ddtrace/contrib/sequel/database'
+require 'ddtrace/contrib/sequel/dataset'
+
+module Datadog
+  module Contrib
+    module Sequel
+      # Patcher enables patching of 'sequel' module.
+      module Patcher
+        include Contrib::Patcher
+
+        module_function
+
+        def target_version
+          Integration.version
+        end
+
+        def patch
+          patch_sequel_database
+          patch_sequel_dataset
+        end
+
+        def patch_sequel_database
+          ::Sequel::Database.send(:include, Database)
+        end
+
+        def patch_sequel_dataset
+          ::Sequel::Dataset.send(:include, Dataset)
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/sequel/utils.rb

@@ -0,0 +1,46 @@
+module Datadog
+  module Contrib
+    module Sequel
+      # General purpose functions for Sequel
+      module Utils
+        class << self
+          def adapter_name(database)
+            Datadog::Utils::Database.normalize_vendor(database.adapter_scheme.to_s)
+          end
+
+          def parse_opts(sql, opts, db_opts)
+            if ::Sequel::VERSION >= '4.37.0' && !sql.is_a?(String)
+              # In 4.37.0, sql was converted to a prepared statement object
+              sql = sql.prepared_sql unless sql.is_a?(Symbol)
+            end
+
+            {
+              name: opts[:type],
+              query: sql,
+              database: db_opts[:database],
+              host: db_opts[:host]
+            }
+          end
+
+          def set_analytics_sample_rate(span)
+            Contrib::Analytics.set_sample_rate(span, analytics_sample_rate) if analytics_enabled?
+          end
+
+          private
+
+          def datadog_configuration
+            Datadog.configuration[:sequel]
+          end
+
+          def analytics_enabled?
+            Contrib::Analytics.enabled?(datadog_configuration[:analytics_enabled])
+          end
+
+          def analytics_sample_rate
+            datadog_configuration[:analytics_sample_rate]
+          end
+        end
+      end
+    end
+  end
+end