lib/new_relic/agent/instrumentation/bunny/instrumentation.rb in newrelic_rpm-8.13.1 vs lib/new_relic/agent/instrumentation/bunny/instrumentation.rb in newrelic_rpm-8.14.0

@@ -51,11 +51,11 @@
             else
               NewRelic::Agent::Tracer.capture_segment_error(segment) do
                 yield
               end
             ensure
-              segment.finish if segment
+              ::NewRelic::Agent::Transaction::Segment.finish(segment)
             end
           end
         end
 
         module Queue
@@ -95,11 +95,11 @@
               if bunny_error
                 segment.notice_error(bunny_error)
                 raise bunny_error
               end
             ensure
-              segment.finish if segment
+              ::NewRelic::Agent::Transaction::Segment.finish(segment)
             end
 
             msg
           end
 
@@ -118,10 +118,10 @@
             else
               NewRelic::Agent::Tracer.capture_segment_error(segment) do
                 yield
               end
             ensure
-              segment.finish if segment
+              ::NewRelic::Agent::Transaction::Segment.finish(segment)
             end
           end
         end
 
         module Consumer