wd_newrelic_rpm 3.3.4.1

data/lib/new_relic/merbtasks.rb

@@ -0,0 +1,6 @@
+namespace :newrelic do
+  desc "Install the developer mode newrelic.yml file"
+  task :default do
+    load File.expand_path(File.join(__FILE__,"..","..","install.rb"))
+  end
+end

data/lib/new_relic/metric_data.rb

@@ -0,0 +1,51 @@
+module NewRelic
+  class MetricData
+    # nil, or a NewRelic::MetricSpec object if we have no cached ID
+    attr_accessor :metric_spec
+    # nil or a cached integer ID for the metric from the collector.
+    attr_accessor :metric_id
+    # the actual statistics object
+    attr_accessor :stats
+
+    def initialize(metric_spec, stats, metric_id)
+      @metric_spec = metric_spec
+      self.stats = stats
+      self.metric_id = metric_id
+    end
+
+    def eql?(o)
+     (metric_spec.eql? o.metric_spec) && (stats.eql? o.stats)
+    end
+
+    def original_spec
+      @original_spec || @metric_spec
+    end
+    
+    # assigns a new metric spec, and retains the old metric spec as
+    # @original_spec if it exists currently
+    def metric_spec= new_spec
+      @original_spec = @metric_spec if @metric_spec
+      @metric_spec = new_spec
+    end
+
+    def hash
+      metric_spec.hash ^ stats.hash
+    end
+
+    # Serialize with all attributes, but if the metric id is not nil, then don't send the metric spec
+    def to_json(*a)
+       %Q[{"metric_spec":#{metric_id ? 'null' : metric_spec.to_json},"stats":{"total_exclusive_time":#{stats.total_exclusive_time},"min_call_time":#{stats.min_call_time},"call_count":#{stats.call_count},"sum_of_squares":#{stats.sum_of_squares},"total_call_time":#{stats.total_call_time},"max_call_time":#{stats.max_call_time}},"metric_id":#{metric_id ? metric_id : 'null'}}]
+    end
+    
+    def to_s
+      if metric_spec
+        "#{metric_spec.name}(#{metric_spec.scope}): #{stats}"
+      else
+        "#{metric_id}: #{stats}"
+      end
+    end
+    def inspect
+      "#<MetricData metric_spec:#{metric_spec.inspect}, stats:#{stats.inspect}, metric_id:#{metric_id.inspect}>"
+    end
+  end
+end

data/lib/new_relic/metric_spec.rb

@@ -0,0 +1,76 @@
+# this struct uniquely defines a metric, optionally inside
+# the call scope of another metric
+class NewRelic::MetricSpec
+  attr_accessor   :name
+  attr_accessor   :scope
+  
+  # the maximum length of a metric name or metric scope
+  MAX_LENGTH = 255
+  LENGTH_RANGE = (0...MAX_LENGTH)
+  # Need a "zero-arg" constructor so it can be instantiated from java (using
+  # jruby) for sending responses to ruby agents from the java collector.
+  #
+  def initialize(metric_name = '', metric_scope = '')
+    self.name = (metric_name || '') && metric_name[LENGTH_RANGE]
+    self.scope = metric_scope && metric_scope[LENGTH_RANGE]
+  end
+  
+  # truncates the name and scope to the MAX_LENGTH
+  def truncate!
+    self.name = name[LENGTH_RANGE] if name && name.size > MAX_LENGTH
+    self.scope = scope[LENGTH_RANGE] if scope && scope.size > MAX_LENGTH
+  end
+  
+  def ==(o)
+    self.eql?(o)
+  end
+
+  def eql? o
+    self.class == o.class &&
+    name.eql?(o.name) &&
+    # coerce scope to a string and compare
+     scope.to_s == o.scope.to_s
+  end
+
+  def hash
+    h = name.hash
+    h ^= scope.hash unless scope.nil?
+    h
+  end
+  # return a new metric spec if the given regex
+  # matches the name or scope.
+  def sub(pattern, replacement, apply_to_scope = true)
+    NewRelic::Control.instance.log.warn("The sub method on metric specs is deprecated") rescue nil
+    return nil if name !~ pattern &&
+     (!apply_to_scope || scope.nil? || scope !~ pattern)
+    new_name = name.sub(pattern, replacement)[LENGTH_RANGE]
+
+    if apply_to_scope
+      new_scope = (scope && scope.sub(pattern, replacement)[LENGTH_RANGE])
+    else
+      new_scope = scope
+    end
+
+    self.class.new new_name, new_scope
+  end
+
+  def to_s
+    return name if scope.empty?
+    "#{name}:#{scope}"
+  end
+
+  def inspect
+    "#<NewRelic::MetricSpec '#{name}':'#{scope}'>"
+  end
+
+  def to_json(*a)
+    {'name' => name,
+    'scope' => scope}.to_json(*a)
+  end
+
+  def <=>(o)
+    namecmp = self.name <=> o.name
+    return namecmp if namecmp != 0
+    return (self.scope || '') <=> (o.scope || '')
+  end
+end

data/lib/new_relic/metrics.rb

@@ -0,0 +1,9 @@
+module NewRelic
+  module Metrics
+    CONTROLLER = "Controller"
+    DISPATCHER = "HttpDispatcher"
+    ACTIVE_RECORD = "ActiveRecord"
+    USER_TIME = "CPU/User Time"
+    MEMORY = "Memory/Physical"
+  end
+end

data/lib/new_relic/noticed_error.rb

@@ -0,0 +1,29 @@
+# This class encapsulates an error that was noticed by New Relic in a managed app.
+class NewRelic::NoticedError
+  extend NewRelic::CollectionHelper
+  attr_accessor :path, :timestamp, :params, :exception_class, :message
+
+  def initialize(path, data, exception, timestamp = Time.now)
+    self.path = path
+    self.params = NewRelic::NoticedError.normalize_params(data)
+
+    self.exception_class = exception.is_a?(Exception) ? exception.class.name : 'Error'
+
+    if exception.respond_to?('original_exception')
+      self.message = exception.original_exception.message.to_s
+    else
+      self.message = (exception || '<no message>').to_s
+    end
+
+    # clamp long messages to 4k so that we don't send a lot of
+    # overhead across the wire
+    self.message = self.message[0..4095] if self.message.length > 4096
+    
+    # obfuscate error message if necessary
+    if NewRelic::Control.instance.fetch('high_security', false)
+      self.message = NewRelic::Agent::Database.obfuscate_sql(self.message)
+    end
+    
+    self.timestamp = timestamp
+  end
+end

data/lib/new_relic/rack/browser_monitoring.rb

@@ -0,0 +1,76 @@
+require 'rack'
+
+module NewRelic::Rack
+  class BrowserMonitoring
+
+    def initialize(app, options = {})
+      @app = app
+    end
+
+    # method required by Rack interface
+    def call(env)
+      # Two experimental options for allowing TT capture based on http params
+      #
+      # if req.params['nr_capture_deep_tt']
+        # NewRelic::Agent::TransactionInfo.get.force_persist = true
+        # NewRelic::Agent::TransactionInfo.get.capture_deep_tt = true
+      # end
+
+      # if req.params['nr_capture_tt']
+        # NewRelic::Agent::TransactionInfo.get.force_persist = true
+      # end
+
+      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
+
+      body_start = source.index("<body")
+      body_close = source.rindex("</body>")
+
+      if body_start && body_close
+        footer = NewRelic::Agent.browser_timing_footer
+        header = NewRelic::Agent.browser_timing_header
+
+        if source.include?('X-UA-Compatible')
+          # put at end of header if UA-Compatible meta tag found
+          head_pos = source.index("</head>")
+        elsif head_open = source.index("<head")
+          # put at the beginning of the header
+          head_pos = source.index(">", head_open) + 1
+        else
+          # put the header right above body start
+          head_pos = body_start
+        end
+
+        source = source[0..(head_pos-1)] + header + source[head_pos..(body_close-1)] + footer + source[body_close..-1]
+
+        headers['Content-Length'] = source.length.to_s if headers['Content-Length']
+      end
+
+      source
+    end
+  end
+
+end

data/lib/new_relic/rack/developer_mode.rb

@@ -0,0 +1,268 @@
+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
+        
+        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