Sha256: 61a1b3081c99b8d0b0ceb9430f8bb8eca6a919f68ad7f3fe174dcc7f7760625b

Contents?: true

Size: 1.78 KB

Versions: 2

Compression:

Stored size: 1.78 KB

Contents

# frozen_string_literal: true

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

module OpenTelemetry
  module Instrumentation
    module Sidekiq
      module Middlewares
        module Client
          # TracerMiddleware propagates context and instruments Sidekiq client
          # by way of its middleware system
          class TracerMiddleware
            def call(_worker_class, job, _queue, _redis_pool)
              tracer.in_span(
                span_name(job),
                attributes: build_attributes(job),
                kind: :producer
              ) do |span|
                OpenTelemetry.propagation.inject(job)
                span.add_event('created_at', timestamp: job['created_at'])
                yield
              end
            end

            private

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

            def span_name(job)
              if config[:enable_job_class_span_names]
                "#{job['wrapped']&.to_s || job['class']} enqueue"
              else
                "#{job['queue']} send"
              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

2 entries across 2 versions & 1 rubygems

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