Contents

# frozen_string_literal: true

# Copyright The OpenTelemetry Authors
#
# SPDX-License-Identifier: Apache-2.0

module OpenTelemetry
  module Instrumentation
    module Sidekiq
      module Middlewares
        module Server
          # TracerMiddleware propagates context and instruments Sidekiq requests
          # by way of its middleware system
          class TracerMiddleware
            def call(_worker, msg, _queue)
              extracted_context = OpenTelemetry.propagation.extract(msg)
              OpenTelemetry::Context.with_current(extracted_context) do
                tracer.in_span(
                  span_name(msg),
                  attributes: build_attributes(msg),
                  kind: :consumer
                ) do |span|
                  span.add_event('created_at', timestamp: msg['created_at'])
                  span.add_event('enqueued_at', timestamp: msg['enqueued_at'])
                  yield
                end
              end
            end

            private

            def build_attributes(msg)
              attributes = {
                'messaging.system' => 'sidekiq',
                'messaging.sidekiq.job_class' => msg['wrapped']&.to_s || msg['class'],
                'messaging.message_id' => msg['jid'],
                'messaging.destination' => msg['queue'],
                'messaging.destination_kind' => 'queue'
              }
              attributes['peer.service'] = config[:peer_service] if config[:peer_service]
              attributes
            end

            def span_name(msg)
              if config[:enable_job_class_span_names]
                "#{msg['wrapped']&.to_s || msg['class']} process"
              else
                "#{msg['queue']} process"
              end
            end

            def config
              Sidekiq::Instrumentation.instance.config
            end

            def tracer
              Sidekiq::Instrumentation.instance.tracer
            end
          end
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
opentelemetry-instrumentation-sidekiq-0.18.0	lib/opentelemetry/instrumentation/sidekiq/middlewares/server/tracer_middleware.rb