Module: NewRelic::Agent::Instrumentation::Rake
- Defined in:
- lib/new_relic/agent/instrumentation/rake/chain.rb,
lib/new_relic/agent/instrumentation/rake/prepend.rb,
lib/new_relic/agent/instrumentation/rake/instrumentation.rb
Defined Under Namespace
Modules: Chain, Prepend, Tracer
Class Method Summary
collapse
Class Method Details
.before_invoke_transaction(task) ⇒ Object
82
83
84
85
86
87
88
89
90
91
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 82
def before_invoke_transaction(task)
ensure_at_exit
instrument_execute_on_prereqs(task)
if task.application.options.always_multitask
instrument_invoke_prerequisites_concurrently(task)
end
rescue => e
NewRelic::Agent.logger.error('Error during Rake task invoke', e)
end
|
.ensure_at_exit ⇒ Object
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 123
def ensure_at_exit
return if @installed_at_exit
at_exit do
NewRelic::Agent.shutdown
end
@installed_at_exit = true
end
|
.instrument_execute(task) ⇒ Object
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 57
def instrument_execute(task)
return if task.instance_variable_get(:@__newrelic_instrumented_execute)
task.instance_variable_set(:@__newrelic_instrumented_execute, true)
task.instance_eval do
def execute(*args, &block)
NewRelic::Agent::MethodTracer.trace_execution_scoped("Rake/execute/#{self.name}") do
super
end
end
end
instrument_execute_on_prereqs(task)
end
|
.instrument_execute_on_prereqs(task) ⇒ Object
51
52
53
54
55
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 51
def instrument_execute_on_prereqs(task)
task.prerequisite_tasks.each do |child_task|
instrument_execute(child_task)
end
end
|
.instrument_invoke_prerequisites_concurrently(task) ⇒ Object
72
73
74
75
76
77
78
79
80
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 72
def instrument_invoke_prerequisites_concurrently(task)
task.instance_eval do
def invoke_prerequisites_concurrently(*_)
NewRelic::Agent::MethodTracer.trace_execution_scoped('Rake/execute/multitask') do
super
end
end
end
end
|
.name_the_args(args, names) ⇒ Object
Expects literal args passed to the task and array of task names If names are present without matching args, still sets them with nils
110
111
112
113
114
115
116
117
118
119
120
121
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 110
def name_the_args(args, names)
unfulfilled_names_length = names.length - args.length
if unfulfilled_names_length > 0
args.concat(Array.new(unfulfilled_names_length))
end
result = {}
args.zip(names).each_with_index do |(value, key), index|
result[key || index.to_s] = value
end
result
end
|
.record_attributes(args, task) ⇒ Object
.safe_from_third_party_gem? ⇒ Boolean
38
39
40
41
42
43
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 38
def safe_from_third_party_gem?
return true unless NewRelic::LanguageSupport.bundled_gem?('newrelic-rake')
::NewRelic::Agent.logger.info('Not installing New Relic supported Rake instrumentation because the third party newrelic-rake gem is present')
false
end
|
.should_install? ⇒ Boolean
34
35
36
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 34
def should_install?
safe_from_third_party_gem?
end
|
.should_trace?(name) ⇒ Boolean
45
46
47
48
49
|
# File 'lib/new_relic/agent/instrumentation/rake/instrumentation.rb', line 45
def should_trace?(name)
NewRelic::Agent.config[:'rake.tasks'].any? do |regex|
regex.match(name)
end
end
|