Created
April 10, 2013 05:12
-
-
Save zanker/5351986 to your computer and use it in GitHub Desktop.
Modification of the original Mongo instrumentation by Alexey Palazhchenko. This handles obfuscating values itself which means Newrelics obfuscator should be set to raw to prevent it breaking anything. This will also make Mongo show up under the overview again.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
DependencyDetection.defer do | |
@name = :mongodb | |
depends_on do | |
defined?(::Mongo) and not NewRelic::Control.instance['disable_mongodb'] | |
end | |
executes do | |
NewRelic::Agent.logger.debug 'Installing MongoDB instrumentation' | |
end | |
executes do | |
::Mongo::Logging.class_eval do | |
include NewRelic::Agent::MethodTracer | |
def sanitize_data(data) | |
parsed = {} | |
data.each do |key, value| | |
if value.is_a?(Array) | |
if key == "$or" or key == "$and" or key == "$and" | |
parsed[key] = [] | |
value.each_index do |i| | |
parsed[key][i] = sanitize_data(value[i]) | |
end | |
else | |
parsed[key] = "?" | |
end | |
elsif value.is_a?(Hash) | |
parsed[key] = sanitize_data(value) | |
else | |
parsed[key] = "?" | |
end | |
end | |
parsed | |
end | |
# This is an inefficient way of instrumenting and logging Mongo right now | |
# but we can't really easily do this with regex because of how Mongo queries work | |
def instrument_with_newrelic_trace(name, payload = {}, &blk) | |
collection = payload[:collection] | |
if collection == '$cmd' | |
f = payload[:selector].first | |
name, collection = f if f | |
end | |
log_payload = {:collection => collection} | |
if payload[:query] | |
log_payload[:query] = sanitize_data(payload[:query]) | |
elsif payload[:documents] | |
log_payload[:documents] = [] | |
payload[:documents].each_index do |i| | |
log_payload[:documents][i] = sanitize_data(payload[:documents][i]) | |
end | |
end | |
if payload[:selector] and payload[:selector][:query] | |
log_payload[:selector] = payload[:selector].dup | |
log_payload[:selector][:query] = sanitize_data(payload[:selector][:query]) | |
elsif payload[:selector] and name != "mapreduce" | |
log_payload[:selector] = sanitize_data(payload[:selector]) | |
end | |
log_payload = log_payload.inspect | |
log_payload.gsub!('"?"', "?") | |
trace_execution_scoped("Database/#{collection}/#{name}") do | |
t0 = Time.now | |
begin | |
res = instrument_without_newrelic_trace(name, payload, &blk) | |
ensure | |
elapsed = (Time.now - t0).to_f | |
NewRelic::Agent.instance.transaction_sampler.notice_sql(log_payload, nil, elapsed) | |
NewRelic::Agent.instance.sql_sampler.notice_sql(log_payload, nil, nil, elapsed) | |
["ActiveRecord/all", "ActiveRecord/#{name}"].each do |metric| | |
NewRelic::Agent.instance.stats_engine.get_stats_no_scope(metric).trace_call(elapsed) | |
end | |
end | |
res | |
end | |
end | |
alias_method :instrument_without_newrelic_trace, :instrument | |
alias_method :instrument, :instrument_with_newrelic_trace | |
end | |
class Mongo::Collection; include Mongo::Logging; end | |
class Mongo::Connection; include Mongo::Logging; end | |
class Mongo::Cursor; include Mongo::Logging; end | |
end | |
end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment