Contents

require 'tracebin/timer'
require 'tracebin/puppet_master'
# require 'tracebin/initializer'

module Tracebin
  class Middleware
    attr_reader :config, :logger

    def initialize(app)
      @app = app

      @config = Tracebin::Agent.config
      @logger = Tracebin::Agent.logger

      start_agent_parent_process
    end

    def call(env)
      dup.__call(env)
    end

    def __call(env)
      start_agent_child_process

      if agent_disabled?(env)
        @logger.debug "TRACEBIN: Tracebin disabled for this request."
        return @app.call env
      else
        @tracebin_timer = Timer.new
        @tracebin_timer.start!

        status, headers, response = @app.call(env)

        @tracebin_timer.transaction_name = fetch_endpoint_name(env)

        @tracebin_timer.stop!

        PuppetMaster.new(@tracebin_timer).process

        return [status, headers, response]
      end
    end

    private

    def fetch_endpoint_name(env)
      if controller = env['action_controller.instance']
        "#{controller.class}##{controller.params['action']}"
      elsif route = env['sinatra.route']
        route
      else
        'RackTransaction'
      end
    end

    def start_agent_child_process
      Tracebin::Agent.start_child_process
    end

    def start_agent_parent_process
      Tracebin::Agent.start_parent_process
    end

    def agent_disabled?(env)
      path = env['REQUEST_PATH']
      ignored_paths = config.ignored_paths.map { |root| %r{^#{root}} }

      !Tracebin::Agent.child_process_started? ||
        ignored_paths.any? { |root| !!root.match(path) }
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
tracebin-0.0.13	lib/tracebin/middleware.rb
tracebin-0.0.12	lib/tracebin/middleware.rb