ddtrace 0.49.0 → 1.12.1

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

@@ -1,33 +0,0 @@
-require 'ddtrace/contrib/patcher'
-require 'ddtrace/contrib/presto/ext'
-require 'ddtrace/contrib/presto/instrumentation'
-require 'ddtrace/utils/only_once'
-
-module Datadog
-  module Contrib
-    module Presto
-      # Patcher enables patching of 'presto-client' module.
-      module Patcher
-        include Contrib::Patcher
-
-        PATCH_ONLY_ONCE = Datadog::Utils::OnlyOnce.new
-
-        module_function
-
-        def patched?
-          PATCH_ONLY_ONCE.ran?
-        end
-
-        def patch
-          PATCH_ONLY_ONCE.run do
-            begin
-              ::Presto::Client::Client.send(:include, Instrumentation::Client)
-            rescue StandardError => e
-              Datadog.logger.error("Unable to apply Presto integration: #{e}")
-            end
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,35 +0,0 @@
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/qless/ext'
-
-module Datadog
-  module Contrib
-    module Qless
-      module Configuration
-        # Custom settings for the Qless 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 :tag_job_data do |o|
-            o.default { env_to_bool(Ext::ENV_TAG_JOB_DATA, false) }
-            o.lazy
-          end
-
-          option :tag_job_tags do |o|
-            o.default { env_to_bool(Ext::ENV_TAG_JOB_TAGS, false) }
-            o.lazy
-          end
-
-          option :service_name, default: Ext::SERVICE_NAME
-        end
-      end
-    end
-  end
-end

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

@@ -1,20 +0,0 @@
-module Datadog
-  module Contrib
-    module Qless
-      # Qless integration constants
-      module Ext
-        APP = 'qless'.freeze
-        ENV_ANALYTICS_ENABLED = 'DD_QLESS_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE = 'DD_QLESS_ANALYTICS_SAMPLE_RATE'.freeze
-        ENV_TAG_JOB_DATA = 'DD_QLESS_TAG_JOB_DATA'.freeze
-        ENV_TAG_JOB_TAGS = 'DD_QLESS_TAG_JOB_TAGS'.freeze
-        SERVICE_NAME = 'qless'.freeze
-        SPAN_JOB = 'qless.job'.freeze
-        TAG_JOB_ID = 'qless.job.id'.freeze
-        TAG_JOB_DATA = 'qless.job.data'.freeze
-        TAG_JOB_QUEUE = 'qless.job.queue'.freeze
-        TAG_JOB_TAGS = 'qless.job.tags'.freeze
-      end
-    end
-  end
-end

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

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

@@ -1,35 +0,0 @@
-require 'ddtrace/contrib/patcher'
-require 'ddtrace/ext/app_types'
-
-module Datadog
-  module Contrib
-    module Qless
-      # Patcher enables patching of 'qless' module.
-      module Patcher
-        include Contrib::Patcher
-
-        module_function
-
-        def target_version
-          Integration.version
-        end
-
-        def patch
-          require_relative 'qless_job'
-          require_relative 'tracer_cleaner'
-
-          # Instrument all Qless Workers
-          ::Qless::Workers::BaseWorker.class_eval do
-            # These are executed in inverse order of listing here
-            include QlessJob
-            include TracerCleaner
-          end
-        end
-
-        def get_option(option)
-          Datadog.configuration[:qless].get_option(option)
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/qless/qless_job.rb

@@ -1,73 +0,0 @@
-require 'ddtrace/ext/app_types'
-require 'ddtrace/contrib/analytics'
-require 'qless'
-
-module Datadog
-  module Contrib
-    module Qless
-      # Uses Qless job hooks to create traces
-      module QlessJob
-        def around_perform(job)
-          return super unless datadog_configuration && tracer
-
-          tracer.trace(Ext::SPAN_JOB, span_options) do |span|
-            span.resource = job.klass_name
-            span.span_type = Datadog::Ext::AppTypes::WORKER
-            span.set_tag(Ext::TAG_JOB_ID, job.jid)
-            span.set_tag(Ext::TAG_JOB_QUEUE, job.queue_name)
-
-            tag_job_tags = datadog_configuration[:tag_job_tags]
-            span.set_tag(Ext::TAG_JOB_TAGS, job.tags) if tag_job_tags
-
-            tag_job_data = datadog_configuration[:tag_job_data]
-            if tag_job_data && !job.data.empty?
-              job_data = job.data.with_indifferent_access
-              formatted_data = job_data.except(:tags).map do |key, value|
-                "#{key}:#{value}".underscore
-              end
-
-              span.set_tag(Ext::TAG_JOB_DATA, formatted_data)
-            end
-
-            # Set analytics sample rate
-            if Contrib::Analytics.enabled?(datadog_configuration[:analytics_enabled])
-              Contrib::Analytics.set_sample_rate(span, datadog_configuration[:analytics_sample_rate])
-            end
-
-            # Measure service stats
-            Contrib::Analytics.set_measured(span)
-
-            super
-          end
-        end
-
-        def after_fork
-          configuration = Datadog.configuration[:qless]
-          return if configuration.nil?
-
-          # Add a pin, marking the job as forked.
-          # Used to trigger shutdown in forks for performance reasons.
-          # Cleanup happens in the TracerCleaner class
-          Datadog::Pin.new(
-            configuration[:service_name],
-            config: { forked: true }
-          ).onto(::Qless)
-        end
-
-        private
-
-        def span_options
-          { service: datadog_configuration[:service_name] }
-        end
-
-        def tracer
-          datadog_configuration.tracer
-        end
-
-        def datadog_configuration
-          Datadog.configuration[:qless]
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/qless/tracer_cleaner.rb

@@ -1,33 +0,0 @@
-module Datadog
-  module Contrib
-    module Qless
-      # Shutdown Tracer in forks for performance reasons
-      module TracerCleaner
-        def around_perform(job)
-          return super unless datadog_configuration && tracer
-
-          super.tap do
-            tracer.shutdown! if forked?
-          end
-        end
-
-        private
-
-        def forked?
-          pin = Datadog::Pin.get_from(::Qless)
-          return false unless pin
-
-          pin.config[:forked] == true
-        end
-
-        def tracer
-          datadog_configuration.tracer
-        end
-
-        def datadog_configuration
-          Datadog.configuration[:qless]
-        end
-      end
-    end
-  end
-end

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

@@ -1,43 +0,0 @@
-# frozen_string_literal: true
-
-require 'ddtrace/contrib/configuration/settings'
-
-module Datadog
-  module Contrib
-    module Que
-      module Configuration
-        # Default settings for the Que integration
-        class Settings < Datadog::Contrib::Configuration::Settings
-          option :service_name, default: Ext::SERVICE_NAME
-          option :distributed_tracing, default: true
-
-          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 :tag_args do |o|
-            o.default { env_to_bool(Ext::ENV_TAG_ARGS_ENABLED, false) }
-            o.lazy
-          end
-
-          option :tag_data do |o|
-            o.default { env_to_bool(Ext::ENV_TAG_DATA_ENABLED, false) }
-            o.lazy
-          end
-          option :error_handler, default: Datadog::Tracer::DEFAULT_ON_ERROR
-        end
-      end
-    end
-  end
-end

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

@@ -1,30 +0,0 @@
-# frozen_string_literal: true
-
-module Datadog
-  module Contrib
-    module Que
-      # Que integration constants
-      module Ext
-        APP = 'que'
-        ENV_ANALYTICS_ENABLED = 'DD_TRACE_QUE_ANALYTICS_ENABLED'
-        ENV_ANALYTICS_ENABLED_OLD = 'DD_QUE_ANALYTICS_ENABLED'
-        ENV_ANALYTICS_SAMPLE_RATE = 'DD_TRACE_QUE_ANALYTICS_SAMPLE_RATE'
-        ENV_ANALYTICS_SAMPLE_RATE_OLD = 'DD_QUE_ANALYTICS_SAMPLE_RATE'
-        ENV_ENABLED = 'DD_TRACE_QUE_ENABLED'
-        ENV_TAG_ARGS_ENABLED = 'DD_TRACE_QUE_TAG_ARGS_ENABLED'
-        ENV_TAG_DATA_ENABLED = 'DD_TRACE_QUE_TAG_DATA_ENABLED'
-        SERVICE_NAME = 'que'
-        SPAN_JOB = 'que.job'
-        TAG_JOB_ARGS = 'que.job.args'
-        TAG_JOB_DATA = 'que.job.data'
-        TAG_JOB_ERROR_COUNT = 'que.job.error_count'
-        TAG_JOB_EXPIRED_AT = 'que.job.expired_at'
-        TAG_JOB_FINISHED_AT = 'que.job.finished_at'
-        TAG_JOB_ID = 'que.job.id'
-        TAG_JOB_PRIORITY = 'que.job.priority'
-        TAG_JOB_QUEUE = 'que.job.queue'
-        TAG_JOB_RUN_AT = 'que.job.run_at'
-      end
-    end
-  end
-end

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

@@ -1,42 +0,0 @@
-# frozen_string_literal: true
-
-require 'ddtrace/contrib/integration'
-require 'ddtrace/contrib/que/ext'
-require 'ddtrace/contrib/que/configuration/settings'
-require 'ddtrace/contrib/que/patcher'
-
-module Datadog
-  module Contrib
-    module Que
-      # Description of Que integration
-      class Integration
-        include Datadog::Contrib::Integration
-
-        MINIMUM_VERSION = Gem::Version.new('1.0.0.beta2')
-
-        register_as :que, auto_patch: true
-
-        def self.version
-          Gem.loaded_specs[Datadog::Contrib::Que::Ext::APP] &&
-            Gem.loaded_specs[Datadog::Contrib::Que::Ext::APP].version
-        end
-
-        def self.loaded?
-          !defined?(::Que).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/que/patcher.rb

@@ -1,24 +0,0 @@
-# frozen_string_literal: true
-
-require 'ddtrace/contrib/que/tracer'
-
-module Datadog
-  module Contrib
-    module Que
-      # Patcher enables patching of 'que' module.
-      module Patcher
-        include Datadog::Contrib::Patcher
-
-        module_function
-
-        def target_version
-          Integration.version
-        end
-
-        def patch
-          ::Que.job_middleware.push(Que::Tracer.new)
-        end
-      end
-    end
-  end
-end

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

@@ -1,57 +0,0 @@
-# frozen_string_literal: true
-
-require 'ddtrace/contrib/analytics'
-
-module Datadog
-  module Contrib
-    module Que
-      # Tracer is a Que's server-side middleware which traces executed jobs
-      class Tracer
-        def call(job)
-          trace_options = {
-            service: configuration[:service_name],
-            span_type: Datadog::Ext::AppTypes::WORKER,
-            on_error: configuration[:error_handler]
-          }
-
-          tracer.trace(Ext::SPAN_JOB, trace_options) do |request_span|
-            request_span.resource = job.class.name.to_s
-            request_span.set_tag(Ext::TAG_JOB_QUEUE, job.que_attrs[:queue])
-            request_span.set_tag(Ext::TAG_JOB_ID, job.que_attrs[:id])
-            request_span.set_tag(Ext::TAG_JOB_PRIORITY, job.que_attrs[:priority])
-            request_span.set_tag(Ext::TAG_JOB_ERROR_COUNT, job.que_attrs[:error_count])
-            request_span.set_tag(Ext::TAG_JOB_RUN_AT, job.que_attrs[:run_at])
-            request_span.set_tag(Ext::TAG_JOB_EXPIRED_AT, job.que_attrs[:expired_at])
-            request_span.set_tag(Ext::TAG_JOB_FINISHED_AT, job.que_attrs[:finished_at])
-            request_span.set_tag(Ext::TAG_JOB_ARGS, job.que_attrs[:args]) if configuration[:tag_args]
-            request_span.set_tag(Ext::TAG_JOB_DATA, job.que_attrs[:data]) if configuration[:tag_data]
-
-            set_sample_rate(request_span)
-            Contrib::Analytics.set_measured(request_span)
-
-            yield
-          end
-        end
-
-        private
-
-        def set_sample_rate(request_span)
-          if Contrib::Analytics.enabled?(configuration[:analytics_enabled])
-            Contrib::Analytics.set_sample_rate(
-              request_span,
-              configuration[:analytics_sample_rate]
-            )
-          end
-        end
-
-        def tracer
-          configuration[:tracer]
-        end
-
-        def configuration
-          Datadog.configuration[Datadog::Contrib::Que::Ext::APP.to_sym]
-        end
-      end
-    end
-  end
-end

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

@@ -1,30 +0,0 @@
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/racecar/ext'
-
-module Datadog
-  module Contrib
-    module Racecar
-      module Configuration
-        # Custom settings for the Racecar 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/racecar/event.rb

@@ -1,76 +0,0 @@
-require 'ddtrace/contrib/analytics'
-require 'ddtrace/contrib/active_support/notifications/event'
-require 'ddtrace/contrib/racecar/ext'
-require 'ddtrace/ext/integration'
-
-module Datadog
-  module Contrib
-    module Racecar
-      # Defines basic behaviors for an ActiveRecord event.
-      module Event
-        def self.included(base)
-          base.send(:include, ActiveSupport::Notifications::Event)
-          base.send(:extend, ClassMethods)
-        end
-
-        # Class methods for Racecar events.
-        # Note, they share the same process method and before_trace method.
-        module ClassMethods
-          def subscription(*args)
-            super.tap do |subscription|
-              subscription.before_trace { ensure_clean_context! }
-            end
-          end
-
-          def span_options
-            { service: configuration[:service_name] }
-          end
-
-          def tracer
-            -> { configuration[:tracer] }
-          end
-
-          def configuration
-            Datadog.configuration[:racecar]
-          end
-
-          def process(span, event, _id, payload)
-            span.service = configuration[:service_name]
-            span.resource = payload[:consumer_class]
-
-            # Tag as an external peer service
-            span.set_tag(Datadog::Ext::Integration::TAG_PEER_SERVICE, span.service)
-
-            # 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.set_tag(Ext::TAG_TOPIC, payload[:topic])
-            span.set_tag(Ext::TAG_CONSUMER, payload[:consumer_class])
-            span.set_tag(Ext::TAG_PARTITION, payload[:partition])
-            span.set_tag(Ext::TAG_OFFSET, payload[:offset]) if payload.key?(:offset)
-            span.set_tag(Ext::TAG_FIRST_OFFSET, payload[:first_offset]) if payload.key?(:first_offset)
-            span.set_tag(Ext::TAG_MESSAGE_COUNT, payload[:message_count]) if payload.key?(:message_count)
-            span.set_error(payload[:exception_object]) if payload[:exception_object]
-          end
-
-          private
-
-          # Context objects are thread-bound.
-          # If Racecar re-uses threads, context from a previous trace
-          # could leak into the new trace. This "cleans" current context,
-          # preventing such a leak.
-          def ensure_clean_context!
-            return unless configuration[:tracer].call_context.current_span
-
-            configuration[:tracer].provider.context = Context.new
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/racecar/events/batch.rb

@@ -1,27 +0,0 @@
-require 'ddtrace/contrib/racecar/ext'
-require 'ddtrace/contrib/racecar/event'
-
-module Datadog
-  module Contrib
-    module Racecar
-      module Events
-        # Defines instrumentation for process_batch.racecar event
-        module Batch
-          include Racecar::Event
-
-          EVENT_NAME = 'process_batch.racecar'.freeze
-
-          module_function
-
-          def event_name
-            self::EVENT_NAME
-          end
-
-          def span_name
-            Ext::SPAN_BATCH
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/racecar/events/consume.rb

@@ -1,27 +0,0 @@
-require 'ddtrace/contrib/racecar/ext'
-require 'ddtrace/contrib/racecar/event'
-
-module Datadog
-  module Contrib
-    module Racecar
-      module Events
-        # Defines instrumentation for main_loop.racecar event
-        module Consume
-          include Racecar::Event
-
-          EVENT_NAME = 'main_loop.racecar'.freeze
-
-          module_function
-
-          def event_name
-            self::EVENT_NAME
-          end
-
-          def span_name
-            Ext::SPAN_CONSUME
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/racecar/events/message.rb

@@ -1,27 +0,0 @@
-require 'ddtrace/contrib/racecar/ext'
-require 'ddtrace/contrib/racecar/event'
-
-module Datadog
-  module Contrib
-    module Racecar
-      module Events
-        # Defines instrumentation for process_message.racecar event
-        module Message
-          include Racecar::Event
-
-          EVENT_NAME = 'process_message.racecar'.freeze
-
-          module_function
-
-          def event_name
-            self::EVENT_NAME
-          end
-
-          def span_name
-            Ext::SPAN_MESSAGE
-          end
-        end
-      end
-    end
-  end
-end