Fix ruby-kafka instrumentation bug

Author: ericfirth

lib/datadog/data_streams.rb

@@ -25,16 +25,13 @@ class << self
       # @return [String, nil] Base64 encoded pathway context or nil if disabled
       # @public_api
       def set_produce_checkpoint(type:, destination:, auto_instrumentation: false, tags: {}, &block)
-        Datadog.logger.debug("[DSM PUBLIC API] set_produce_checkpoint called: type=#{type}, dest=#{destination}, processor=#{processor.class}")
-        result = processor&.set_produce_checkpoint(
+        processor&.set_produce_checkpoint(
           type: type,
           destination: destination,
           manual_checkpoint: !auto_instrumentation,
           tags: tags,
           &block
         )
-        Datadog.logger.debug("[DSM PUBLIC API] set_produce_checkpoint returned: #{result.inspect}")
-        result
       end
 
       # Set a consume checkpoint for Data Streams Monitoring

lib/datadog/data_streams/processor.rb

@@ -100,15 +100,13 @@ def track_kafka_consume(topic, partition, offset, now)
       # @yield [key, value] Block to inject context into carrier
       # @return [String] Base64 encoded pathway context
       def set_produce_checkpoint(type:, destination:, manual_checkpoint: true, tags: {}, &block)
-        @logger.debug("[DSM PROCESSOR] set_produce_checkpoint called: type=#{type}, dest=#{destination}")
         checkpoint_tags = ["type:#{type}", "topic:#{destination}", 'direction:out']
         checkpoint_tags << 'manual_checkpoint:true' if manual_checkpoint
         checkpoint_tags.concat(tags.map { |k, v| "#{k}:#{v}" }) unless tags.empty?
 
         span = Datadog::Tracing.active_span
         pathway = set_checkpoint(tags: checkpoint_tags, span: span)
 
-        @logger.debug("[DSM PROCESSOR] set_produce_checkpoint pathway=#{pathway.inspect}")
         yield(PROPAGATION_KEY, pathway) if pathway && block
 
         pathway

lib/datadog/tracing/contrib/kafka/instrumentation/consumer.rb

@@ -7,7 +7,7 @@ module Kafka
         module Instrumentation
           # Instrumentation for Kafka::Consumer
           module Consumer
-            def self.included(base)
+            def self.prepended(base)
               base.prepend(InstanceMethods)
             end
 

lib/datadog/tracing/contrib/kafka/instrumentation/producer.rb

@@ -7,7 +7,7 @@ module Kafka
         module Instrumentation
           # Instrumentation for Kafka::Producer
           module Producer
-            def self.included(base)
+            def self.prepended(base)
               base.prepend(InstanceMethods)
             end