lib/datadog/tracing/contrib/kafka/events.rb


# frozen_string_literal: true

require_relative 'events/connection/request'
require_relative 'events/consumer/process_batch'
require_relative 'events/consumer/process_message'
require_relative 'events/consumer_group/heartbeat'
require_relative 'events/consumer_group/join_group'
require_relative 'events/consumer_group/leave_group'
require_relative 'events/consumer_group/sync_group'
require_relative 'events/produce_operation/send_messages'
require_relative 'events/producer/deliver_messages'

module Datadog
  module Tracing
    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
end