opentelemetry-instrumentation-ruby_kafka 0.20.1 → 0.20.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4b7a6f767f7bc01631fb7c795420b641fe40a45985a37ffe02cf6f6bbe7e186c
-  data.tar.gz: ae102bd72261240ca875f18a824bf0d35180aa65d444c6c416fe4e9f1f926fd1
+  metadata.gz: 3fc93b53a9e290e09a9682342efefccab2eeca8ec6de527eae81c2f16fca74d7
+  data.tar.gz: 60cd6ff58f39eac71c8163a2e734aa0ac3d6f51eae6de0fc855c448088f69bab
 SHA512:
-  metadata.gz: 12f07057792fad0e23b53e50259fca79aa36571b191ea2418f89d92704cb233d256bffa43c98a793c4bc2eec8d4ed1810ec7ae1deae3a66fa08d4df49d72a53c
-  data.tar.gz: f52ce0dadc1acfc3062f723379fae88dd9bde6fe2ccff6113f59b5c45831709bd5503ffb25399ce82484c976b877e8cc38819e1e8d59c5c67062762a7f9a9f91
+  metadata.gz: c8b62e4319b6970795cb314c0550e07d4d83be4a753850a8c0bbe0a1cfe7f231d2c6239558dfeb7201c0355b35b3f8bb3e663daa4d2b7501694a4e2f572b6b01
+  data.tar.gz: 7db39cffc72d33aab5fad426af7eb05f2457b1a2ede39986ac4f535dbb8e0c9b12651019892acc5fe87429ad5d577af4d21a2a463f9d76b82d72d693887bbe5d

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Release History: opentelemetry-instrumentation-ruby_kafka
 
+### v0.20.2 / 2023-08-09
+
+* FIXED: propagate context from async producer
+
 ### v0.20.1 / 2023-06-05
 
 * FIXED: Base config options 

data/lib/opentelemetry/instrumentation/ruby_kafka/instrumentation.rb

@@ -32,12 +32,14 @@ module OpenTelemetry
         end
 
         def require_patches
+          require_relative 'patches/async_producer'
           require_relative 'patches/producer'
           require_relative 'patches/consumer'
           require_relative 'patches/client'
         end
 
         def patch
+          ::Kafka::AsyncProducer.prepend(Patches::AsyncProducer)
           ::Kafka::Producer.prepend(Patches::Producer)
           ::Kafka::Consumer.prepend(Patches::Consumer)
           ::Kafka::Client.prepend(Patches::Client)

data/lib/opentelemetry/instrumentation/ruby_kafka/patches/async_producer.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+module OpenTelemetry
+  module Instrumentation
+    module RubyKafka
+      module Patches
+        # The AsyncProducer module contains the instrumentation patch the AsyncProducer#produce method
+        module AsyncProducer
+          def produce(value, topic:, **options)
+            options = __otel_merge_options!(**options)
+            super
+          end
+
+          def __otel_merge_options!(**options)
+            options ||= { headers: {} }
+            # The propagator mutates the carrier (first positional argument), so we need to set headers to empty hash so
+            # that there's something to mutate
+            options[:headers] = {} unless options[:headers]
+            OpenTelemetry.propagation.inject(options[:headers])
+            options
+          end
+        end
+      end
+    end
+  end
+end

data/lib/opentelemetry/instrumentation/ruby_kafka/patches/producer.rb

@@ -17,9 +17,15 @@ module OpenTelemetry
               'messaging.destination_kind' => 'topic'
             }
 
-            tracer.in_span("#{topic} send", attributes: attributes, kind: :producer) do
-              OpenTelemetry.propagation.inject(headers)
-              super
+            # If trace context is present in headers, extract and use it as parent. If there is _no_ trace context key
+            # in the headers, OpenTelemetry.propagation.extract will return an unmodified copy of the the current
+            # Thread's context, so the next two lines preserve the correct Thread-local context.
+            ctx = OpenTelemetry.propagation.extract(headers)
+            OpenTelemetry::Context.with_current(ctx) do
+              tracer.in_span("#{topic} send", attributes: attributes, kind: :producer) do
+                OpenTelemetry.propagation.inject(headers)
+                super
+              end
             end
           end
 

data/lib/opentelemetry/instrumentation/ruby_kafka/version.rb

@@ -7,7 +7,7 @@
 module OpenTelemetry
   module Instrumentation
     module RubyKafka
-      VERSION = '0.20.1'
+      VERSION = '0.20.2'
     end
   end
 end

data/lib/opentelemetry/instrumentation/ruby_kafka.rb

@@ -15,5 +15,5 @@ module OpenTelemetry
   end
 end
 
-require_relative './ruby_kafka/instrumentation'
-require_relative './ruby_kafka/version'
+require_relative 'ruby_kafka/instrumentation'
+require_relative 'ruby_kafka/version'

data/lib/opentelemetry/instrumentation.rb

@@ -16,4 +16,4 @@ module OpenTelemetry
   end
 end
 
-require_relative './instrumentation/ruby_kafka'
+require_relative 'instrumentation/ruby_kafka'

data/lib/opentelemetry-instrumentation-ruby_kafka.rb

@@ -4,4 +4,4 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-require_relative './opentelemetry/instrumentation'
+require_relative 'opentelemetry/instrumentation'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: opentelemetry-instrumentation-ruby_kafka
 version: !ruby/object:Gem::Version
-  version: 0.20.1
+  version: 0.20.2
 platform: ruby
 authors:
 - OpenTelemetry Authors
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-06-05 00:00:00.000000000 Z
+date: 2023-08-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: opentelemetry-api
@@ -128,14 +128,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.50.2
+        version: 1.55.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.50.2
+        version: 1.55.1
 - !ruby/object:Gem::Dependency
   name: ruby-kafka
   requirement: !ruby/object:Gem::Requirement
@@ -207,6 +207,7 @@ files:
 - lib/opentelemetry/instrumentation.rb
 - lib/opentelemetry/instrumentation/ruby_kafka.rb
 - lib/opentelemetry/instrumentation/ruby_kafka/instrumentation.rb
+- lib/opentelemetry/instrumentation/ruby_kafka/patches/async_producer.rb
 - lib/opentelemetry/instrumentation/ruby_kafka/patches/client.rb
 - lib/opentelemetry/instrumentation/ruby_kafka/patches/consumer.rb
 - lib/opentelemetry/instrumentation/ruby_kafka/patches/producer.rb
@@ -216,10 +217,10 @@ homepage: https://github.com/open-telemetry/opentelemetry-ruby-contrib
 licenses:
 - Apache-2.0
 metadata:
-  changelog_uri: https://rubydoc.info/gems/opentelemetry-instrumentation-ruby_kafka/0.20.1/file/CHANGELOG.md
+  changelog_uri: https://rubydoc.info/gems/opentelemetry-instrumentation-ruby_kafka/0.20.2/file/CHANGELOG.md
   source_code_uri: https://github.com/open-telemetry/opentelemetry-ruby-contrib/tree/main/instrumentation/ruby_kafka
   bug_tracker_uri: https://github.com/open-telemetry/opentelemetry-ruby-contrib/issues
-  documentation_uri: https://rubydoc.info/gems/opentelemetry-instrumentation-ruby_kafka/0.20.1
+  documentation_uri: https://rubydoc.info/gems/opentelemetry-instrumentation-ruby_kafka/0.20.2
 post_install_message: 
 rdoc_options: []
 require_paths: