module NewRelic::Agent::Instrumentation::Rack::Chain

Public Class Methods

instrument!(builder_class) click to toggle source
# File lib/new_relic/agent/instrumentation/rack/chain.rb, line 8
def self.instrument!(builder_class)
  NewRelic::Agent::Instrumentation::RackBuilder.track_deferred_detection(builder_class)

  builder_class.class_eval do
    include ::NewRelic::Agent::Instrumentation::RackBuilder

    def to_app_with_newrelic_deferred_dependency_detection
      with_deferred_dependency_detection { to_app_without_newrelic }
    end

    alias_method(:to_app_without_newrelic, :to_app)
    alias_method(:to_app, :to_app_with_newrelic_deferred_dependency_detection)

    if ::NewRelic::Agent::Instrumentation::RackHelpers.middleware_instrumentation_enabled?
      ::NewRelic::Agent.logger.info("Installing #{builder_class} middleware instrumentation")

      def run_with_newrelic(app = nil, *args, &block)
        app_or_block = app || block
        run_with_tracing(app_or_block) do |wrapped|
          # Rack::Builder#run for Rack v3+ supports a block, and does not
          # support args. Whether a block or an app is provided, that
          # callable object will be wrapped into a MiddlewareProxy
          # instance. That proxy instance must then be passed to
          # run_without_newrelic as the app argument.
          block ? run_without_newrelic(wrapped, &nil) : run_without_newrelic(wrapped, *args)
        end
      end

      alias_method(:run_without_newrelic, :run)
      alias_method(:run, :run_with_newrelic)

      def use_with_newrelic(middleware_class, *args, &block)
        use_with_tracing(middleware_class) { |wrapped_class| use_without_newrelic(wrapped_class, *args, &block) }
      end
      ruby2_keywords(:use_with_newrelic) if respond_to?(:ruby2_keywords, true)

      alias_method(:use_without_newrelic, :use)
      alias_method(:use, :use_with_newrelic)
    end
  end
end

Public Instance Methods

run_with_newrelic(app = nil, *args, &block) click to toggle source
# File lib/new_relic/agent/instrumentation/rack/chain.rb, line 24
def run_with_newrelic(app = nil, *args, &block)
  app_or_block = app || block
  run_with_tracing(app_or_block) do |wrapped|
    # Rack::Builder#run for Rack v3+ supports a block, and does not
    # support args. Whether a block or an app is provided, that
    # callable object will be wrapped into a MiddlewareProxy
    # instance. That proxy instance must then be passed to
    # run_without_newrelic as the app argument.
    block ? run_without_newrelic(wrapped, &nil) : run_without_newrelic(wrapped, *args)
  end
end
to_app_with_newrelic_deferred_dependency_detection() click to toggle source
# File lib/new_relic/agent/instrumentation/rack/chain.rb, line 14
def to_app_with_newrelic_deferred_dependency_detection
  with_deferred_dependency_detection { to_app_without_newrelic }
end
use_with_newrelic(middleware_class, *args, &block) click to toggle source
# File lib/new_relic/agent/instrumentation/rack/chain.rb, line 39
def use_with_newrelic(middleware_class, *args, &block)
  use_with_tracing(middleware_class) { |wrapped_class| use_without_newrelic(wrapped_class, *args, &block) }
end