newrelic_rpm 8.9.0 → 8.10.1

data/lib/new_relic/agent/instrumentation/curb/chain.rb

@@ -16,46 +16,46 @@ module NewRelic::Agent::Instrumentation
             http_head_with_tracing { http_head_without_newrelic(*args, &blk) }
           end
 
-          alias_method :http_head_without_newrelic, :http_head
-          alias_method :http_head, :http_head_with_newrelic
+          alias_method(:http_head_without_newrelic, :http_head)
+          alias_method(:http_head, :http_head_with_newrelic)
 
           def http_post_with_newrelic(*args, &blk)
             http_post_with_tracing { http_post_without_newrelic(*args, &blk) }
           end
 
-          alias_method :http_post_without_newrelic, :http_post
-          alias_method :http_post, :http_post_with_newrelic
+          alias_method(:http_post_without_newrelic, :http_post)
+          alias_method(:http_post, :http_post_with_newrelic)
 
           def http_put_with_newrelic(*args, &blk)
             http_put_with_tracing { http_put_without_newrelic(*args, &blk) }
           end
 
-          alias_method :http_put_without_newrelic, :http_put
-          alias_method :http_put, :http_put_with_newrelic
+          alias_method(:http_put_without_newrelic, :http_put)
+          alias_method(:http_put, :http_put_with_newrelic)
 
           # Hook the #http method to set the verb.
-          def http_with_newrelic verb
+          def http_with_newrelic(verb)
             http_with_tracing(verb) { http_without_newrelic(verb) }
           end
 
-          alias_method :http_without_newrelic, :http
-          alias_method :http, :http_with_newrelic
+          alias_method(:http_without_newrelic, :http)
+          alias_method(:http, :http_with_newrelic)
 
           # Hook the #perform method to mark the request as non-parallel.
           def perform_with_newrelic
             perform_with_tracing { perform_without_newrelic }
           end
 
-          alias_method :perform_without_newrelic, :perform
-          alias_method :perform, :perform_with_newrelic
+          alias_method(:perform_without_newrelic, :perform)
+          alias_method(:perform, :perform_with_newrelic)
 
           # Record the HTTP verb for future #perform calls
-          def method_with_newrelic verb
+          def method_with_newrelic(verb)
             method_with_tracing(verb) { method_without_newrelic(verb) }
           end
 
-          alias_method :method_without_newrelic, :method
-          alias_method :method, :method_with_newrelic
+          alias_method(:method_without_newrelic, :method)
+          alias_method(:method, :method_with_newrelic)
 
           # We override this method in order to ensure access to header_str even
           # though we use an on_header callback
@@ -63,8 +63,8 @@ module NewRelic::Agent::Instrumentation
             header_str_with_tracing { header_str_without_newrelic }
           end
 
-          alias_method :header_str_without_newrelic, :header_str
-          alias_method :header_str, :header_str_with_newrelic
+          alias_method(:header_str_without_newrelic, :header_str)
+          alias_method(:header_str, :header_str_with_newrelic)
         end
 
         Curl::Multi.class_eval do
@@ -72,19 +72,19 @@ module NewRelic::Agent::Instrumentation
 
           # Add CAT with callbacks if the request is serial
           def add_with_newrelic(curl)
-            add_with_tracing(curl) { add_without_newrelic curl }
+            add_with_tracing(curl) { add_without_newrelic(curl) }
           end
 
-          alias_method :add_without_newrelic, :add
-          alias_method :add, :add_with_newrelic
+          alias_method(:add_without_newrelic, :add)
+          alias_method(:add, :add_with_newrelic)
 
           # Trace as an External/Multiple call if the first request isn't serial.
           def perform_with_newrelic(&blk)
             perform_with_tracing { perform_without_newrelic(&blk) }
           end
 
-          alias_method :perform_without_newrelic, :perform
-          alias_method :perform, :perform_with_newrelic
+          alias_method(:perform_without_newrelic, :perform)
+          alias_method(:perform, :perform_with_newrelic)
         end
       end
     end

data/lib/new_relic/agent/instrumentation/curb/instrumentation.rb

@@ -35,7 +35,7 @@ module NewRelic
           end
 
           # Hook the #http method to set the verb.
-          def http_with_tracing verb
+          def http_with_tracing(verb)
             self._nr_http_verb = verb.to_s.upcase
             yield
           end
@@ -48,7 +48,7 @@ module NewRelic
           end
 
           # Record the HTTP verb for future #perform calls
-          def method_with_tracing verb
+          def method_with_tracing(verb)
             self._nr_http_verb = verb.upcase
             yield
           end
@@ -99,7 +99,7 @@ module NewRelic
               procedure: wrapped_request.method
             )
 
-            segment.add_request_headers wrapped_request
+            segment.add_request_headers(wrapped_request)
 
             # install all callbacks
             unless request._nr_instrumented
@@ -129,9 +129,9 @@ module NewRelic
             request._nr_header_str = nil
             request.on_header do |header_data|
               if original_callback
-                original_callback.call header_data
+                original_callback.call(header_data)
               else
-                wrapped_response.append_header_data header_data
+                wrapped_response.append_header_data(header_data)
                 header_data.length
               end
             end
@@ -143,7 +143,7 @@ module NewRelic
             request._nr_original_on_complete = original_callback
             request.on_complete do |finished_request|
               begin
-                segment.process_response_headers wrapped_response
+                segment.process_response_headers(wrapped_response)
               ensure
                 segment.finish if segment
                 # Make sure the existing completion callback is run, and restore the
@@ -167,8 +167,8 @@ module NewRelic
             request.on_failure do |failed_request, error|
               begin
                 if segment
-                  noticible_error = NewRelic::Agent::NoticibleError.new error[0].name, error[-1]
-                  segment.notice_error noticible_error
+                  noticible_error = NewRelic::Agent::NoticibleError.new(error[0].name, error[-1])
+                  segment.notice_error(noticible_error)
                 end
               ensure
                 original_callback.call(failed_request, error) if original_callback

data/lib/new_relic/agent/instrumentation/curb/prepend.rb

@@ -25,7 +25,7 @@ module NewRelic
               http_put_with_tracing { super }
             end
 
-            def http verb
+            def http(verb)
               http_with_tracing(verb) { super }
             end
 
@@ -33,7 +33,7 @@ module NewRelic
               perform_with_tracing { super }
             end
 
-            def method verb
+            def method(verb)
               method_with_tracing(verb) { super }
             end
 

data/lib/new_relic/agent/instrumentation/curb.rb

@@ -1,6 +1,7 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 require_relative 'curb/chain'
 require_relative 'curb/prepend'
@@ -16,7 +17,7 @@ DependencyDetection.defer do
   end
 
   executes do
-    ::NewRelic::Agent.logger.info 'Installing Curb instrumentation'
+    ::NewRelic::Agent.logger.info('Installing Curb instrumentation')
     require 'new_relic/agent/distributed_tracing/cross_app_tracing'
     require 'new_relic/agent/http_clients/curb_wrappers'
   end

data/lib/new_relic/agent/instrumentation/data_mapper.rb

@@ -1,6 +1,7 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 DependencyDetection.defer do
   named :data_mapper
@@ -13,7 +14,7 @@ DependencyDetection.defer do
   end
 
   executes do
-    ::NewRelic::Agent.logger.info 'Installing DataMapper instrumentation'
+    ::NewRelic::Agent.logger.info('Installing DataMapper instrumentation')
     require 'new_relic/agent/datastores/metric_helper'
 
     deprecation_msg = 'The instrumentation for DataMapper is deprecated. ' \
@@ -30,69 +31,69 @@ DependencyDetection.defer do
   end
 
   executes do
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :get
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :first
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :last
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :all
-
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :create
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :create!
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :update
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :update!
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :destroy
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :destroy!
-
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :aggregate
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :find
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Model, :find_by_sql
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :get)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :first)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :last)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :all)
+
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :create)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :create!)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :update)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :update!)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :destroy)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :destroy!)
+
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :aggregate)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :find)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Model, :find_by_sql)
   end
 
   executes do
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Resource, :update
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Resource, :update!
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Resource, :save
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Resource, :save!
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Resource, :destroy
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Resource, :destroy!
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Resource, :update)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Resource, :update!)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Resource, :save)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Resource, :save!)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Resource, :destroy)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Resource, :destroy!)
   end
 
   executes do
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :get
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :first
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :last
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :all
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :get)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :first)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :last)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :all)
 
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :lazy_load
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :lazy_load)
 
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :create
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :create!
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :update
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :update!
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :destroy
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :destroy!
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :create)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :create!)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :update)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :update!)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :destroy)
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :destroy!)
 
-    NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Collection, :aggregate
+    NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Collection, :aggregate)
   end
 
   executes do
     # Catch direct SQL calls that bypass CRUD
     if defined?(::DataMapper::Adapters::DataObjectsAdapter)
-      NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Adapters::DataObjectsAdapter, :select, true
-      NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Adapters::DataObjectsAdapter, :execute, true
+      NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Adapters::DataObjectsAdapter, :select, true)
+      NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Adapters::DataObjectsAdapter, :execute, true)
     end
   end
 
   executes do
     # DM::Validations overrides Model#create, so we patch it here as well
     if defined?(::DataMapper::Validations::ClassMethods)
-      NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Validations::ClassMethods, :create
+      NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Validations::ClassMethods, :create)
     end
   end
 
   executes do
     # DM::Transaction calls commit() twice, so potentially shows up twice.
     if defined?(::DataMapper::Transaction)
-      NewRelic::Agent::DataMapperTracing.add_tracer ::DataMapper::Transaction, :commit, true
+      NewRelic::Agent::DataMapperTracing.add_tracer(::DataMapper::Transaction, :commit, true)
     end
   end
 
@@ -114,8 +115,8 @@ module NewRelic
             define_method("#{method_name}_with_newrelic",
               NewRelic::Agent::DataMapperTracing.method_body(clazz, method_name, operation_only))
 
-            alias_method "#{method_name}_without_newrelic", method_name
-            alias_method method_name, "#{method_name}_with_newrelic"
+            alias_method("#{method_name}_without_newrelic", method_name)
+            alias_method(method_name, "#{method_name}_with_newrelic")
           end
         end
       end
@@ -152,7 +153,7 @@ module NewRelic
             )
 
             begin
-              NewRelic::Agent::Tracer.capture_segment_error segment do
+              NewRelic::Agent::Tracer.capture_segment_error(segment) do
                 self.send("#{method_name}_without_newrelic", *args, &blk)
               end
             rescue ::DataObjects::ConnectionError => e
@@ -205,7 +206,7 @@ module NewRelic
           txn = state.current_transaction
 
           if txn && txn.current_segment.respond_to?(:notice_sql)
-            txn.current_segment.notice_sql msg.query
+            txn.current_segment.notice_sql(msg.query)
           end
         ensure
           super

data/lib/new_relic/agent/instrumentation/delayed_job/chain.rb

@@ -22,7 +22,7 @@ module NewRelic::Agent::Instrumentation
             Delayed::Job.class_eval do
               include NewRelic::Agent::Instrumentation::DelayedJobTracer
 
-              alias_method :invoke_job_without_new_relic, :invoke_job
+              alias_method(:invoke_job_without_new_relic, :invoke_job)
 
               def invoke_job(*args, &block)
                 invoke_job_with_tracing { invoke_job_without_new_relic(*args, &block) }

data/lib/new_relic/agent/instrumentation/delayed_job/instrumentation.rb

@@ -19,9 +19,9 @@ module NewRelic
           if defined?(::Delayed::Job) && ::Delayed::Job.method_defined?(:invoke_job) &&
               !(::Delayed::Job.method_defined?(:invoke_job_without_new_relic))
 
-            ::NewRelic::Agent.logger.info 'Installing DelayedJob instrumentation [part 2/2]'
+            ::NewRelic::Agent.logger.info('Installing DelayedJob instrumentation [part 2/2]')
             install_newrelic_job_tracer
-            NewRelic::Control.instance.init_plugin :dispatcher => :delayed_job
+            NewRelic::Control.instance.init_plugin(:dispatcher => :delayed_job)
           else
             NewRelic::Agent.logger.warn("Did not find a Delayed::Job class responding to invoke_job, aborting DJ instrumentation")
           end

data/lib/new_relic/agent/instrumentation/delayed_job/prepend.rb

@@ -17,7 +17,7 @@ module NewRelic
           end
 
           def install_newrelic_job_tracer
-            Delayed::Job.send :prepend, ::NewRelic::Agent::Instrumentation::DelayedJobTracerPrepend
+            Delayed::Job.send(:prepend, ::NewRelic::Agent::Instrumentation::DelayedJobTracerPrepend)
           end
         end
       end

data/lib/new_relic/agent/instrumentation/delayed_job_instrumentation.rb

@@ -1,6 +1,7 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 require_relative 'delayed_job/chain'
 require_relative 'delayed_job/prepend'
@@ -21,7 +22,7 @@ module NewRelic
           LEGACY_DJ_DEFAULT_CLASS = '(unknown class)'.freeze
 
           def name_from_payload(payload_object)
-            if payload_object.is_a? ::Delayed::PerformableMethod
+            if payload_object.is_a?(::Delayed::PerformableMethod)
               # payload_object contains a reference to an object
               # that received an asynchronous method call via .delay or .handle_asynchronously
               "#{object_name(payload_object)}#{delimiter(payload_object)}#{method_name(payload_object)}"
@@ -83,7 +84,7 @@ DependencyDetection.defer do
   end
 
   executes do
-    ::NewRelic::Agent.logger.info 'Installing DelayedJob instrumentation [part 1/2]'
+    ::NewRelic::Agent.logger.info('Installing DelayedJob instrumentation [part 1/2]')
   end
 
   executes do

data/lib/new_relic/agent/instrumentation/excon/middleware.rb

@@ -1,6 +1,7 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 module ::Excon
   module Middleware
@@ -25,7 +26,7 @@ module ::Excon
               procedure: wrapped_request.method
             )
 
-            segment.add_request_headers wrapped_request
+            segment.add_request_headers(wrapped_request)
 
             datum[:connection].instance_variable_set(TRACE_DATA_IVAR, segment)
           end
@@ -54,7 +55,7 @@ module ::Excon
 
             if datum[:response]
               wrapped_response = ::NewRelic::Agent::HTTPClients::ExconHTTPResponse.new(datum[:response])
-              segment.process_response_headers wrapped_response
+              segment.process_response_headers(wrapped_response)
             end
           ensure
             segment.finish if segment

data/lib/new_relic/agent/instrumentation/excon.rb

@@ -1,6 +1,7 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 DependencyDetection.defer do
   named :excon
@@ -40,7 +41,7 @@ DependencyDetection.defer do
   end
 
   def install_middleware_excon_instrumentation
-    ::NewRelic::Agent.logger.info 'Installing middleware-based Excon instrumentation'
+    ::NewRelic::Agent.logger.info('Installing middleware-based Excon instrumentation')
     require 'new_relic/agent/instrumentation/excon/middleware'
     defaults = Excon.defaults
 

data/lib/new_relic/agent/instrumentation/grape/chain.rb

@@ -1,22 +1,23 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 module NewRelic::Agent::Instrumentation
   module Grape
     module Chain
       def self.instrument!
         Grape::Instrumentation.instrumented_class.class_eval do
-          def call_with_new_relic env
+          def call_with_new_relic(env)
             begin
-              call_without_new_relic env
+              call_without_new_relic(env)
             ensure
-              Grape::Instrumentation.capture_transaction env, self
+              Grape::Instrumentation.capture_transaction(env, self)
             end
           end
 
-          alias_method :call_without_new_relic, :call
-          alias_method :call, :call_with_new_relic
+          alias_method(:call_without_new_relic, :call)
+          alias_method(:call, :call_with_new_relic)
         end
       end
     end

data/lib/new_relic/agent/instrumentation/grape/instrumentation.rb

@@ -1,6 +1,7 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 require 'new_relic/agent/parameter_filtering'
 
@@ -15,7 +16,7 @@ module NewRelic::Agent::Instrumentation
         defined?(::Grape::API::Instance) ? ::Grape::API::Instance : ::Grape::API
       end
 
-      def capture_transaction env, context
+      def capture_transaction(env, context)
         begin
           endpoint = env[API_ENDPOINT]
           version = env[API_VERSION]
@@ -29,10 +30,10 @@ module NewRelic::Agent::Instrumentation
 
       def prepare!
         if defined?(::Grape::VERSION) && Gem::Version.new(::Grape::VERSION) >= Gem::Version.new("0.16.0")
-          send :remove_method, :name_for_transaction_deprecated
+          send(:remove_method, :name_for_transaction_deprecated)
         else
-          send :remove_method, :name_for_transaction
-          send :alias_method, :name_for_transaction, :name_for_transaction_deprecated
+          send(:remove_method, :name_for_transaction)
+          send(:alias_method, :name_for_transaction, :name_for_transaction_deprecated)
         end
       end
 

data/lib/new_relic/agent/instrumentation/grape/prepend.rb

@@ -1,15 +1,16 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 module NewRelic::Agent::Instrumentation
   module Grape
     module Prepend
-      def call env
+      def call(env)
         begin
-          super env
+          super(env)
         ensure
-          Grape::Instrumentation.capture_transaction env, self
+          Grape::Instrumentation.capture_transaction(env, self)
         end
       end
     end

data/lib/new_relic/agent/instrumentation/grape.rb

@@ -1,6 +1,7 @@
 # encoding: utf-8
 # This file is distributed under New Relic's license terms.
 # See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+# frozen_string_literal: true
 
 require_relative 'grape/instrumentation'
 require_relative 'grape/chain'

data/lib/new_relic/agent/instrumentation/grpc/client/chain.rb

@@ -0,0 +1,98 @@
+# encoding: utf-8
+# frozen_string_literal: true
+# This file is distributed under New Relic's license terms.
+# See https://github.com/newrelic/newrelic-ruby-agent/blob/main/LICENSE for complete details.
+
+require_relative 'instrumentation'
+
+module NewRelic::Agent::Instrumentation
+  module GRPC
+    module Client
+      module Chain
+        def self.instrument!
+          ::GRPC::ClientStub.class_eval do
+            include NewRelic::Agent::Instrumentation::GRPC::Client
+
+            def bidi_streamer_with_new_relic_trace(method, requests, marshal, unmarshal,
+              deadline: nil,
+              return_op: false,
+              parent: nil,
+              credentials: nil,
+              metadata: {},
+              &blk)
+
+              issue_request_with_tracing(:bidi_streamer, method, requests, marshal, unmarshal,
+                deadline: deadline,
+                return_op: return_op,
+                parent: parent,
+                credentials: credentials,
+                metadata: metadata)
+
+              # TODO: gRPC - confirm that &blk is being invoked correctly
+            end
+
+            alias bidi_streamer_without_newrelic_trace bidi_streamer
+            alias bidi_streamer bidi_streamer_with_newrelic_trace
+
+            def client_streamer_with_newrelic_trace(method, requests, marshal, unmarshal,
+              deadline: nil,
+              return_op: false,
+              parent: nil,
+              credentials: nil,
+              metadata: {})
+
+              issue_request_with_tracing(:client_streamer, method, requests, marshal, unmarshal,
+                deadline: deadline,
+                return_op: return_op,
+                parent: parent,
+                credentials: credentials,
+                metadata: metadata)
+            end
+
+            alias client_streamer_without_newrelic_trace client_streamer
+            alias client_streamer client_streamer_with_newrelic_trace
+
+            def request_response_with_newrelic_trace(method, req, marshal, unmarshal,
+              deadline: nil,
+              return_op: false,
+              parent: nil,
+              credentials: nil,
+              metadata: {})
+
+              issue_request_with_tracing(:request_response, method, req, marshal, unmarshal,
+                deadline: deadline,
+                return_op: return_op,
+                parent: parent,
+                credentials: credentials,
+                metadata: metadata)
+            end
+
+            alias request_response_without_newrelic_trace request_response
+            alias request_response request_response_with_newrelic_trace
+
+            def server_streamer_with_newrelic_trace(method, req, marshal, unmarshal,
+              deadline: nil,
+              return_op: false,
+              parent: nil,
+              credentials: nil,
+              metadata: {},
+              &blk)
+
+              issue_request_with_tracing(:server_streamer, method, req, marshal, unmarshal,
+                deadline: deadline,
+                return_op: return_op,
+                parent: parent,
+                credentials: credentials,
+                metadata: metadata)
+
+              # TODO: gRPC - confirm that &blk is being invoked correctly
+            end
+
+            alias server_streamer_without_newrelic_trace server_streamer
+            alias server_streamer server_streamer_with_newrelic_trace
+          end
+        end
+      end
+    end
+  end
+end