Software Público Brasileiro

class ProposalsDiscussionPlugin::ApiNewRelicInstrumenter < Grape::Middleware::Base
  include NewRelic::Agent::Instrumentation::ControllerInstrumentation

  def call_with_newrelic(&block)
    trace_options = {
      :category => :rack,
      :path => "#{route_path}\##{route_method}",
      :request => Grape::Request.new(@env)
    }

    perform_action_with_newrelic_trace(trace_options) do
      result = yield
      MetricFrame.abort_transaction! if result.first == 404 # ignore cascaded calls
      result
    end
  end

  def call(env)
    @env = env
    if ENV['NEW_RELIC_ID']
      call_with_newrelic do
        super
      end
    else
      super
    end
  end

  def env
    @env
  end

  def route
    env['api.endpoint'].routes.first
  end

  def route_method
    route.route_method.downcase
  end

  def route_path
    path = route.route_path.gsub(/^.+:version\/|^\/|:|\(.+\)/, '').tr('/', '-')
    "api.#{route.route_version}.#{path}"
  end

end