ddtrace 0.32.0 → 0.33.0

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

@@ -0,0 +1,36 @@
+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
+
+        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)
+        end
+
+        def self.compatible?
+          super && version >= Gem::Version.new('0.5.14')
+        end
+
+        def default_configuration
+          Configuration::Settings.new
+        end
+
+        def patcher
+          Patcher
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,30 @@
+require 'ddtrace/contrib/patcher'
+require 'ddtrace/contrib/presto/ext'
+require 'ddtrace/contrib/presto/instrumentation'
+
+module Datadog
+  module Contrib
+    module Presto
+      # Patcher enables patching of 'presto-client' module.
+      module Patcher
+        include Contrib::Patcher
+
+        module_function
+
+        def patched?
+          done?(:presto)
+        end
+
+        def patch
+          do_once(:presto) do
+            begin
+              ::Presto::Client::Client.send(:include, Instrumentation::Client)
+            rescue StandardError => e
+              Datadog::Logger.log.error("Unable to apply Presto integration: #{e}")
+            end
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -15,12 +15,12 @@ module Datadog
           Gem.loaded_specs['racecar'] && Gem.loaded_specs['racecar'].version
         end
 
-        def self.present?
-          super && defined?(::Racecar)
+        def self.loaded?
+          defined?(::Racecar) && defined?(::ActiveSupport::Notifications)
         end
 
         def self.compatible?
-          super && defined?(::ActiveSupport::Notifications)
+          super && version >= Gem::Version.new('0.3.5')
         end
 
         def default_configuration

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

@@ -15,8 +15,12 @@ module Datadog
           Gem.loaded_specs['rack'] && Gem.loaded_specs['rack'].version
         end
 
-        def self.present?
-          super && defined?(::Rack)
+        def self.loaded?
+          defined?(::Rack)
+        end
+
+        def self.compatible?
+          super && version >= Gem::Version.new('1.1.0')
         end
 
         def default_configuration

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

@@ -15,13 +15,16 @@ module Datadog
           Gem.loaded_specs['rails'] && Gem.loaded_specs['rails'].version
         end
 
-        def self.present?
+        def self.loaded?
           defined?(::Rails)
         end
 
         def self.compatible?
-          return false if ENV['DISABLE_DATADOG_RAILS']
-          super && defined?(::Rails::VERSION) && ::Rails::VERSION::MAJOR.to_i >= 3
+          super && version >= Gem::Version.new('3.0')
+        end
+
+        def self.patchable?
+          super && !ENV.key?('DISABLE_DATADOG_RAILS')
         end
 
         def default_configuration

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

@@ -15,8 +15,12 @@ module Datadog
           Gem.loaded_specs['rake'] && Gem.loaded_specs['rake'].version
         end
 
-        def self.present?
-          super && defined?(::Rake)
+        def self.loaded?
+          defined?(::Rake)
+        end
+
+        def self.compatible?
+          super && version >= Gem::Version.new('12.0')
         end
 
         def default_configuration

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

@@ -0,0 +1,36 @@
+require 'ddtrace/contrib/redis/vendor/resolver'
+
+module Datadog
+  module Contrib
+    module Redis
+      module Configuration
+        # Converts Symbols, Strings, and Hashes to a normalized connection settings Hash.
+        class Resolver < Contrib::Configuration::Resolver
+          def resolve(key_or_hash)
+            return :default if key_or_hash == :default
+
+            normalize(connection_resolver.resolve(key_or_hash))
+          end
+
+          def normalize(hash)
+            return { url: hash[:url] } if hash[:scheme] == 'unix'
+
+            # Connexion strings are always converted to host, port, db and scheme
+            # but the host, port, db and scheme will generate the :url only after
+            # establishing a first connexion
+            {
+              host: hash[:host],
+              port: hash[:port],
+              db: hash[:db],
+              scheme: hash[:scheme]
+            }
+          end
+
+          def connection_resolver
+            @connection_resolver ||= ::Datadog::Contrib::Redis::Vendor::Resolver.new
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -15,12 +15,12 @@ module Datadog
           Gem.loaded_specs['redis'] && Gem.loaded_specs['redis'].version
         end
 
-        def self.present?
-          super && defined?(::Redis)
+        def self.loaded?
+          defined?(::Redis)
         end
 
         def self.compatible?
-          !version.nil? && version >= Gem::Version.new('3.0.0')
+          super && version >= Gem::Version.new('3.2')
         end
 
         def default_configuration
@@ -30,6 +30,10 @@ module Datadog
         def patcher
           Patcher
         end
+
+        def resolver
+          @resolver ||= Configuration::Resolver.new
+        end
       end
     end
   end

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

@@ -1,5 +1,6 @@
 require 'ddtrace/contrib/patcher'
 require 'ddtrace/contrib/redis/ext'
+require 'ddtrace/contrib/redis/configuration/resolver'
 
 module Datadog
   module Contrib
@@ -72,16 +73,24 @@ module Datadog
             def datadog_pin
               @datadog_pin ||= begin
                 pin = Datadog::Pin.new(
-                  Datadog.configuration[:redis][:service_name],
+                  datadog_configuration[:service_name],
                   app: Ext::APP,
                   app_type: Datadog::Ext::AppTypes::DB,
-                  tracer: Datadog.configuration[:redis][:tracer]
+                  tracer: datadog_configuration[:tracer]
                 )
                 pin.onto(self)
               end
             end
+
+            private
+
+            def datadog_configuration
+              Datadog.configuration[:redis, options]
+            end
           end
         end
+        # rubocop:enable Metrics/MethodLength
+        # rubocop:enable Metrics/BlockLength
       end
     end
   end

data/lib/ddtrace/contrib/redis/vendor/resolver.rb

@@ -0,0 +1,159 @@
+require 'uri'
+
+# NOTE: This code is copied directly from Redis.
+#       Its purpose is to resolve connection information.
+#       It exists here only because it doesn't exist in the redis
+#       library as a separated module and it allows to avoid
+#       instantiating a new Redis::Client for resolving the connection
+module Datadog
+  module Contrib
+    module Redis
+      module Vendor
+        class Resolver # :nodoc:
+          # Connection DEFAULTS for a Redis::Client are unchanged for
+          # the integration supported options.
+          # https://github.com/redis/redis-rb/blob/v3.0.0/lib/redis/client.rb#L6-L14
+          # https://github.com/redis/redis-rb/blob/v4.1.3/lib/redis/client.rb#L10-L26
+          # Since the integration takes in consideration only few attributes, all
+          # versions are compatible for :url, :scheme, :host, :port, :db
+          DEFAULTS = {
+            url: -> { ENV['REDIS_URL'] },
+            scheme: 'redis',
+            host: '127.0.0.1',
+            port: 6379,
+            path: nil,
+            # :timeout => 5.0,
+            password: nil,
+            db: 0 # ,
+            # :driver => nil,
+            # :id => nil,
+            # :tcp_keepalive => 0,
+            # :reconnect_attempts => 1,
+            # :reconnect_delay => 0,
+            # :reconnect_delay_max => 0.5,
+            # :inherit_socket => false
+          }.freeze
+
+          def resolve(options)
+            _parse_options(options)
+          end
+
+          # rubocop:disable Metrics/AbcSize
+          # rubocop:disable Metrics/MethodLength
+          # rubocop:disable Metrics/PerceivedComplexity
+          #
+          # This method is a subset of the implementation provided in v3.0.0
+          # https://github.com/redis/redis-rb/blob/v3.0.0/lib/redis/client.rb
+          # https://github.com/redis/redis-rb/blob/v4.1.3/lib/redis/client.rb
+          #
+          # Since it has been backported from the original gem, some linting
+          # cops have been disabled
+          def _parse_options(options)
+            # https://github.com/redis/redis-rb/blob/v4.1.3/lib/redis/client.rb#L404
+            # Early return for modern client options
+            return options if options[:_parsed]
+
+            defaults = DEFAULTS.dup
+            options = options.dup
+
+            defaults.keys.each do |key|
+              # Fill in defaults if needed
+              if defaults[key].respond_to?(:call)
+                defaults[key] = defaults[key].call
+              end
+
+              # Symbolize only keys that are needed
+              options[key] = options[key.to_s] if options.key?(key.to_s)
+            end
+
+            url = options[:url]
+            url = defaults[:url] if url.nil?
+
+            # Override defaults from URL if given
+            if url
+              uri = URI(url)
+
+              if uri.scheme == 'unix'
+                defaults[:path] = uri.path
+              elsif uri.scheme == 'redis' || uri.scheme == 'rediss'
+                defaults[:scheme]   = uri.scheme
+                defaults[:host]     = uri.host if uri.host
+                defaults[:port]     = uri.port if uri.port
+                defaults[:password] = CGI.unescape(uri.password) if uri.password
+                defaults[:db]       = uri.path[1..-1].to_i if uri.path
+                defaults[:role] = :master
+              else
+                raise ArgumentError, "invalid uri scheme '#{uri.scheme}'"
+              end
+
+              # defaults[:ssl] = true if uri.scheme == "rediss"
+            end
+
+            # Use default when option is not specified or nil
+            defaults.keys.each do |key|
+              options[key] = defaults[key] if options[key].nil?
+            end
+
+            if options[:path]
+              # Unix socket
+              options[:scheme] = 'unix'
+              options.delete(:host)
+              options.delete(:port)
+            else
+              # TCP socket
+              options[:host] = options[:host].to_s
+              options[:port] = options[:port].to_i
+            end
+
+            # Options ignored by the integration
+            #
+            # if options.has_key?(:timeout)
+            #   options[:connect_timeout] ||= options[:timeout]
+            #   options[:read_timeout]    ||= options[:timeout]
+            #   options[:write_timeout]   ||= options[:timeout]
+            # end
+
+            # options[:connect_timeout] = Float(options[:connect_timeout])
+            # options[:read_timeout]    = Float(options[:read_timeout])
+            # options[:write_timeout]   = Float(options[:write_timeout])
+
+            # options[:reconnect_attempts] = options[:reconnect_attempts].to_i
+            # options[:reconnect_delay] = options[:reconnect_delay].to_f
+            # options[:reconnect_delay_max] = options[:reconnect_delay_max].to_f
+
+            options[:db] = options[:db].to_i
+            # options[:driver] = _parse_driver(options[:driver]) || Connection.drivers.last
+
+            # case options[:tcp_keepalive]
+            # when Hash
+            #   [:time, :intvl, :probes].each do |key|
+            #     unless options[:tcp_keepalive][key].is_a?(Integer)
+            #       raise "Expected the #{key.inspect} key in :tcp_keepalive to be an Integer"
+            #     end
+            #   end
+
+            # when Integer
+            #   if options[:tcp_keepalive] >= 60
+            #     options[:tcp_keepalive] = {:time => options[:tcp_keepalive] - 20, :intvl => 10, :probes => 2}
+
+            #   elsif options[:tcp_keepalive] >= 30
+            #     options[:tcp_keepalive] = {:time => options[:tcp_keepalive] - 10, :intvl => 5, :probes => 2}
+
+            #   elsif options[:tcp_keepalive] >= 5
+            #     options[:tcp_keepalive] = {:time => options[:tcp_keepalive] - 2, :intvl => 2, :probes => 1}
+            #   end
+            # end
+
+            options[:_parsed] = true
+
+            options
+          end
+
+          # rubocop:enable Metrics/AbcSize
+          # rubocop:enable Metrics/MethodLength
+          # rubocop:enable Metrics/PerceivedComplexity
+        end
+      end
+    end
+  end
+end

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

@@ -15,8 +15,14 @@ module Datadog
           Gem.loaded_specs['resque'] && Gem.loaded_specs['resque'].version
         end
 
-        def self.present?
-          super && defined?(::Resque)
+        def self.loaded?
+          defined?(::Resque)
+        end
+
+        def self.compatible?
+          super \
+            && version >= Gem::Version.new('1.0') \
+            && version < Gem::Version.new('2.0')
         end
 
         def default_configuration

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

@@ -14,8 +14,12 @@ module Datadog
           Gem.loaded_specs['rest-client'] && Gem.loaded_specs['rest-client'].version
         end
 
-        def self.present?
-          super && defined?(::RestClient::Request)
+        def self.loaded?
+          defined?(::RestClient::Request)
+        end
+
+        def self.compatible?
+          super && version >= Gem::Version.new('1.8')
         end
 
         def default_configuration

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

@@ -15,8 +15,12 @@ module Datadog
           Gem.loaded_specs['sequel'] && Gem.loaded_specs['sequel'].version
         end
 
-        def self.present?
-          super && defined?(::Sequel)
+        def self.loaded?
+          defined?(::Sequel)
+        end
+
+        def self.compatible?
+          super && version >= Gem::Version.new('3.41')
         end
 
         def default_configuration

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

@@ -12,14 +12,16 @@ module Datadog
 
         register_as :shoryuken
 
-        class << self
-          def version
-            Gem.loaded_specs['shoryuken'] && Gem.loaded_specs['shoryuken'].version
-          end
+        def self.version
+          Gem.loaded_specs['shoryuken'] && Gem.loaded_specs['shoryuken'].version
+        end
+
+        def self.loaded?
+          defined?(::Shoryuken)
+        end
 
-          def present?
-            super && defined?(::Shoryuken)
-          end
+        def self.compatible?
+          super && version >= Gem::Version.new('4.0.2')
         end
 
         def default_configuration