ddtrace 0.16.1 → 0.17.0

data/lib/ddtrace.rb

@@ -44,30 +44,31 @@ module Datadog
 end
 
 require 'ddtrace/contrib/base'
-require 'ddtrace/contrib/active_model_serializers/patcher'
+require 'ddtrace/contrib/active_model_serializers/integration'
 require 'ddtrace/contrib/active_record/integration'
-require 'ddtrace/contrib/aws/patcher'
+require 'ddtrace/contrib/aws/integration'
 require 'ddtrace/contrib/concurrent_ruby/integration'
-require 'ddtrace/contrib/dalli/patcher'
-require 'ddtrace/contrib/delayed_job/patcher'
-require 'ddtrace/contrib/elasticsearch/patcher'
-require 'ddtrace/contrib/excon/patcher'
-require 'ddtrace/contrib/faraday/patcher'
-require 'ddtrace/contrib/grape/patcher'
-require 'ddtrace/contrib/graphql/patcher'
-require 'ddtrace/contrib/grpc/patcher'
-require 'ddtrace/contrib/http/patcher'
+require 'ddtrace/contrib/dalli/integration'
+require 'ddtrace/contrib/delayed_job/integration'
+require 'ddtrace/contrib/elasticsearch/integration'
+require 'ddtrace/contrib/excon/integration'
+require 'ddtrace/contrib/faraday/integration'
+require 'ddtrace/contrib/grape/integration'
+require 'ddtrace/contrib/graphql/integration'
+require 'ddtrace/contrib/grpc/integration'
+require 'ddtrace/contrib/http/integration'
 require 'ddtrace/contrib/integration'
-require 'ddtrace/contrib/mongodb/patcher'
-require 'ddtrace/contrib/mysql2/patcher'
-require 'ddtrace/contrib/racecar/patcher'
-require 'ddtrace/contrib/rack/patcher'
-require 'ddtrace/contrib/rails/patcher'
-require 'ddtrace/contrib/rake/patcher'
-require 'ddtrace/contrib/redis/patcher'
-require 'ddtrace/contrib/resque/patcher'
+require 'ddtrace/contrib/mysql2/integration'
+require 'ddtrace/contrib/mongodb/integration'
+require 'ddtrace/contrib/racecar/integration'
+require 'ddtrace/contrib/rack/integration'
+require 'ddtrace/contrib/rails/integration'
+require 'ddtrace/contrib/rake/integration'
+require 'ddtrace/contrib/redis/integration'
+require 'ddtrace/contrib/resque/integration'
 require 'ddtrace/contrib/rest_client/integration'
 require 'ddtrace/contrib/sequel/integration'
-require 'ddtrace/contrib/sidekiq/patcher'
-require 'ddtrace/contrib/sucker_punch/patcher'
+require 'ddtrace/contrib/sidekiq/integration'
+require 'ddtrace/contrib/sinatra/integration'
+require 'ddtrace/contrib/sucker_punch/integration'
 require 'ddtrace/monkey'

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

@@ -0,0 +1,23 @@
+require 'ddtrace/contrib/configuration/settings'
+require 'ddtrace/contrib/active_model_serializers/ext'
+
+module Datadog
+  module Contrib
+    module ActiveModelSerializers
+      module Configuration
+        # Custom settings for the ActiveModelSerializers integration
+        class Settings < Contrib::Configuration::Settings
+          option :service_name, default: Ext::SERVICE_NAME
+          option :tracer, default: Datadog.tracer do |value|
+            (value || Datadog.tracer).tap do |v|
+              # Make sure to update tracers of all subscriptions
+              Events.subscriptions.each do |subscription|
+                subscription.tracer = v
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/active_model_serializers/event.rb

@@ -1,4 +1,6 @@
+require 'ddtrace/ext/http'
 require 'ddtrace/contrib/active_support/notifications/event'
+require 'ddtrace/contrib/active_model_serializers/ext'
 
 module Datadog
   module Contrib
@@ -31,12 +33,12 @@ module Datadog
             # Set the resource name and serializer name
             res = resource(payload[:serializer])
             span.resource = res
-            span.set_tag('active_model_serializers.serializer', res)
+            span.set_tag(Ext::TAG_SERIALIZER, res)
 
             span.span_type = Datadog::Ext::HTTP::TEMPLATE
 
             # Will be nil in 0.9
-            span.set_tag('active_model_serializers.adapter', payload[:adapter].class) unless payload[:adapter].nil?
+            span.set_tag(Ext::TAG_ADAPTER, payload[:adapter].class) unless payload[:adapter].nil?
           end
 
           private

data/lib/ddtrace/contrib/active_model_serializers/events/render.rb

@@ -1,3 +1,4 @@
+require 'ddtrace/contrib/active_model_serializers/ext'
 require 'ddtrace/contrib/active_model_serializers/event'
 
 module Datadog
@@ -9,7 +10,6 @@ module Datadog
           include ActiveModelSerializers::Event
 
           EVENT_NAME = 'render.active_model_serializers'.freeze
-          SPAN_NAME = 'active_model_serializers.render'.freeze
 
           module_function
 
@@ -23,7 +23,7 @@ module Datadog
           end
 
           def span_name
-            self::SPAN_NAME
+            Ext::SPAN_RENDER
           end
         end
       end

data/lib/ddtrace/contrib/active_model_serializers/events/serialize.rb

@@ -1,3 +1,4 @@
+require 'ddtrace/contrib/active_model_serializers/ext'
 require 'ddtrace/contrib/active_model_serializers/event'
 
 module Datadog
@@ -9,7 +10,6 @@ module Datadog
           include ActiveModelSerializers::Event
 
           EVENT_NAME = '!serialize.active_model_serializers'.freeze
-          SPAN_NAME = 'active_model_serializers.serialize'.freeze
 
           module_function
 
@@ -26,7 +26,7 @@ module Datadog
           end
 
           def span_name
-            self::SPAN_NAME
+            Ext::SPAN_SERIALIZE
           end
         end
       end

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

@@ -0,0 +1,17 @@
+module Datadog
+  module Contrib
+    module ActiveModelSerializers
+      # ActiveModelSerializers integration constants
+      module Ext
+        APP = 'active_model_serializers'.freeze
+        SERVICE_NAME = 'active_model_serializers'.freeze
+
+        SPAN_RENDER = 'active_model_serializers.render'.freeze
+        SPAN_SERIALIZE = 'active_model_serializers.serialize'.freeze
+
+        TAG_ADAPTER = 'active_model_serializers.adapter'.freeze
+        TAG_SERIALIZER = 'active_model_serializers.serializer'.freeze
+      end
+    end
+  end
+end

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

@@ -0,0 +1,39 @@
+require 'ddtrace/contrib/integration'
+require 'ddtrace/contrib/active_model_serializers/configuration/settings'
+require 'ddtrace/contrib/active_model_serializers/patcher'
+
+module Datadog
+  module Contrib
+    module ActiveModelSerializers
+      # Description of ActiveModelSerializers integration
+      class Integration
+        include Contrib::Integration
+
+        register_as :active_model_serializers
+
+        def self.version
+          Gem.loaded_specs['active_model_serializers'] \
+            && Gem.loaded_specs['active_model_serializers'].version
+        end
+
+        def self.present?
+          super && defined?(::ActiveModel::Serializer)
+        end
+
+        def self.compatible?
+          super \
+            && defined?(::ActiveSupport::Notifications) \
+            && version >= Gem::Version.new('0.9.0')
+        end
+
+        def default_configuration
+          Configuration::Settings.new
+        end
+
+        def patcher
+          Patcher
+        end
+      end
+    end
+  end
+end

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

@@ -1,60 +1,41 @@
+require 'ddtrace/contrib/patcher'
 require 'ddtrace/ext/app_types'
-require 'ddtrace/ext/http'
+require 'ddtrace/contrib/active_model_serializers/ext'
 require 'ddtrace/contrib/active_model_serializers/events'
 
 module Datadog
   module Contrib
     module ActiveModelSerializers
-      # Provides instrumentation for ActiveModelSerializers through ActiveSupport instrumentation signals
+      # Patcher enables patching of 'active_model_serializers' module.
       module Patcher
-        include Base
+        include Contrib::Patcher
 
-        VERSION_REQUIRED = Gem::Version.new('0.9.0')
+        module_function
 
-        register_as :active_model_serializers
+        def patched?
+          done?(:active_model_serializers)
+        end
 
-        option :service_name, default: 'active_model_serializers'
-        option :tracer, default: Datadog.tracer do |value|
-          (value || Datadog.tracer).tap do |v|
-            # Make sure to update tracers of all subscriptions
-            Events.subscriptions.each do |subscription|
-              subscription.tracer = v
+        def patch
+          do_once(:active_model_serializers) do
+            begin
+              # Subscribe to ActiveModelSerializers events
+              Events.subscribe!
+
+              # Set service info
+              get_option(:tracer).set_service_info(
+                get_option(:service_name),
+                Ext::APP,
+                Datadog::Ext::AppTypes::WEB
+              )
+            rescue StandardError => e
+              Datadog::Tracer.log.error("Unable to apply ActiveModelSerializers integration: #{e}")
             end
           end
         end
 
-        class << self
-          def patch
-            return patched? if patched? || !compatible?
-
-            # Subscribe to ActiveModelSerializers events
-            Events.subscribe!
-
-            # Set service info
-            configuration[:tracer].set_service_info(
-              configuration[:service_name],
-              'active_model_serializers',
-              Ext::AppTypes::WEB
-            )
-
-            @patched = true
-          end
-
-          def patched?
-            return @patched if defined?(@patched)
-            @patched = false
-          end
-
-          private
-
-          def configuration
-            Datadog.configuration[:active_model_serializers]
-          end
-
-          def compatible?
-            Gem.loaded_specs['active_model_serializers'] && Gem.loaded_specs['activesupport'] \
-              && Gem.loaded_specs['active_model_serializers'].version >= VERSION_REQUIRED
-          end
+        def get_option(option)
+          Datadog.configuration[:active_model_serializers].get_option(option)
         end
       end
     end

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

@@ -1,16 +1,17 @@
 require 'ddtrace/contrib/configuration/settings'
+require 'ddtrace/contrib/active_record/ext'
 require 'ddtrace/contrib/active_record/utils'
 
 module Datadog
   module Contrib
     module ActiveRecord
       module Configuration
-        # Unique settings for ActiveRecord
+        # Custom settings for the ActiveRecord integration
         class Settings < Contrib::Configuration::Settings
           option :orm_service_name
           option :service_name, depends_on: [:tracer] do |value|
             (value || Utils.adapter_name).tap do |service_name|
-              tracer.set_service_info(service_name, 'active_record', Ext::AppTypes::DB)
+              tracer.set_service_info(service_name, Ext::APP, Datadog::Ext::AppTypes::DB)
             end
           end
 

data/lib/ddtrace/contrib/active_record/events/instantiation.rb

@@ -1,3 +1,4 @@
+require 'ddtrace/contrib/active_record/ext'
 require 'ddtrace/contrib/active_record/event'
 
 module Datadog
@@ -9,8 +10,6 @@ module Datadog
           include ActiveRecord::Event
 
           EVENT_NAME = 'instantiation.active_record'.freeze
-          SPAN_NAME = 'active_record.instantiation'.freeze
-          DEFAULT_SERVICE_NAME = 'active_record'.freeze
 
           module_function
 
@@ -24,7 +23,7 @@ module Datadog
           end
 
           def span_name
-            self::SPAN_NAME
+            Ext::SPAN_INSTANTIATION
           end
 
           def process(span, event, _id, payload)
@@ -34,13 +33,13 @@ module Datadog
                            elsif span.parent
                              span.parent.service
                            else
-                             self::DEFAULT_SERVICE_NAME
+                             Ext::SERVICE_NAME
                            end
 
             span.resource = payload.fetch(:class_name)
-            span.span_type = 'custom'
-            span.set_tag('active_record.instantiation.class_name', payload.fetch(:class_name))
-            span.set_tag('active_record.instantiation.record_count', payload.fetch(:record_count))
+            span.span_type = Ext::SPAN_TYPE_INSTANTIATION
+            span.set_tag(Ext::TAG_INSTANTIATION_CLASS_NAME, payload.fetch(:class_name))
+            span.set_tag(Ext::TAG_INSTANTIATION_RECORD_COUNT, payload.fetch(:record_count))
           rescue StandardError => e
             Datadog::Tracer.log.debug(e.message)
           end

data/lib/ddtrace/contrib/active_record/events/sql.rb

@@ -1,3 +1,5 @@
+require 'ddtrace/ext/net'
+require 'ddtrace/contrib/active_record/ext'
 require 'ddtrace/contrib/active_record/event'
 
 module Datadog
@@ -9,7 +11,7 @@ module Datadog
           include ActiveRecord::Event
 
           EVENT_NAME = 'sql.active_record'.freeze
-          SPAN_NAME = 'active_record.sql'.freeze
+          PAYLOAD_CACHE = 'CACHE'.freeze
 
           module_function
 
@@ -18,7 +20,7 @@ module Datadog
           end
 
           def span_name
-            self::SPAN_NAME
+            Ext::SPAN_SQL
           end
 
           def process(span, event, _id, payload)
@@ -35,13 +37,13 @@ module Datadog
             # Find out if the SQL query has been cached in this request. This meta is really
             # helpful to users because some spans may have 0ns of duration because the query
             # is simply cached from memory, so the notification is fired with start == finish.
-            cached = payload[:cached] || (payload[:name] == 'CACHE')
+            cached = payload[:cached] || (payload[:name] == PAYLOAD_CACHE)
 
-            span.set_tag('active_record.db.vendor', adapter_name)
-            span.set_tag('active_record.db.name', config[:database])
-            span.set_tag('active_record.db.cached', cached) if cached
-            span.set_tag('out.host', config[:host]) if config[:host]
-            span.set_tag('out.port', config[:port]) if config[:port]
+            span.set_tag(Ext::TAG_DB_VENDOR, adapter_name)
+            span.set_tag(Ext::TAG_DB_NAME, config[:database])
+            span.set_tag(Ext::TAG_DB_CACHED, cached) if cached
+            span.set_tag(Datadog::Ext::NET::TARGET_HOST, config[:host]) if config[:host]
+            span.set_tag(Datadog::Ext::NET::TARGET_PORT, config[:port]) if config[:port]
           rescue StandardError => e
             Datadog::Tracer.log.debug(e.message)
           end

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

@@ -0,0 +1,22 @@
+module Datadog
+  module Contrib
+    module ActiveRecord
+      # ActiveRecord integration constants
+      module Ext
+        APP = 'active_record'.freeze
+        SERVICE_NAME = 'active_record'.freeze
+
+        SPAN_INSTANTIATION = 'active_record.instantiation'.freeze
+        SPAN_SQL = 'active_record.sql'.freeze
+
+        SPAN_TYPE_INSTANTIATION = 'custom'.freeze
+
+        TAG_DB_CACHED = 'active_record.db.cached'.freeze
+        TAG_DB_NAME = 'active_record.db.name'.freeze
+        TAG_DB_VENDOR = 'active_record.db.vendor'.freeze
+        TAG_INSTANTIATION_CLASS_NAME = 'active_record.instantiation.class_name'.freeze
+        TAG_INSTANTIATION_RECORD_COUNT = 'active_record.instantiation.record_count'.freeze
+      end
+    end
+  end
+end

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

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

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

@@ -0,0 +1,21 @@
+module Datadog
+  module Contrib
+    module Aws
+      # AWS integration constants
+      module Ext
+        APP = 'aws'.freeze
+        SERVICE_NAME = 'aws'.freeze
+
+        SPAN_COMMAND = 'aws.command'.freeze
+
+        TAG_AGENT = 'aws.agent'.freeze
+        TAG_OPERATION = 'aws.operation'.freeze
+        TAG_REGION = 'aws.region'.freeze
+        TAG_PATH = 'path'.freeze
+        TAG_HOST = 'host'.freeze
+
+        TAG_DEFAULT_AGENT = 'aws-sdk-ruby'.freeze
+      end
+    end
+  end
+end