ddtrace 0.35.2 → 1.1.0

data/lib/ddtrace/contrib/aws/services.rb

@@ -1,115 +0,0 @@
-module Datadog
-  module Contrib
-    # rubocop:disable Metrics/ModuleLength:
-    module Aws
-      SERVICES = %w[
-        ACM
-        APIGateway
-        AppStream
-        ApplicationAutoScaling
-        ApplicationDiscoveryService
-        Athena
-        AutoScaling
-        Batch
-        Budgets
-        CloudDirectory
-        CloudFormation
-        CloudFront
-        CloudHSM
-        CloudHSMV2
-        CloudSearch
-        CloudSearchDomain
-        CloudTrail
-        CloudWatch
-        CloudWatchEvents
-        CloudWatchLogs
-        CodeBuild
-        CodeCommit
-        CodeDeploy
-        CodePipeline
-        CodeStar
-        CognitoIdentity
-        CognitoIdentityProvider
-        CognitoSync
-        ConfigService
-        CostandUsageReportService
-        DAX
-        DataPipeline
-        DatabaseMigrationService
-        DeviceFarm
-        DirectConnect
-        DirectoryService
-        DynamoDB
-        DynamoDBStreams
-        EC2
-        ECR
-        ECS
-        EFS
-        EMR
-        ElastiCache
-        ElasticBeanstalk
-        ElasticLoadBalancing
-        ElasticLoadBalancingV2
-        ElasticTranscoder
-        ElasticsearchService
-        Firehose
-        GameLift
-        Glacier
-        Glue
-        Greengrass
-        Health
-        IAM
-        ImportExport
-        Inspector
-        IoT
-        IoTDataPlane
-        KMS
-        Kinesis
-        KinesisAnalytics
-        Lambda
-        LambdaPreview
-        Lex
-        LexModelBuildingService
-        Lightsail
-        MTurk
-        MachineLearning
-        MarketplaceCommerceAnalytics
-        MarketplaceEntitlementService
-        MarketplaceMetering
-        MigrationHub
-        Mobile
-        OpsWorks
-        OpsWorksCM
-        Organizations
-        Pinpoint
-        Polly
-        RDS
-        Redshift
-        Rekognition
-        ResourceGroupsTaggingAPI
-        Route53
-        Route53Domains
-        S3
-        SES
-        SMS
-        SNS
-        SQS
-        SSM
-        STS
-        SWF
-        ServiceCatalog
-        Shield
-        SimpleDB
-        Snowball
-        States
-        StorageGateway
-        Support
-        WAF
-        WAFRegional
-        WorkDocs
-        WorkSpaces
-        XRay
-      ].freeze
-    end
-  end
-end

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

@@ -1,15 +0,0 @@
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/concurrent_ruby/ext'
-
-module Datadog
-  module Contrib
-    module ConcurrentRuby
-      module Configuration
-        # Custom settings for the ConcurrentRuby integration
-        class Settings < Contrib::Configuration::Settings
-          option :service_name, default: Ext::SERVICE_NAME
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/concurrent_ruby/context_composite_executor_service.rb

@@ -1,35 +0,0 @@
-require 'concurrent/executor/executor_service'
-
-module Datadog
-  module Contrib
-    module ConcurrentRuby
-      # wraps existing executor to carry over trace context
-      class ContextCompositeExecutorService
-        extend Forwardable
-        include Concurrent::ExecutorService
-
-        attr_accessor :composited_executor
-
-        def initialize(composited_executor)
-          @composited_executor = composited_executor
-        end
-
-        # post method runs the task within composited executor - in a different thread
-        def post(*args, &task)
-          context = datadog_configuration.tracer.provider.context
-
-          @composited_executor.post(*args) do
-            datadog_configuration.tracer.provider.context = context
-            yield
-          end
-        end
-
-        def datadog_configuration
-          Datadog.configuration[:concurrent_ruby]
-        end
-
-        delegate [:can_overflow?, :serialized?] => :composited_executor
-      end
-    end
-  end
-end

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

@@ -1,11 +0,0 @@
-module Datadog
-  module Contrib
-    module ConcurrentRuby
-      # ConcurrentRuby integration constants
-      module Ext
-        APP = 'concurrent-ruby'.freeze
-        SERVICE_NAME = 'concurrent-ruby'.freeze
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/concurrent_ruby/future_patch.rb

@@ -1,23 +0,0 @@
-require 'ddtrace/contrib/concurrent_ruby/context_composite_executor_service'
-
-module Datadog
-  module Contrib
-    module ConcurrentRuby
-      # This patches the Future - to wrap executor service using ContextCompositeExecutorService
-      module FuturePatch
-        def self.included(base)
-          base.class_eval do
-            alias_method :ns_initialize_without_datadog, :ns_initialize
-            remove_method(:ns_initialize)
-
-            def ns_initialize(value, opts)
-              ns_initialize_without_datadog(value, opts)
-
-              @executor = ContextCompositeExecutorService.new(@executor)
-            end
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,38 +0,0 @@
-require 'ddtrace/contrib/integration'
-require 'ddtrace/contrib/concurrent_ruby/patcher'
-require 'ddtrace/contrib/concurrent_ruby/configuration/settings'
-
-module Datadog
-  module Contrib
-    module ConcurrentRuby
-      # Describes the ConcurrentRuby integration
-      class Integration
-        include Contrib::Integration
-
-        MINIMUM_VERSION = Gem::Version.new('0.9')
-
-        register_as :concurrent_ruby
-
-        def self.version
-          Gem.loaded_specs['concurrent-ruby'] && Gem.loaded_specs['concurrent-ruby'].version
-        end
-
-        def self.loaded?
-          !defined?(::Concurrent::Future).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/concurrent_ruby/patcher.rb

@@ -1,28 +0,0 @@
-require 'ddtrace/contrib/patcher'
-
-module Datadog
-  module Contrib
-    module ConcurrentRuby
-      # Patcher enables patching of 'Future' class.
-      module Patcher
-        include Contrib::Patcher
-
-        module_function
-
-        def target_version
-          Integration.version
-        end
-
-        def patch
-          require 'ddtrace/contrib/concurrent_ruby/future_patch'
-          patch_future
-        end
-
-        # Propagate tracing context in Concurrent::Future
-        def patch_future
-          ::Concurrent::Future.send(:include, FuturePatch)
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/configurable.rb

@@ -1,76 +0,0 @@
-require 'ddtrace/contrib/configuration/resolver'
-require 'ddtrace/contrib/configuration/settings'
-
-module Datadog
-  module Contrib
-    # Defines configurable behavior for integrations
-    module Configurable
-      def self.included(base)
-        base.send(:include, InstanceMethods)
-      end
-
-      # Configurable instance behavior for integrations
-      module InstanceMethods
-        def default_configuration
-          Configuration::Settings.new
-        end
-
-        def reset_configuration!
-          @configurations = nil
-          @resolver = nil
-        end
-
-        # Get matching configuration for key.
-        # If no match, returns default configuration.
-        def configuration(key = :default)
-          configurations[configuration_key(key)]
-        end
-
-        # If the key has matching configuration explicitly defined for it,
-        # then return true. Otherwise return false.
-        def configuration_for?(key)
-          key = resolver.resolve(key) unless key == :default
-          configurations.key?(key)
-        end
-
-        def configurations
-          @configurations ||= {
-            default: default_configuration
-          }
-        end
-
-        # Create or update configuration with provided settings.
-        def configure(key, options = {}, &block)
-          key ||= :default
-
-          # Get or add the configuration
-          config = configuration_for?(key) ? configuration(key) : add_configuration(key)
-
-          # Apply the settings
-          config.configure(options, &block)
-          config
-        end
-
-        protected
-
-        def resolver
-          @resolver ||= Configuration::Resolver.new
-        end
-
-        def add_configuration(key)
-          resolver.add(key)
-          config_key = resolver.resolve(key)
-          configurations[config_key] = default_configuration
-        end
-
-        def configuration_key(key)
-          return :default if key.nil? || key == :default
-
-          key = resolver.resolve(key)
-          key = :default unless configurations.key?(key)
-          key
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/configuration/resolver.rb

@@ -1,16 +0,0 @@
-module Datadog
-  module Contrib
-    module Configuration
-      # Resolves a configuration key to a Datadog::Contrib::Configuration:Settings object
-      class Resolver
-        def resolve(key)
-          key
-        end
-
-        def add(key)
-          key
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/configuration/resolvers/pattern_resolver.rb

@@ -1,39 +0,0 @@
-require 'ddtrace/contrib/configuration/resolver'
-
-module Datadog
-  module Contrib
-    module Configuration
-      # Resolves a value to a configuration key
-      module Resolvers
-        # Matches strings against Regexps.
-        class PatternResolver < Datadog::Contrib::Configuration::Resolver
-          def resolve(name)
-            # Try to find a matching pattern
-            matching_pattern = patterns.find do |pattern|
-              # Rubocop incorrectly thinks assignment is done here...
-              # rubocop:disable Style/ConditionalAssignment
-              if pattern.is_a?(Proc)
-                pattern === name
-              else
-                pattern === name.to_s # Co-erce to string
-              end
-            end
-
-            # Return match or default
-            matching_pattern || :default
-          end
-
-          def add(pattern)
-            patterns << (pattern.is_a?(Regexp) || pattern.is_a?(Proc) ? pattern : pattern.to_s)
-          end
-
-          private
-
-          def patterns
-            @patterns ||= Set.new
-          end
-        end
-      end
-    end
-  end
-end

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

@@ -1,35 +0,0 @@
-require 'ddtrace/configuration/base'
-
-module Datadog
-  module Contrib
-    module Configuration
-      # Common settings for all integrations
-      class Settings
-        include Datadog::Configuration::Base
-
-        option :analytics_enabled, default: false
-        option :analytics_sample_rate, default: 1.0
-        option :service_name
-        option :tracer do |o|
-          o.delegate_to { Datadog.tracer }
-        end
-
-        def configure(options = {})
-          self.class.options.dependency_order.each do |name|
-            self[name] = options[name] if options.key?(name)
-          end
-
-          yield(self) if block_given?
-        end
-
-        def [](name)
-          respond_to?(name) ? send(name) : get_option(name)
-        end
-
-        def []=(name, value)
-          respond_to?("#{name}=") ? send("#{name}=", value) : set_option(name, value)
-        end
-      end
-    end
-  end
-end

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

@@ -1,25 +0,0 @@
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/dalli/ext'
-
-module Datadog
-  module Contrib
-    module Dalli
-      module Configuration
-        # Custom settings for the Dalli 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 :service_name, default: Ext::SERVICE_NAME
-        end
-      end
-    end
-  end
-end

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

@@ -1,17 +0,0 @@
-module Datadog
-  module Contrib
-    module Dalli
-      # Dalli integration constants
-      module Ext
-        APP = 'dalli'.freeze
-        ENV_ANALYTICS_ENABLED = 'DD_DALLI_ANALYTICS_ENABLED'.freeze
-        ENV_ANALYTICS_SAMPLE_RATE = 'DD_DALLI_ANALYTICS_SAMPLE_RATE'.freeze
-        QUANTIZE_MAX_CMD_LENGTH = 100
-        SERVICE_NAME = 'memcached'.freeze
-        SPAN_COMMAND = 'memcached.command'.freeze
-        SPAN_TYPE_COMMAND = 'memcached'.freeze
-        TAG_COMMAND = 'memcached.command'.freeze
-      end
-    end
-  end
-end

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

@@ -1,50 +0,0 @@
-require 'ddtrace/ext/net'
-require 'ddtrace/contrib/analytics'
-require 'ddtrace/contrib/dalli/ext'
-require 'ddtrace/contrib/dalli/quantize'
-
-module Datadog
-  module Contrib
-    module Dalli
-      # Instruments every interaction with the memcached server
-      module Instrumentation
-        def self.included(base)
-          base.send(:prepend, InstanceMethods)
-        end
-
-        # InstanceMethods - implementing instrumentation
-        module InstanceMethods
-          def request(op, *args)
-            tracer.trace(Datadog::Contrib::Dalli::Ext::SPAN_COMMAND) do |span|
-              span.resource = op.to_s.upcase
-              span.service = datadog_configuration[:service_name]
-              span.span_type = Datadog::Contrib::Dalli::Ext::SPAN_TYPE_COMMAND
-
-              # Set analytics sample rate
-              if Contrib::Analytics.enabled?(datadog_configuration[:analytics_enabled])
-                Contrib::Analytics.set_sample_rate(span, datadog_configuration[:analytics_sample_rate])
-              end
-
-              span.set_tag(Datadog::Ext::NET::TARGET_HOST, hostname)
-              span.set_tag(Datadog::Ext::NET::TARGET_PORT, port)
-              cmd = Datadog::Contrib::Dalli::Quantize.format_command(op, args)
-              span.set_tag(Datadog::Contrib::Dalli::Ext::TAG_COMMAND, cmd)
-
-              super
-            end
-          end
-
-          private
-
-          def tracer
-            datadog_configuration[:tracer]
-          end
-
-          def datadog_configuration
-            Datadog.configuration[:dalli, "#{hostname}:#{port}"] || Datadog.configuration[:dalli]
-          end
-        end
-      end
-    end
-  end
-end

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

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

@@ -1,67 +0,0 @@
-require 'ddtrace/contrib/patcher'
-require 'ddtrace/ext/app_types'
-require 'ddtrace/contrib/dalli/ext'
-require 'ddtrace/contrib/dalli/instrumentation'
-
-module Datadog
-  module Contrib
-    module Dalli
-      # Patcher enables patching of 'dalli' module.
-      module Patcher
-        include Contrib::Patcher
-
-        module_function
-
-        def target_version
-          Integration.version
-        end
-
-        def patch
-          add_pin!
-          ::Dalli::Server.send(:include, Instrumentation)
-        end
-
-        # DEPRECATED: Only kept for users still using `Dalli.datadog_pin` to configure.
-        #             Replaced by configuration API, i.e. `c.use :dalli`.
-        def add_pin!
-          DeprecatedPin
-            .new(
-              get_option(:service_name),
-              app: Ext::APP,
-              app_type: Datadog::Ext::AppTypes::CACHE,
-              tracer: get_option(:tracer)
-            ).onto(::Dalli)
-        end
-
-        def get_option(option)
-          Datadog.configuration[:dalli].get_option(option)
-        end
-
-        # Implementation of deprecated Pin, which raises warnings when accessed.
-        # To be removed when support for Datadog::Pin with Dalli is removed.
-        class DeprecatedPin < Datadog::Pin
-          include Datadog::DeprecatedPin
-
-          DEPRECATION_WARNING = %(
-            Use of Datadog::Pin with Dalli is DEPRECATED.
-            Upgrade to the configuration API using the migration guide here:
-            https://github.com/DataDog/dd-trace-rb/releases/tag/v0.11.0).freeze
-
-          def tracer=(tracer)
-            Datadog.configuration[:dalli][:tracer] = tracer
-          end
-
-          def service_name=(service_name)
-            Datadog.configuration[:dalli][:service_name] = service_name
-          end
-
-          def log_deprecation_warning(method_name)
-            do_once(method_name) do
-              Datadog.logger.warn("#{method_name}:#{DEPRECATION_WARNING}")
-            end
-          end
-        end
-      end
-    end
-  end
-end

data/lib/ddtrace/contrib/dalli/quantize.rb

@@ -1,22 +0,0 @@
-require 'ddtrace/contrib/dalli/ext'
-
-module Datadog
-  module Contrib
-    module Dalli
-      # Quantize contains dalli-specic quantization tools.
-      module Quantize
-        module_function
-
-        def format_command(operation, args)
-          placeholder = "#{operation} BLOB (OMITTED)"
-          command = [operation, *args].join(' ').strip
-          command = Utils.utf8_encode(command, binary: true, placeholder: placeholder)
-          Utils.truncate(command, Ext::QUANTIZE_MAX_CMD_LENGTH)
-        rescue => e
-          Datadog.logger.debug("Error sanitizing Dalli operation: #{e}")
-          placeholder
-        end
-      end
-    end
-  end
-end

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

@@ -1,25 +0,0 @@
-require 'ddtrace/contrib/configuration/settings'
-require 'ddtrace/contrib/delayed_job/ext'
-
-module Datadog
-  module Contrib
-    module DelayedJob
-      module Configuration
-        # Custom settings for the DelayedJob 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 :service_name, default: Ext::SERVICE_NAME
-        end
-      end
-    end
-  end
-end