sundawg_newrelic_rpm 3.5.8.1

data/lib/new_relic/rack.rb

@@ -0,0 +1,4 @@
+module NewRelic
+  module Rack
+  end
+end

data/lib/new_relic/rack/agent_hooks.rb

@@ -0,0 +1,20 @@
+require 'new_relic/agent/event_listener'
+
+module NewRelic::Rack
+  class AgentHooks
+    def initialize(app, options = {})
+      @app = app
+    end
+
+    # method required by Rack interface
+    # [status, headers, response]
+    def call(env)
+      events = NewRelic::Agent.instance.events
+      events.notify(:before_call, env)
+      result = @app.call(env)
+      events.notify(:after_call, env, result)
+      result
+    end
+  end
+end
+

data/lib/new_relic/rack/browser_monitoring.rb

@@ -0,0 +1,85 @@
+require 'rack'
+
+module NewRelic::Rack
+  class BrowserMonitoring
+
+    def initialize(app, options = {})
+      @app = app
+    end
+
+    # method required by Rack interface
+    def call(env)
+      result = @app.call(env)   # [status, headers, response]
+
+      if (NewRelic::Agent.browser_timing_header != "") && should_instrument?(result[0], result[1])
+        response_string = autoinstrument_source(result[2], result[1])
+
+        if response_string
+          response = Rack::Response.new(response_string, result[0], result[1])
+          response.finish
+        else
+          result
+        end
+      else
+        result
+      end
+    end
+
+    def should_instrument?(status, headers)
+      status == 200 && headers["Content-Type"] && headers["Content-Type"].include?("text/html") &&
+        !headers['Content-Disposition'].to_s.include?('attachment')
+    end
+
+    def autoinstrument_source(response, headers)
+      source = nil
+      response.each {|fragment| source ? (source << fragment.to_s) : (source = fragment.to_s)}
+      return nil unless source
+
+
+      # Only scan the first 50k (roughly) then give up.
+      beginning_of_source = source[0..50_000]
+      # Don't scan for body close unless we find body start
+      if (body_start = beginning_of_source.index("<body")) && (body_close = source.rindex("</body>"))
+
+        footer = NewRelic::Agent.browser_timing_footer
+        header = NewRelic::Agent.browser_timing_header
+
+        x_ua_compatible_found = beginning_of_source.include?('X-UA-Compatible')
+
+        head_pos = if x_ua_compatible_found
+          # put at end of header if X-UA-Compatible meta tag found
+          ::NewRelic::Agent.logger.debug "Detected X-UA-Compatible meta tag. Attempting to insert RUM header at end of head."
+          beginning_of_source.index("</head>")
+        elsif head_open = beginning_of_source.index("<head")
+          ::NewRelic::Agent.logger.debug "Attempting to insert RUM header at beginning of head."
+          # put at the beginning of the header
+          beginning_of_source.index(">", head_open) + 1
+        else
+          ::NewRelic::Agent.logger.debug "Failed to detect head tag. Attempting to insert RUM header at above body tag."
+          # otherwise put the header right above body start
+          body_start
+        end
+
+        # check that head_pos is less than body close.  If it's not something
+        # is really weird and we should punt.
+        if head_pos && (head_pos < body_close)
+          # rebuild the source
+          source = source[0..(head_pos-1)] <<
+            header <<
+            source[head_pos..(body_close-1)] <<
+            footer <<
+            source[body_close..-1]
+        else
+          if head_pos
+            ::NewRelic::Agent.logger.debug "Skipping RUM instrumentation. Failed to detect head tags."
+          else
+            ::NewRelic::Agent.logger.debug "Skipping RUM instrumentation. Detected head is after detected body close."
+          end
+        end
+      end
+      headers['Content-Length'] = source.length.to_s if headers['Content-Length']
+      source
+    end
+  end
+
+end

data/lib/new_relic/rack/developer_mode.rb

@@ -0,0 +1,271 @@
+require 'rack'
+require 'rack/request'
+require 'rack/response'
+require 'rack/file'
+require 'new_relic/metric_parser/metric_parser'
+require 'new_relic/collection_helper'
+
+module NewRelic
+  module Rack
+    class DeveloperMode
+
+      VIEW_PATH = File.expand_path('../../../../ui/views/', __FILE__)
+      HELPER_PATH = File.expand_path('../../../../ui/helpers/', __FILE__)
+      require File.join(HELPER_PATH, 'developer_mode_helper.rb')
+
+
+      include NewRelic::DeveloperModeHelper
+
+      def initialize(app)
+        @app = app
+      end
+
+      def call(env)
+        return @app.call(env) unless /^\/newrelic/ =~ ::Rack::Request.new(env).path_info
+        dup._call(env)
+      end
+
+      protected
+
+      def _call(env)
+        @req = ::Rack::Request.new(env)
+        @rendered = false
+        case @req.path_info
+        when /profile/
+          profile
+        when /file/
+          ::Rack::File.new(VIEW_PATH).call(env)
+        when /index/
+          index
+        when /threads/
+          threads
+        when /reset/
+          reset
+        when /show_sample_detail/
+          show_sample_data
+        when /show_sample_summary/
+          show_sample_data
+        when /show_sample_sql/
+          show_sample_data
+        when /explain_sql/
+          explain_sql
+        when /show_source/
+          show_source
+        when /^\/newrelic\/?$/
+          index
+        else
+          @app.call(env)
+        end
+      end
+
+      private
+
+      def index
+        get_samples
+        render(:index)
+      end
+
+      def reset
+        NewRelic::Agent.instance.transaction_sampler.reset!
+        NewRelic::Agent.instance.sql_sampler.reset!
+        ::Rack::Response.new{|r| r.redirect('/newrelic/')}.finish
+      end
+
+      def explain_sql
+        get_segment
+
+        return render(:sample_not_found) unless @sample
+
+        @sql = @segment[:sql]
+        @trace = @segment[:backtrace]
+
+        if NewRelic::Agent.agent.record_sql == :obfuscated
+          @obfuscated_sql = @segment.obfuscated_sql
+        end
+
+        headers, explanations = @segment.explain_sql
+        if explanations
+          @explanation = explanations
+          if !@explanation.blank?
+            first_row = @explanation.first
+            # Show the standard headers if it looks like a mysql explain plan
+            # Otherwise show blank headers
+            if first_row.length < NewRelic::MYSQL_EXPLAIN_COLUMNS.length
+              @row_headers = nil
+            else
+              @row_headers = NewRelic::MYSQL_EXPLAIN_COLUMNS
+            end
+          end
+        end
+        render(:explain_sql)
+      end
+
+      def profile
+        NewRelic::Control.instance.profiling = params['start'] == 'true'
+        index
+      end
+
+      def threads
+        render(:threads)
+      end
+
+      def render(view, layout=true)
+        add_rack_array = true
+        if view.is_a? Hash
+          layout = false
+          if view[:object]
+            object = view[:object]
+          end
+
+          if view[:collection]
+            return view[:collection].map do |object|
+              render({:partial => view[:partial], :object => object})
+            end.join(' ')
+          end
+
+          if view[:partial]
+            add_rack_array = false
+            view = "_#{view[:partial]}"
+          end
+        end
+        binding = Proc.new {}.binding
+        if layout
+          body = render_with_layout(view) do
+            render_without_layout(view, binding)
+          end
+        else
+          body = render_without_layout(view, binding)
+        end
+        if add_rack_array
+          ::Rack::Response.new(body).finish
+        else
+          body
+        end
+      end
+
+      # You have to call this with a block - the contents returned from
+      # that block are interpolated into the layout
+      def render_with_layout(view)
+        body = ERB.new(File.read(File.join(VIEW_PATH, 'layouts/newrelic_default.rhtml')))
+        body.result(Proc.new {}.binding)
+      end
+
+      # you have to pass a binding to this (a proc) so that ERB can have
+      # access to helper functions and local variables
+      def render_without_layout(view, binding)
+        ERB.new(File.read(File.join(VIEW_PATH, 'newrelic', view.to_s + '.rhtml')), nil, nil, 'frobnitz').result(binding)
+      end
+
+      def content_tag(tag, contents, opts={})
+        opt_values = opts.map {|k, v| "#{k}=\"#{v}\"" }.join(' ')
+        "<#{tag} #{opt_values}>#{contents}</#{tag}>"
+      end
+
+      def sample
+        @sample || @samples[0]
+      end
+
+      def params
+        @req.params
+      end
+
+      def segment
+        @segment
+      end
+
+
+      # show the selected source file with the highlighted selected line
+      def show_source
+        @filename = params['file']
+        line_number = params['line'].to_i
+
+        if !File.readable?(@filename)
+          @source="<p>Unable to read #{@filename}.</p>"
+          return
+        end
+        begin
+          file = File.new(@filename, 'r')
+        rescue => e
+          @source="<p>Unable to access the source file #{@filename} (#{e.message}).</p>"
+          return
+        end
+        @source = ""
+
+        @source << "<pre>"
+        file.each_line do |line|
+          # place an anchor 6 lines above the selected line (if the line # < 6)
+          if file.lineno == line_number - 6
+            @source << "</pre><pre id = 'selected_line'>"
+            @source << line.rstrip
+            @source << "</pre><pre>"
+
+            # highlight the selected line
+          elsif file.lineno == line_number
+            @source << "</pre><pre class = 'selected_source_line'>"
+            @source << line.rstrip
+            @source << "</pre><pre>"
+          else
+            @source << line
+          end
+        end
+        render(:show_source)
+      end
+
+      def show_sample_data
+        get_sample
+
+        return render(:sample_not_found) unless @sample
+
+        @request_params = @sample.params['request_params'] || {}
+        @custom_params = @sample.params['custom_params'] || {}
+
+        controller_metric = @sample.root_segment.called_segments.first.metric_name
+
+        metric_parser = NewRelic::MetricParser::MetricParser.for_metric_named controller_metric
+        @sample_controller_name = metric_parser.controller_name
+        @sample_action_name = metric_parser.action_name
+
+        @sql_segments = @sample.sql_segments
+        if params['d']
+          @sql_segments.sort!{|a,b| b.duration <=> a.duration }
+        end
+
+        sort_method = params['sort'] || :total_time
+        @profile_options = {:min_percent => 0.5, :sort_method => sort_method.to_sym}
+        
+        render(:show_sample)
+      end
+
+      def get_samples
+        @samples = NewRelic::Agent.instance.transaction_sampler.samples.select do |sample|
+          sample.params[:path] != nil
+        end
+
+        return @samples = @samples.sort{|x,y| y.omit_segments_with('(Rails/Application Code Loading)|(Database/.*/.+ Columns)').duration <=>
+          x.omit_segments_with('(Rails/Application Code Loading)|(Database/.*/.+ Columns)').duration} if params['h']
+        return @samples = @samples.sort{|x,y| x.params[:uri] <=> y.params[:uri]} if params['u']
+        @samples = @samples.reverse
+      end
+
+      def get_sample
+        get_samples
+        id = params['id']
+        sample_id = id.to_i
+        @samples.each do |s|
+          if s.sample_id == sample_id
+            @sample = stripped_sample(s)
+            return
+          end
+        end
+      end
+
+      def get_segment
+        get_sample
+        return unless @sample
+
+        segment_id = params['segment'].to_i
+        @segment = @sample.find_segment(segment_id)
+      end
+    end
+  end
+end

data/lib/new_relic/rack/error_collector.rb

@@ -0,0 +1,66 @@
+module NewRelic::Rack
+  class ErrorCollector
+    def initialize(app, options={})
+      @app = app
+    end
+
+    def call(env)
+      @app.call(env)
+    rescue Exception => exception
+      NewRelic::Agent.logger.debug "collecting %p: %s" % [ exception.class, exception.message ]
+      request = Rack::Request.new(env)
+
+      if !should_ignore_error?(exception, request)
+        params = begin
+          request.params
+        rescue => err
+          warning = "failed to capture request parameters: %p: %s" % [ err.class, err.message ]
+          NewRelic::Agent.logger.warn(warning)
+          {'error' => warning}
+        end
+
+        NewRelic::Agent.instance.error_collector.notice_error(exception,
+                                                              :uri => request.path,
+                                                          :referer => request.referer,
+                                                   :request_params => params)
+      end
+      raise exception
+    end
+
+    def should_ignore_error?(error, request)
+      NewRelic::Agent.instance.error_collector.error_is_ignored?(error) ||
+        ignored_in_controller?(error, request)
+    end
+
+    def ignored_in_controller?(exception, request)
+      if request.env['action_dispatch.request.parameters']
+        ignore_actions = newrelic_ignore_for_controller(request.env['action_dispatch.request.parameters']['controller'])
+        action_name = request.env['action_dispatch.request.parameters']['action']
+
+        case ignore_actions
+        when nil; false
+        when Hash
+          only_actions = Array(ignore_actions[:only])
+          except_actions = Array(ignore_actions[:except])
+          only_actions.include?(action_name.to_sym) ||
+            (except_actions.any? &&
+             !except_actions.include?(action_name.to_sym))
+        else
+          true
+        end
+      end
+    end
+
+    def newrelic_ignore_for_controller(controller_name)
+      if controller_name
+        controller_constant_name = (controller_name + "_controller").camelize
+        if Object.const_defined?(controller_constant_name)
+          controller = controller_constant_name.constantize
+          controller.instance_variable_get(:@do_not_trace)
+        end
+      end
+    rescue NameError
+      nil
+    end
+  end
+end