From 978b3c650cd6041fec6b7444b3230c800d87b9a0 Mon Sep 17 00:00:00 2001 From: Tanna McClure Date: Wed, 11 Sep 2024 16:39:43 -0500 Subject: [PATCH] fix for rdkafka on load tests --- lib/new_relic/agent/instrumentation/rdkafka/chain.rb | 2 ++ lib/new_relic/agent/instrumentation/rdkafka/prepend.rb | 4 +++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/lib/new_relic/agent/instrumentation/rdkafka/chain.rb b/lib/new_relic/agent/instrumentation/rdkafka/chain.rb index 5b0617fb86..fc9b826aa8 100644 --- 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 index b9fe845e1d..3397c2436c 100644 --- 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)