ddtrace 0.54.2 → 1.2.0

data/lib/ddtrace/contrib/patcher.rb

@@ -1,66 +0,0 @@
-# typed: false
-require 'ddtrace/utils/only_once'
-
-module Datadog
-  module Contrib
-    # Common behavior for patcher modules
-    module Patcher
-      def self.included(base)
-        base.singleton_class.prepend(CommonMethods)
-        base.prepend(CommonMethods) if base.instance_of?(Class)
-      end
-
-      # Prepended instance methods for all patchers
-      module CommonMethods
-        def patch_name
-          self.class != Class && self.class != Module ? self.class.name : name
-        end
-
-        def patched?
-          patch_only_once.ran?
-        end
-
-        def patch
-          return unless defined?(super)
-
-          patch_only_once.run do
-            begin
-              super.tap do
-                # Emit a metric
-                Datadog.health_metrics.instrumentation_patched(1, tags: default_tags)
-              end
-            rescue StandardError => e
-              on_patch_error(e)
-            end
-          end
-        end
-
-        # Processes patching errors. This default implementation logs the error and reports relevant metrics.
-        # @param e [Exception]
-        def on_patch_error(e)
-          # Log the error
-          Datadog.logger.error("Failed to apply #{patch_name} patch. Cause: #{e} Location: #{Array(e.backtrace).first}")
-
-          # Emit a metric
-          tags = default_tags
-          tags << "error:#{e.class.name}"
-
-          Datadog.health_metrics.error_instrumentation_patch(1, tags: tags)
-        end
-
-        private
-
-        def default_tags
-          ["patcher:#{patch_name}"].tap do |tags|
-            tags << "target_version:#{target_version}" if respond_to?(:target_version) && !target_version.nil?
-          end
-        end
-
-        def patch_only_once
-          # NOTE: This is not thread-safe
-          @patch_only_once ||= Datadog::Utils::OnlyOnce.new
-        end
-      end
-    end
-  end
-end

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

@@ -1,31 +0,0 @@
-# typed: false
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/presto/ext'
-
-module Datadog
-  module Contrib
-    module Presto
-      module Configuration
-        # Custom settings for the Presto 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/presto/ext.rb

@@ -1,29 +0,0 @@
-# typed: true
-module Datadog
-  module Contrib
-    module Presto
-      # Presto integration constants
-      module Ext
-        APP = 'presto'.freeze
-        ENV_ENABLED = 'DD_TRACE_PRESTO_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED = 'DD_TRACE_PRESTO_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_ENABLED_OLD = 'DD_PRESTO_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE = 'DD_TRACE_PRESTO_ANALYTICS_SAMPLE_RATE'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE_OLD = 'DD_PRESTO_ANALYTICS_SAMPLE_RATE'.freeze
-        SERVICE_NAME = 'presto'.freeze
-        SPAN_QUERY = 'presto.query'.freeze
-        SPAN_RESUME = 'presto.resume_query'.freeze
-        SPAN_KILL = 'presto.kill_query'.freeze
-        TAG_SCHEMA_NAME = 'presto.schema'.freeze
-        TAG_CATALOG_NAME = 'presto.catalog'.freeze
-        TAG_USER_NAME = 'presto.user'.freeze
-        TAG_TIME_ZONE = 'presto.time_zone'.freeze
-        TAG_LANGUAGE = 'presto.language'.freeze
-        TAG_PROXY = 'presto.http_proxy'.freeze
-        TAG_MODEL_VERSION = 'presto.model_version'.freeze
-        TAG_QUERY_ID = 'presto.query.id'.freeze
-        TAG_QUERY_ASYNC = 'presto.query.async'.freeze
-      end
-    end
-  end
-end

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

@@ -1,111 +0,0 @@
-# typed: true
-require 'ddtrace/ext/net'
-require 'ddtrace/ext/sql'
-require 'ddtrace/ext/app_types'
-require 'ddtrace/contrib/presto/ext'
-
-module Datadog
-  module Contrib
-    module Presto
-      # Instrumentation for Presto integration
-      module Instrumentation
-        # Instrumentation for Presto::Client::Client
-        module Client
-          def self.included(base)
-            base.prepend(InstanceMethods)
-          end
-
-          # Instance methods for Presto::Client
-          module InstanceMethods
-            def run(query)
-              tracer.trace(Ext::SPAN_QUERY, span_options) do |span|
-                begin
-                  decorate!(span)
-                  span.resource = query
-                  span.span_type = Datadog::Ext::SQL::TYPE
-                  span.set_tag(Ext::TAG_QUERY_ASYNC, false)
-                rescue StandardError => e
-                  Datadog.logger.debug("error preparing span for presto: #{e}")
-                end
-
-                super(query)
-              end
-            end
-
-            def query(query, &blk)
-              tracer.trace(Ext::SPAN_QUERY, span_options) do |span|
-                begin
-                  decorate!(span)
-                  span.resource = query
-                  span.span_type = Datadog::Ext::SQL::TYPE
-                  span.set_tag(Ext::TAG_QUERY_ASYNC, !blk.nil?)
-                rescue StandardError => e
-                  Datadog.logger.debug("error preparing span for presto: #{e}")
-                end
-
-                super(query, &blk)
-              end
-            end
-
-            def kill(query_id)
-              tracer.trace(Ext::SPAN_KILL, span_options) do |span|
-                begin
-                  decorate!(span)
-                  span.resource = Ext::SPAN_KILL
-                  span.span_type = Datadog::Ext::AppTypes::DB
-                  # ^ not an SQL type span, since there's no SQL query
-                  span.set_tag(Ext::TAG_QUERY_ID, query_id)
-                rescue StandardError => e
-                  Datadog.logger.debug("error preparing span for presto: #{e}")
-                end
-
-                super(query_id)
-              end
-            end
-
-            private
-
-            def datadog_configuration
-              Datadog.configuration[:presto]
-            end
-
-            def span_options
-              {
-                service: datadog_configuration[:service_name],
-                app: Ext::APP,
-                app_type: Datadog::Ext::AppTypes::DB
-              }
-            end
-
-            def tracer
-              datadog_configuration.tracer
-            end
-
-            def decorate!(span)
-              set_nilable_tag!(span, :server, Datadog::Ext::NET::TARGET_HOST)
-              set_nilable_tag!(span, :user, Ext::TAG_USER_NAME)
-              set_nilable_tag!(span, :schema, Ext::TAG_SCHEMA_NAME)
-              set_nilable_tag!(span, :catalog, Ext::TAG_CATALOG_NAME)
-              set_nilable_tag!(span, :time_zone, Ext::TAG_TIME_ZONE)
-              set_nilable_tag!(span, :language, Ext::TAG_LANGUAGE)
-              set_nilable_tag!(span, :http_proxy, Ext::TAG_PROXY)
-              set_nilable_tag!(span, :model_version, Ext::TAG_MODEL_VERSION)
-
-              # 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?(datadog_configuration[:analytics_enabled])
-                Contrib::Analytics.set_sample_rate(span, datadog_configuration[:analytics_sample_rate])
-              end
-            end
-
-            def set_nilable_tag!(span, key, tag_name)
-              @options[key].tap { |val| span.set_tag(tag_name, val) if val }
-            end
-          end
-        end
-      end
-    end
-  end
-end

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

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

@@ -1,34 +0,0 @@
-# typed: true
-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.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,36 +0,0 @@
-# typed: false
-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,21 +0,0 @@
-# typed: true
-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,39 +0,0 @@
-# typed: false
-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 @@
-# typed: true
-require 'ddtrace/contrib/patcher'
-require 'ddtrace/ext/app_types'
-
-module Datadog
-  module Contrib
-    module Qless
-      # Patcher enables patching of 'qless' module.
-      module Patcher
-        include Kernel # Ensure that kernel methods are always available (https://sorbet.org/docs/error-reference#7003)
-        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
-          # These are executed in inverse order of listing here
-          ::Qless::Workers::BaseWorker.include(QlessJob)
-          ::Qless::Workers::BaseWorker.include(TracerCleaner)
-        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,74 +0,0 @@
-# typed: false
-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,34 +0,0 @@
-# typed: true
-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,44 +0,0 @@
-# typed: false
-# 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,31 +0,0 @@
-# typed: strict
-# 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,43 +0,0 @@
-# typed: false
-# 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