Sha256: 1a3a4e6f98668bbbc4e7f5e69568f5acb3405136d7a998a84817d90d726d068e
Contents?: true
Size: 1.96 KB
Versions: 49
Compression:
Stored size: 1.96 KB
Contents
module ScoutApm
module Instruments
class Grape
attr_reader :logger
def initalize(logger=ScoutApm::Agent.instance.logger)
@logger = logger
@installed = false
end
def installed?
@installed
end
def install
@installed = true
if defined?(::Grape) && defined?(::Grape::Endpoint)
ScoutApm::Agent.instance.logger.info "Instrumenting Grape::Endpoint"
::Grape::Endpoint.class_eval do
include ScoutApm::Instruments::GrapeEndpointInstruments
alias run_without_scout_instruments run
alias run run_with_scout_instruments
end
end
end
end
module GrapeEndpointInstruments
def run_with_scout_instruments(*args)
request = ::Grape::Request.new(env || args.first)
req = ScoutApm::RequestManager.lookup
path = ScoutApm::Agent.instance.config.value("uri_reporting") == 'path' ? request.path : request.fullpath
req.annotate_request(:uri => path)
# IP Spoofing Protection can throw an exception, just move on w/o remote ip
req.context.add_user(:ip => request.ip) rescue nil
req.set_headers(request.headers)
req.web!
begin
name = ["Grape",
self.options[:method].first,
self.options[:for].to_s,
self.namespace.sub(%r{\A/}, ''), # removing leading slashes
self.options[:path].first,
].compact.map{ |n| n.to_s }.join("/")
rescue => e
ScoutApm::Agent.instance.logger.info("Error getting Grape Endpoint Name. Error: #{e.message}. Options: #{self.options.inspect}")
name = "Grape/Unknown"
end
req.start_layer( ScoutApm::Layer.new("Controller", name) )
begin
run_without_scout_instruments(*args)
rescue
req.error!
raise
ensure
req.stop_layer
end
end
end
end
end
Version data entries
49 entries across 49 versions & 1 rubygems