lib/ting_yun/instrumentation/rake.rb in tingyun_rpm-1.2.0 vs lib/ting_yun/instrumentation/rake.rb in tingyun_rpm-1.3.0

@@ -11,10 +11,11 @@
         ::TingYun::Agent::Instrumentation::RakeInstrumentation.supported_version?
   end
 
   executes do
     ::TingYun::Agent.logger.info 'Installing deferred Rake instrumentation'
+    require 'ting_yun/agent/method_tracer_helpers'
   end
 
   executes do
     module Rake
       class Task
@@ -85,11 +86,11 @@
           return if task.instance_variable_get(:@__tingyun_instrumented_execute)
 
           task.instance_variable_set(:@__tingyun_instrumented_execute, true)
           task.instance_eval do
             def execute(*args, &block)
-              TingYun::Agent::MethodTracer.trace_execution_scoped("Rake/execute/#{self.name}") do
+              TingYun::Agent::MethodTracerHelpers.trace_execution_scoped("Rake/execute/#{self.name}") do
                 super
               end
             end
           end
 
@@ -97,10 +98,10 @@
         end
 
         def self.instrument_invoke_prerequisites_concurrently(task)
           task.instance_eval do
             def invoke_prerequisites_concurrently(*_)
-              TingYun::Agent::MethodTracer.trace_execution_scoped("Rake/execute/multitask") do
+              TingYun::Agent::MethodTracerHelpers.trace_execution_scoped("Rake/execute/multitask") do
                 super
               end
             end
           end
         end
\ No newline at end of file