ddtrace 0.38.0

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

@@ -0,0 +1,43 @@
+require 'ddtrace/contrib/integration'
+require 'ddtrace/contrib/httprb/configuration/settings'
+require 'ddtrace/contrib/configuration/resolvers/pattern_resolver'
+require 'ddtrace/contrib/httprb/patcher'
+
+module Datadog
+  module Contrib
+    module Httprb
+      # Description of Httprb integration
+      class Integration
+        include Contrib::Integration
+
+        MINIMUM_VERSION = Gem::Version.new('2.0.0')
+
+        register_as :httprb
+
+        def self.version
+          Gem.loaded_specs['http'] && Gem.loaded_specs['http'].version
+        end
+
+        def self.loaded?
+          !defined?(::HTTP).nil? && !defined?(::HTTP::Client).nil?
+        end
+
+        def self.compatible?
+          super && version >= MINIMUM_VERSION
+        end
+
+        def default_configuration
+          Configuration::Settings.new
+        end
+
+        def patcher
+          Patcher
+        end
+
+        def resolver
+          @resolver ||= Contrib::Configuration::Resolvers::PatternResolver.new
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,35 @@
+require 'ddtrace/contrib/patcher'
+require 'ddtrace/contrib/httprb/instrumentation'
+
+module Datadog
+  module Contrib
+    # Datadog Httprb integration.
+    module Httprb
+      # Patcher enables patching of 'httprb' module.
+      module Patcher
+        include Contrib::Patcher
+
+        module_function
+
+        def patched?
+          done?(:httprb)
+        end
+
+        def target_version
+          Integration.version
+        end
+
+        # patch applies our patch
+        def patch
+          do_once(:httprb) do
+            begin
+              ::HTTP::Client.send(:include, Instrumentation)
+            rescue StandardError => e
+              Datadog::Logger.error("Unable to apply httprb integration: #{e}")
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/integration.rb

@@ -0,0 +1,16 @@
+require 'ddtrace/contrib/configurable'
+require 'ddtrace/contrib/patchable'
+require 'ddtrace/contrib/registerable'
+
+module Datadog
+  module Contrib
+    # Base provides features that are shared across all integrations
+    module Integration
+      def self.included(base)
+        base.send(:include, Configurable)
+        base.send(:include, Patchable)
+        base.send(:include, Registerable)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,25 @@
+require 'ddtrace/contrib/configuration/settings'
+require 'ddtrace/contrib/kafka/ext'
+
+module Datadog
+  module Contrib
+    module Kafka
+      module Configuration
+        # Custom settings for the Kafka 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/kafka/consumer_event.rb

@@ -0,0 +1,14 @@
+module Datadog
+  module Contrib
+    module Kafka
+      # Defines basic behaviors for an event for a consumer.
+      module ConsumerEvent
+        def process(span, _event, _id, payload)
+          super
+
+          span.set_tag(Ext::TAG_GROUP, payload[:group_id])
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/kafka/consumer_group_event.rb

@@ -0,0 +1,14 @@
+module Datadog
+  module Contrib
+    module Kafka
+      # Defines basic behaviors for an event for a consumer group.
+      module ConsumerGroupEvent
+        def process(span, _event, _id, payload)
+          super
+
+          span.resource = payload[:group_id]
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,51 @@
+require 'ddtrace/contrib/analytics'
+require 'ddtrace/contrib/active_support/notifications/event'
+require 'ddtrace/contrib/kafka/ext'
+
+module Datadog
+  module Contrib
+    module Kafka
+      # Defines basic behaviors for an ActiveSupport event.
+      module Event
+        def self.included(base)
+          base.send(:include, ActiveSupport::Notifications::Event)
+          base.send(:extend, ClassMethods)
+        end
+
+        # Class methods for Kafka events.
+        module ClassMethods
+          def event_name
+            self::EVENT_NAME
+          end
+
+          def span_options
+            { service: configuration[:service_name] }
+          end
+
+          def tracer
+            -> { configuration[:tracer] }
+          end
+
+          def configuration
+            Datadog.configuration[:kafka]
+          end
+
+          def process(span, _event, _id, payload)
+            span.service = configuration[:service_name]
+            span.set_tag(Ext::TAG_CLIENT, payload[:client_id])
+
+            # 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_error(payload[:exception_object]) if payload[:exception_object]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/kafka/events.rb

@@ -0,0 +1,44 @@
+require 'ddtrace/contrib/kafka/events/connection/request'
+require 'ddtrace/contrib/kafka/events/consumer/process_batch'
+require 'ddtrace/contrib/kafka/events/consumer/process_message'
+require 'ddtrace/contrib/kafka/events/consumer_group/heartbeat'
+require 'ddtrace/contrib/kafka/events/consumer_group/join_group'
+require 'ddtrace/contrib/kafka/events/consumer_group/leave_group'
+require 'ddtrace/contrib/kafka/events/consumer_group/sync_group'
+require 'ddtrace/contrib/kafka/events/produce_operation/send_messages'
+require 'ddtrace/contrib/kafka/events/producer/deliver_messages'
+
+module Datadog
+  module Contrib
+    module Kafka
+      # Defines collection of instrumented Kafka events
+      module Events
+        ALL = [
+          Events::Connection::Request,
+          Events::Consumer::ProcessBatch,
+          Events::Consumer::ProcessMessage,
+          Events::ConsumerGroup::Heartbeat,
+          Events::ConsumerGroup::JoinGroup,
+          Events::ConsumerGroup::LeaveGroup,
+          Events::ConsumerGroup::SyncGroup,
+          Events::ProduceOperation::SendMessages,
+          Events::Producer::DeliverMessages
+        ].freeze
+
+        module_function
+
+        def all
+          self::ALL
+        end
+
+        def subscriptions
+          all.collect(&:subscriptions).collect(&:to_a).flatten
+        end
+
+        def subscribe!
+          all.each(&:subscribe!)
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/kafka/events/connection/request.rb

@@ -0,0 +1,34 @@
+require 'ddtrace/contrib/kafka/ext'
+require 'ddtrace/contrib/kafka/event'
+
+module Datadog
+  module Contrib
+    module Kafka
+      module Events
+        module Connection
+          # Defines instrumentation for request.connection.kafka event
+          module Request
+            include Kafka::Event
+
+            EVENT_NAME = 'request.connection.kafka'.freeze
+
+            def self.process(span, _event, _id, payload)
+              super
+
+              span.resource = payload[:api]
+
+              span.set_tag(Ext::TAG_REQUEST_SIZE, payload[:request_size]) if payload.key?(:request_size)
+              span.set_tag(Ext::TAG_RESPONSE_SIZE, payload[:response_size]) if payload.key?(:response_size)
+            end
+
+            module_function
+
+            def span_name
+              Ext::SPAN_CONNECTION_REQUEST
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/kafka/events/consumer/process_batch.rb

@@ -0,0 +1,41 @@
+require 'ddtrace/contrib/kafka/ext'
+require 'ddtrace/contrib/kafka/event'
+require 'ddtrace/contrib/kafka/consumer_event'
+
+module Datadog
+  module Contrib
+    module Kafka
+      module Events
+        module Consumer
+          # Defines instrumentation for process_batch.consumer.kafka event
+          module ProcessBatch
+            include Kafka::Event
+            extend Kafka::ConsumerEvent
+
+            EVENT_NAME = 'process_batch.consumer.kafka'.freeze
+
+            def self.process(span, _event, _id, payload)
+              super
+
+              span.resource = payload[:topic]
+
+              span.set_tag(Ext::TAG_TOPIC, payload[:topic]) if payload.key?(:topic)
+              span.set_tag(Ext::TAG_MESSAGE_COUNT, payload[:message_count]) if payload.key?(:message_count)
+              span.set_tag(Ext::TAG_PARTITION, payload[:partition]) if payload.key?(:partition)
+              if payload.key?(:highwater_mark_offset)
+                span.set_tag(Ext::TAG_HIGHWATER_MARK_OFFSET, payload[:highwater_mark_offset])
+              end
+              span.set_tag(Ext::TAG_OFFSET_LAG, payload[:offset_lag]) if payload.key?(:offset_lag)
+            end
+
+            module_function
+
+            def span_name
+              Ext::SPAN_PROCESS_BATCH
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/kafka/events/consumer/process_message.rb

@@ -0,0 +1,39 @@
+require 'ddtrace/contrib/kafka/ext'
+require 'ddtrace/contrib/kafka/event'
+require 'ddtrace/contrib/kafka/consumer_event'
+
+module Datadog
+  module Contrib
+    module Kafka
+      module Events
+        module Consumer
+          # Defines instrumentation for process_message.consumer.kafka event
+          module ProcessMessage
+            include Kafka::Event
+            extend Kafka::ConsumerEvent
+
+            EVENT_NAME = 'process_message.consumer.kafka'.freeze
+
+            def self.process(span, _event, _id, payload)
+              super
+
+              span.resource = payload[:topic]
+
+              span.set_tag(Ext::TAG_TOPIC, payload[:topic]) if payload.key?(:topic)
+              span.set_tag(Ext::TAG_MESSAGE_KEY, payload[:key]) if payload.key?(:key)
+              span.set_tag(Ext::TAG_PARTITION, payload[:partition]) if payload.key?(:partition)
+              span.set_tag(Ext::TAG_OFFSET, payload[:offset]) if payload.key?(:offset)
+              span.set_tag(Ext::TAG_OFFSET_LAG, payload[:offset_lag]) if payload.key?(:offset_lag)
+            end
+
+            module_function
+
+            def span_name
+              Ext::SPAN_PROCESS_MESSAGE
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/kafka/events/consumer_group/heartbeat.rb

@@ -0,0 +1,39 @@
+require 'ddtrace/contrib/kafka/ext'
+require 'ddtrace/contrib/kafka/event'
+require 'ddtrace/contrib/kafka/consumer_event'
+require 'ddtrace/contrib/kafka/consumer_group_event'
+
+module Datadog
+  module Contrib
+    module Kafka
+      module Events
+        module ConsumerGroup
+          # Defines instrumentation for heartbeat.consumer.kafka event
+          module Heartbeat
+            include Kafka::Event
+            extend Kafka::ConsumerEvent
+            extend Kafka::ConsumerGroupEvent
+
+            EVENT_NAME = 'heartbeat.consumer.kafka'.freeze
+
+            def self.process(span, _event, _id, payload)
+              super
+
+              if payload.key?(:topic_partitions)
+                payload[:topic_partitions].each do |topic, partitions|
+                  span.set_tag("#{Ext::TAG_TOPIC_PARTITIONS}.#{topic}", partitions)
+                end
+              end
+            end
+
+            module_function
+
+            def span_name
+              Ext::SPAN_CONSUMER_HEARTBEAT
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/kafka/events/consumer_group/join_group.rb

@@ -0,0 +1,29 @@
+require 'ddtrace/contrib/kafka/ext'
+require 'ddtrace/contrib/kafka/event'
+require 'ddtrace/contrib/kafka/consumer_event'
+require 'ddtrace/contrib/kafka/consumer_group_event'
+
+module Datadog
+  module Contrib
+    module Kafka
+      module Events
+        module ConsumerGroup
+          # Defines instrumentation for join_group.consumer.kafka event
+          module JoinGroup
+            include Kafka::Event
+            extend Kafka::ConsumerEvent
+            extend Kafka::ConsumerGroupEvent
+
+            EVENT_NAME = 'join_group.consumer.kafka'.freeze
+
+            module_function
+
+            def span_name
+              Ext::SPAN_CONSUMER_JOIN_GROUP
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ddtrace/contrib/kafka/events/consumer_group/leave_group.rb

@@ -0,0 +1,29 @@
+require 'ddtrace/contrib/kafka/ext'
+require 'ddtrace/contrib/kafka/event'
+require 'ddtrace/contrib/kafka/consumer_event'
+require 'ddtrace/contrib/kafka/consumer_group_event'
+
+module Datadog
+  module Contrib
+    module Kafka
+      module Events
+        module ConsumerGroup
+          # Defines instrumentation for leave_group.consumer.kafka event
+          module LeaveGroup
+            include Kafka::Event
+            extend Kafka::ConsumerEvent
+            extend Kafka::ConsumerGroupEvent
+
+            EVENT_NAME = 'leave_group.consumer.kafka'.freeze
+
+            module_function
+
+            def span_name
+              Ext::SPAN_CONSUMER_LEAVE_GROUP
+            end
+          end
+        end
+      end
+    end
+  end
+end