Sha256: 0a8a6b767cf4a3a6c7945d2896abdaee0b1748aa370c78c732266eedef84b773

Contents?: true

Size: 1.92 KB

Versions: 4

Compression:

Stored size: 1.92 KB

Contents

# frozen_string_literal: true

# typed: true

require 'datadog/tracing'
require 'datadog/tracing/metadata/ext'
require 'datadog/tracing/contrib/analytics'

module Datadog
  module Tracing
    module Contrib
      module Sneakers
        # Tracer is a Sneakers server-side middleware which traces executed jobs
        class Tracer
          def initialize(app, *args)
            @app = app
            @args = args
          end

          def call(deserialized_msg, delivery_info, metadata, handler)
            trace_options = {
              service: configuration[:service_name],
              span_type: Tracing::Metadata::Ext::AppTypes::TYPE_WORKER,
              on_error: configuration[:error_handler]
            }

            Tracing.trace(Ext::SPAN_JOB, **trace_options) do |request_span|
              request_span.set_tag(Tracing::Metadata::Ext::TAG_COMPONENT, Ext::TAG_COMPONENT)
              request_span.set_tag(Tracing::Metadata::Ext::TAG_OPERATION, Ext::TAG_OPERATION_JOB)

              # Set analytics sample rate
              if Contrib::Analytics.enabled?(configuration[:analytics_enabled])
                Contrib::Analytics.set_sample_rate(request_span, configuration[:analytics_sample_rate])
              end

              # Measure service stats
              Contrib::Analytics.set_measured(request_span)

              request_span.resource = @app.to_proc.binding.eval('self.class').to_s
              request_span.set_tag(Ext::TAG_JOB_ROUTING_KEY, delivery_info.routing_key)
              request_span.set_tag(Ext::TAG_JOB_QUEUE, delivery_info.consumer.queue.name)

              request_span.set_tag(Ext::TAG_JOB_BODY, deserialized_msg) if configuration[:tag_body]

              @app.call(deserialized_msg, delivery_info, metadata, handler)
            end
          end

          private

          def configuration
            Datadog.configuration.tracing[:sneakers]
          end
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version Path
ddtrace-1.2.0 lib/datadog/tracing/contrib/sneakers/tracer.rb
ddtrace-1.1.0 lib/datadog/tracing/contrib/sneakers/tracer.rb
ddtrace-1.0.0 lib/datadog/tracing/contrib/sneakers/tracer.rb
ddtrace-1.0.0.beta2 lib/datadog/tracing/contrib/sneakers/tracer.rb