Contents

module Vitals::Integrations::Rack
  class Requests
    REQUEST_METHOD = 'REQUEST_METHOD'.freeze

    RACK_PATH_INFO = 'PATH_INFO'.freeze
    SINATRA_PATH_INFO = 'sinatra.route'.freeze
    GRAPE_PATH_INFO = 'api.endpoint'.freeze
    RAILS_PATH_INFO = 'action_controller.instance'.freeze

    def initialize(app, options = {})
      @app = app
      @prefix = options[:prefix] ? options[:prefix] + "." : nil
    end

    def call(env)
      start = Time.now
      status, header, body = @app.call(env)
      t = Time.now - start
      path = if env[SINATRA_PATH_INFO]
            Requests.sinatra_path(env)
          elsif env[GRAPE_PATH_INFO]
            Requests.grape_path(env)
          elsif env[RAILS_PATH_INFO]
            Requests.rails_path(env)
          else
            Requests.rack_path(env)
          end
      m = "requests.#{@prefix}#{path}_#{env[REQUEST_METHOD].downcase}.#{status}"

      # TODO add option to customize 'requests' through options
      Vitals.timing(m, Vitals::Utils.sec_to_ms(t))

      [status, header, body]
    end

    private

    def self.sinatra_path(env)
      env[SINATRA_PATH_INFO].gsub(/^\w+\s+\//, '')
    end

    def self.grape_path(env)
      Vitals::Utils.grape_path(env[GRAPE_PATH_INFO].route)
    end

    def self.rails_path(env)
      ctrl = env[RAILS_PATH_INFO]
      "#{ctrl.controller_name}_#{ctrl.action_name}"
    end

    def self.rack_path(env)
      ''
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
vitals-0.5.0	lib/vitals/integrations/rack/requests.rb
vitals-0.4.0	lib/vitals/integrations/rack/requests.rb