Merge pull request #2849 from newrelic/fix_load_tests

fix for rdkafka on load tests
newrelic · Sep 11, 2024 · b3f45e4 · b3f45e4
2 parents d121224 + 978b3c6
commit b3f45e4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/lib/new_relic/agent/instrumentation/rdkafka/chain.rb b/lib/new_relic/agent/instrumentation/rdkafka/chain.rb
@@ -2,6 +2,8 @@
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
 # frozen_string_literal: true
 
+require_relative 'instrumentation'
+
 module NewRelic::Agent::Instrumentation
   module Rdkafka::Chain
     def self.instrument!

diff --git a/lib/new_relic/agent/instrumentation/rdkafka/prepend.rb b/lib/new_relic/agent/instrumentation/rdkafka/prepend.rb
@@ -2,6 +2,8 @@
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
 # frozen_string_literal: true
 
+require_relative 'instrumentation'
+
 module NewRelic::Agent::Instrumentation
   module RdkafkaProducer
     module Prepend
@@ -34,7 +36,7 @@ module RdkafkaConfig
     module Prepend
       include NewRelic::Agent::Instrumentation::RdkafkaConfig
 
-      if Gem::Version.new(::Rdkafka::VERSION) >= Gem::Version.new('0.16.0')
+      if defined?(::Rdkafka) && Gem::Version.new(::Rdkafka::VERSION) >= Gem::Version.new('0.16.0')
         def producer(**kwargs)
           super.tap do |producer|
             set_nr_config(producer)