Module: Datadog::Tracing::Contrib::Karafka::MessagesPatch

Defined in:
lib/datadog/tracing/contrib/karafka/patcher.rb

Overview

Patch to add tracing to Karafka::Messages::Messages

Instance Method Summary collapse

Instance Method Details

#configuration ⇒ Object 



13
14
15
 
# File 'lib/datadog/tracing/contrib/karafka/patcher.rb', line 13

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

#each(&block) ⇒ Object

each is the most popular access point to Karafka messages, but not the only one Other access patterns do not have a straightforward tracing avenue (e.g. my_batch_operation messages.payloads)

See Also:



26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
 
# File 'lib/datadog/tracing/contrib/karafka/patcher.rb', line 26

def each(&block)
  @messages_array.each do |message|
    if configuration[:distributed_tracing]
      headers = if message.metadata.respond_to?(:raw_headers)
        message.metadata.raw_headers
      else
        message.metadata.headers
      end
      trace_digest = Karafka.extract(headers)
      Datadog::Tracing.continue_trace!(trace_digest) if trace_digest
    end

    if Datadog::DataStreams.enabled?
      begin
        headers = if message.metadata.respond_to?(:raw_headers)
          message.metadata.raw_headers
        else
          message.metadata.headers
        end

        Datadog::DataStreams.set_consume_checkpoint(
          type: 'kafka',
          source: message.topic,
          auto_instrumentation: true
        ) { |key| headers[key] }
      rescue => e
        Datadog.logger.debug("Error setting DSM checkpoint: #{e.class}: #{e}")
      end
    end

    Tracing.trace(Ext::SPAN_MESSAGE_CONSUME) do |span|
      span.set_tag(Ext::TAG_OFFSET, message.metadata.offset)
      span.set_tag(Contrib::Ext::Messaging::TAG_DESTINATION, message.topic)
      span.set_tag(Contrib::Ext::Messaging::TAG_SYSTEM, Ext::TAG_SYSTEM)

      span.resource = message.topic

      yield message
    end
  end
end

#propagation ⇒ Object 



17
18
19
 
# File 'lib/datadog/tracing/contrib/karafka/patcher.rb', line 17

def propagation
  @propagation ||= Contrib::Karafka::Distributed::Propagation.new
end