ddtrace 0.49.0 → 1.12.1

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

@@ -1,28 +0,0 @@
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/rspec/ext'
-
-module Datadog
-  module Contrib
-    module RSpec
-      module Configuration
-        # Custom settings for the RSpec integration
-        class Settings < Contrib::Configuration::Settings
-          option :enabled do |o|
-            o.default { env_to_bool(Ext::ENV_ENABLED, true) }
-            o.lazy
-          end
-
-          option :service_name do |o|
-            o.default { Datadog.configuration.service || Ext::SERVICE_NAME }
-            o.lazy
-          end
-
-          option :operation_name do |o|
-            o.default { ENV.key?(Ext::ENV_OPERATION_NAME) ? ENV[Ext::ENV_OPERATION_NAME] : Ext::OPERATION_NAME }
-            o.lazy
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/rspec/example.rb

@@ -1,75 +0,0 @@
-module Datadog
-  module Contrib
-    module RSpec
-      # Instrument RSpec::Core::Example
-      module Example
-        def self.included(base)
-          base.send(:prepend, InstanceMethods)
-        end
-
-        # Instance methods for configuration
-        module InstanceMethods
-          def run(example_group_instance, reporter)
-            return super unless configuration[:enabled]
-
-            test_name = full_description.strip
-            if metadata[:description].empty?
-              # for unnamed it blocks this appends something like "example at ./spec/some_spec.rb:10"
-              test_name += " #{description}"
-            end
-
-            trace_options = {
-              app: Ext::APP,
-              resource: test_name,
-              service: configuration[:service_name],
-              span_type: Datadog::Ext::AppTypes::TEST,
-              tags: tags.merge(Datadog.configuration.tags)
-            }
-
-            tracer.trace(configuration[:operation_name], trace_options) do |span|
-              span.set_tag(Datadog::Ext::Test::TAG_FRAMEWORK, Ext::FRAMEWORK)
-              span.set_tag(Datadog::Ext::Test::TAG_NAME, test_name)
-              span.set_tag(Datadog::Ext::Test::TAG_SUITE, file_path)
-              span.set_tag(Datadog::Ext::Test::TAG_TYPE, Ext::TEST_TYPE)
-              span.set_tag(Datadog::Ext::Test::TAG_SPAN_KIND, Datadog::Ext::AppTypes::TEST)
-
-              # Measure service stats
-              Contrib::Analytics.set_measured(span)
-
-              result = super
-
-              case execution_result.status
-              when :passed
-                span.set_tag(Datadog::Ext::Test::TAG_STATUS, Datadog::Ext::Test::Status::PASS)
-              when :failed
-                span.status = 1
-                span.set_tag(Datadog::Ext::Test::TAG_STATUS, Datadog::Ext::Test::Status::FAIL)
-                span.set_error(execution_result.exception)
-              else
-                if execution_result.example_skipped?
-                  span.set_tag(Datadog::Ext::Test::TAG_STATUS, Datadog::Ext::Test::Status::SKIP)
-                end
-              end
-
-              result
-            end
-          end
-
-          private
-
-          def configuration
-            Datadog.configuration[:rspec]
-          end
-
-          def tracer
-            configuration[:tracer]
-          end
-
-          def tags
-            @tags ||= Datadog::Ext::CI.tags(ENV)
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,16 +0,0 @@
-module Datadog
-  module Contrib
-    module RSpec
-      # RSpec integration constants
-      module Ext
-        APP = 'rspec'.freeze
-        ENV_ENABLED = 'DD_TRACE_RSPEC_ENABLED'.freeze
-        ENV_OPERATION_NAME = 'DD_TRACE_RSPEC_OPERATION_NAME'.freeze
-        FRAMEWORK = 'rspec'.freeze
-        OPERATION_NAME = 'rspec.example'.freeze
-        SERVICE_NAME = 'rspec'.freeze
-        TEST_TYPE = 'test'.freeze
-      end
-    end
-  end
-end

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

@@ -1,46 +0,0 @@
-require 'ddtrace/contrib/integration'
-require 'ddtrace/contrib/rspec/configuration/settings'
-require 'ddtrace/contrib/rspec/patcher'
-require 'ddtrace/contrib/integration'
-
-module Datadog
-  module Contrib
-    module RSpec
-      # Description of RSpec integration
-      class Integration
-        include Contrib::Integration
-
-        MINIMUM_VERSION = Gem::Version.new('3.0.0')
-
-        register_as :rspec, auto_patch: true
-
-        def self.version
-          Gem.loaded_specs['rspec'] \
-            && Gem.loaded_specs['rspec'].version
-        end
-
-        def self.loaded?
-          !defined?(::RSpec).nil? && !defined?(::RSpec::Core).nil? && \
-            !defined?(::RSpec::Core::Example).nil?
-        end
-
-        def self.compatible?
-          super && version >= MINIMUM_VERSION
-        end
-
-        # test environments should not auto instrument test libraries
-        def auto_instrument?
-          false
-        end
-
-        def default_configuration
-          Configuration::Settings.new
-        end
-
-        def patcher
-          Patcher
-        end
-      end
-    end
-  end
-end

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

@@ -1,23 +0,0 @@
-require 'ddtrace/contrib/patcher'
-require 'ddtrace/contrib/rspec/example'
-
-module Datadog
-  module Contrib
-    module RSpec
-      # Patcher enables patching of 'rspec' module.
-      module Patcher
-        include Contrib::Patcher
-
-        module_function
-
-        def target_version
-          Integration.version
-        end
-
-        def patch
-          ::RSpec::Core::Example.send(:include, Example)
-        end
-      end
-    end
-  end
-end

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

@@ -1,28 +0,0 @@
-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 :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
-        end
-      end
-    end
-  end
-end

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

@@ -1,63 +0,0 @@
-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_common_tags(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
-            sql = sql.is_a?(::Sequel::SQL::Expression) ? literal(sql) : sql.to_s
-
-            Utils.parse_opts(sql, opts, db_opts)
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,63 +0,0 @@
-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, self)
-            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_common_tags(span)
-              span.set_tag(Ext::TAG_DB_VENDOR, adapter_name)
-              span.set_tag(Ext::TAG_PREPARED_NAME, opts[:prepared_name]) if opts[:prepared_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

@@ -1,19 +0,0 @@
-module Datadog
-  module Contrib
-    module Sequel
-      # Sequel integration constants
-      module Ext
-        APP = 'sequel'.freeze
-        ENV_ENABLED = 'DD_TRACE_SEQUEL_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED = 'DD_TRACE_SEQUEL_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED_OLD = 'DD_SEQUEL_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE = 'DD_TRACE_SEQUEL_ANALYTICS_SAMPLE_RATE'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE_OLD = 'DD_SEQUEL_ANALYTICS_SAMPLE_RATE'.freeze
-        SERVICE_NAME = 'sequel'.freeze
-        SPAN_QUERY = 'sequel.query'.freeze
-        TAG_DB_VENDOR = 'sequel.db.vendor'.freeze
-        TAG_PREPARED_NAME = 'sequel.prepared.name'.freeze
-      end
-    end
-  end
-end

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

@@ -1,38 +0,0 @@
-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

@@ -1,33 +0,0 @@
-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

@@ -1,74 +0,0 @@
-require 'ddtrace/ext/integration'
-
-module Datadog
-  module Contrib
-    module Sequel
-      # General purpose functions for Sequel
-      module Utils
-        class << self
-          # Ruby database connector library
-          #
-          # e.g. adapter:mysql2 (database:mysql), adapter:jdbc (database:postgres)
-          def adapter_name(database)
-            scheme = database.adapter_scheme.to_s
-
-            if scheme == 'jdbc'.freeze
-              # The subtype is more important in this case,
-              # otherwise all database adapters will be 'jdbc'.
-              database_type(database)
-            else
-              Datadog::Utils::Database.normalize_vendor(scheme)
-            end
-          end
-
-          # Database engine
-          #
-          # e.g. database:mysql (adapter:mysql2), database:postgres (adapter:jdbc)
-          def database_type(database)
-            Datadog::Utils::Database.normalize_vendor(database.database_type.to_s)
-          end
-
-          def parse_opts(sql, opts, db_opts, dataset = nil)
-            # Prepared statements don't provide their sql query in the +sql+ parameter.
-            if !sql.is_a?(String) && (dataset && dataset.respond_to?(:prepared_sql) &&
-              (resolved_sql = dataset.prepared_sql))
-              # The dataset contains the resolved SQL query and prepared statement name.
-              prepared_name = dataset.prepared_statement_name
-              sql = resolved_sql
-            end
-
-            {
-              name: opts[:type],
-              query: sql,
-              prepared_name: prepared_name,
-              database: db_opts[:database],
-              host: db_opts[:host]
-            }
-          end
-
-          def set_common_tags(span)
-            # Tag as an external peer service
-            span.set_tag(Datadog::Ext::Integration::TAG_PEER_SERVICE, span.service)
-
-            # Set analytics sample rate
-            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

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

@@ -1,31 +0,0 @@
-require 'ddtrace/contrib/configuration/settings'
-
-module Datadog
-  module Contrib
-    module Shoryuken
-      module Configuration
-        # Default settings for the Shoryuken 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
-          option :error_handler, default: Datadog::Tracer::DEFAULT_ON_ERROR
-          option :tag_body, default: false
-        end
-      end
-    end
-  end
-end

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

@@ -1,21 +0,0 @@
-module Datadog
-  module Contrib
-    module Shoryuken
-      # Shoryuken integration constants
-      module Ext
-        APP = 'shoryuken'.freeze
-        ENV_ENABLED = 'DD_TRACE_SHORYUKEN_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED = 'DD_TRACE_SHORYUKEN_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED_OLD = 'DD_SHORYUKEN_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE = 'DD_TRACE_SHORYUKEN_ANALYTICS_SAMPLE_RATE'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE_OLD = 'DD_SHORYUKEN_ANALYTICS_SAMPLE_RATE'.freeze
-        SERVICE_NAME = 'shoryuken'.freeze
-        SPAN_JOB = 'shoryuken.job'.freeze
-        TAG_JOB_ID = 'shoryuken.id'.freeze
-        TAG_JOB_QUEUE = 'shoryuken.queue'.freeze
-        TAG_JOB_ATTRIBUTES = 'shoryuken.attributes'.freeze
-        TAG_JOB_BODY = 'shoryuken.body'.freeze
-      end
-    end
-  end
-end

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

@@ -1,39 +0,0 @@
-require 'ddtrace/contrib/integration'
-require 'ddtrace/contrib/shoryuken/ext'
-require 'ddtrace/contrib/shoryuken/configuration/settings'
-require 'ddtrace/contrib/shoryuken/patcher'
-
-module Datadog
-  module Contrib
-    module Shoryuken
-      # Description of Shoryuken integration
-      class Integration
-        include Contrib::Integration
-
-        MINIMUM_VERSION = Gem::Version.new('3.2')
-
-        register_as :shoryuken
-
-        def self.version
-          Gem.loaded_specs['shoryuken'] && Gem.loaded_specs['shoryuken'].version
-        end
-
-        def self.loaded?
-          !defined?(::Shoryuken).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/shoryuken/patcher.rb

@@ -1,24 +0,0 @@
-require 'ddtrace/contrib/shoryuken/tracer'
-
-module Datadog
-  module Contrib
-    module Shoryuken
-      # Patcher enables patching of 'shoryuken' module.
-      module Patcher
-        include Contrib::Patcher
-
-        module_function
-
-        def target_version
-          Integration.version
-        end
-
-        def patch
-          ::Shoryuken.server_middleware do |chain|
-            chain.add Shoryuken::Tracer
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/shoryuken/tracer.rb

@@ -1,55 +0,0 @@
-require 'ddtrace/contrib/analytics'
-
-module Datadog
-  module Contrib
-    module Shoryuken
-      # Tracer is a Shoryuken server-side middleware which traces executed jobs
-      class Tracer
-        def initialize(options = {})
-          @tracer = options[:tracer] || configuration[:tracer]
-          @shoryuken_service = options[:service_name] || configuration[:service_name]
-          @error_handler = options[:error_handler] || configuration[:error_handler]
-        end
-
-        def call(worker_instance, queue, sqs_msg, body)
-          @tracer.trace(
-            Ext::SPAN_JOB,
-            service: @shoryuken_service,
-            span_type: Datadog::Ext::AppTypes::WORKER,
-            on_error: @error_handler
-          ) do |span|
-            # Set analytics sample rate
-            if Contrib::Analytics.enabled?(configuration[:analytics_enabled])
-              Contrib::Analytics.set_sample_rate(span, configuration[:analytics_sample_rate])
-            end
-
-            # Measure service stats
-            Contrib::Analytics.set_measured(span)
-
-            span.resource = resource(worker_instance, body)
-            span.set_tag(Ext::TAG_JOB_ID, sqs_msg.message_id)
-            span.set_tag(Ext::TAG_JOB_QUEUE, queue)
-            span.set_tag(Ext::TAG_JOB_ATTRIBUTES, sqs_msg.attributes) if sqs_msg.respond_to?(:attributes)
-            span.set_tag(Ext::TAG_JOB_BODY, body) if configuration[:tag_body]
-
-            yield
-          end
-        end
-
-        private
-
-        def resource(worker_instance, body)
-          # If it's a Hash, try to get the job class from it.
-          # This is for ActiveJob compatibility.
-          job_class = body['job_class'] if body.is_a?(Hash)
-          # If nothing is available, use the worker class name.
-          job_class || worker_instance.class.name
-        end
-
-        def configuration
-          Datadog.configuration[:shoryuken]
-        end
-      end
-    end
-  end
-end