ghazel-newrelic_rpm 3.1.0.1

data/lib/new_relic/agent/stats_engine/samplers.rb

@@ -0,0 +1,89 @@
+module NewRelic
+module Agent
+  class StatsEngine
+    module Shim # :nodoc:
+      def add_sampler(*args); end
+      def add_harvest_sampler(*args); end
+      def start_sampler_thread(*args); end
+    end
+
+    module Samplers
+
+      # By default a sampler polls on harvest time, once a minute.  However you can
+      # override #use_harvest_sampler? to return false and it will sample
+      # every POLL_PERIOD seconds on a background thread.
+      POLL_PERIOD = 20
+
+      def start_sampler_thread
+
+        return if @sampler_thread && @sampler_thread.alive?
+
+        # start up a thread that will periodically poll for metric samples
+        return if periodic_samplers.empty?
+
+        @sampler_thread = Thread.new do
+          while true do
+            begin
+              sleep POLL_PERIOD
+              poll periodic_samplers
+            end
+          end
+        end
+        @sampler_thread['newrelic_label'] = 'Sampler Tasks'
+      end
+
+      private
+
+      def add_sampler_to(sampler_array, sampler)
+        raise "Sampler #{sampler.inspect} is already registered.  Don't call add_sampler directly anymore." if sampler_array.include?(sampler)
+        sampler_array << sampler
+        sampler.stats_engine = self
+      end
+
+      def log_added_sampler(type, sampler)
+        log.debug "Adding #{type} sampler: #{sampler.inspect}"
+      end
+
+      public
+
+      # Add an instance of Sampler to be invoked about every 10 seconds on a background
+      # thread.
+      def add_sampler(sampler)
+        add_sampler_to(periodic_samplers, sampler)
+        log_added_sampler('periodic', sampler)
+      end
+
+      # Add a sampler to be invoked just before each harvest.
+      def add_harvest_sampler(sampler)
+        add_sampler_to(harvest_samplers, sampler)
+        log_added_sampler('harvest-time', sampler)
+      end
+
+      private
+
+      # Call poll on each of the samplers.  Remove
+      # the sampler if it raises.
+      def poll(samplers)
+        samplers.delete_if do |sampled_item|
+          begin
+            sampled_item.poll
+            false # it's okay.  don't delete it.
+          rescue Exception => e
+            log.error "Removing #{sampled_item} from list"
+            log.error e
+            log.debug e.backtrace.to_s
+            true # remove the sampler
+          end
+        end
+      end
+
+      def harvest_samplers
+        @harvest_samplers ||= []
+      end
+      def periodic_samplers
+        @periodic_samplers ||= []
+      end
+    end
+  end
+end
+end

data/lib/new_relic/agent/stats_engine/transactions.rb

@@ -0,0 +1,184 @@
+module NewRelic
+module Agent
+  class StatsEngine
+
+    # Defines methods that stub out the stats engine methods
+    # when the agent is disabled
+
+    class ScopeStackElement
+      attr_reader :name, :deduct_call_time_from_parent
+      attr_accessor :children_time
+      def initialize(name, deduct_call_time)
+        @name = name
+        @deduct_call_time_from_parent = deduct_call_time
+        @children_time = 0
+      end
+    end
+
+    module Transactions
+      module Shim # :nodoc:
+        def start_transaction(*args); end
+        def end_transaction; end
+        def push_scope(*args); end
+        def transaction_sampler=(*args); end
+        def scope_name=(*args); end
+        def scope_name; end
+        def pop_scope(*args); end
+      end
+
+      def transaction_sampler= sampler
+        fail "Can't add a scope listener midflight in a transaction" if scope_stack.any?
+        @transaction_sampler = sampler
+      end
+
+      def remove_transaction_sampler(l)
+        @transaction_sampler = nil
+      end
+
+      def push_scope(metric, time = Time.now.to_f, deduct_call_time_from_parent = true)
+
+        stack = scope_stack
+        if collecting_gc?
+          if stack.empty?
+            # reset the gc time so we only include gc time spent during this call
+            @last_gc_timestamp = gc_time
+            @last_gc_count = gc_collections
+          else
+            capture_gc_time
+          end
+        end
+        @transaction_sampler.notice_push_scope metric, time if @transaction_sampler
+        scope = ScopeStackElement.new(metric, deduct_call_time_from_parent)
+        stack.push scope
+        scope
+      end
+
+      def pop_scope(expected_scope, duration, time=Time.now.to_f)
+        capture_gc_time if collecting_gc?
+        stack = scope_stack
+        scope = stack.pop
+        fail "unbalanced pop from blame stack, got #{scope ? scope.name : 'nil'}, expected #{expected_scope ? expected_scope.name : 'nil'}" if scope != expected_scope
+
+        if !stack.empty?
+          if scope.deduct_call_time_from_parent
+            stack.last.children_time += duration
+          else
+            stack.last.children_time += scope.children_time
+          end
+        end
+        @transaction_sampler.notice_pop_scope(scope.name, time) if @transaction_sampler
+        scope
+      end
+
+      def peek_scope
+        scope_stack.last
+      end
+
+      # set the name of the transaction for the current thread, which will be used
+      # to define the scope of all traced methods called on this thread until the
+      # scope stack is empty.
+      #
+      # currently the transaction name is the name of the controller action that
+      # is invoked via the dispatcher, but conceivably we could use other transaction
+      # names in the future if the traced application does more than service http request
+      # via controller actions
+      def scope_name=(transaction)
+        Thread::current[:newrelic_scope_name] = transaction
+        Thread::current[:newrelic_most_recent_transaction] = transaction
+      end
+
+      def scope_name
+        Thread::current[:newrelic_scope_name]
+      end
+
+      # Start a new transaction, unless one is already in progress
+      def start_transaction(name = nil)
+        Thread::current[:newrelic_scope_stack] ||= []
+        self.scope_name = name if name
+      end
+
+      # Try to clean up gracefully, otherwise we leave things hanging around on thread locals.
+      # If it looks like a transaction is still in progress, then maybe this is an inner transaction
+      # and is ignored.
+      #
+      def end_transaction
+        stack = scope_stack
+
+        if stack && stack.empty?
+          Thread::current[:newrelic_scope_stack] = nil
+          Thread::current[:newrelic_scope_name] = nil
+        end
+      end
+
+      private
+
+      # Make sure we don't do this in a multi-threaded environment
+      def collecting_gc?
+        if !defined?(@@collecting_gc)
+          @@collecting_gc = false
+          if !NewRelic::Control.instance.multi_threaded?
+            @@collecting_gc = true if GC.respond_to?(:time) && GC.respond_to?(:collections) # 1.8.x
+            @@collecting_gc = true if defined?(GC::Profiler) && GC::Profiler.enabled? # 1.9.2
+          end
+        end
+        @@collecting_gc
+      end
+
+      # The total number of times the garbage collector has run since
+      # profiling was enabled
+      def gc_collections
+        if GC.respond_to?(:count)
+          GC.count
+        elsif GC.respond_to?(:collections)
+          GC.collections
+        end
+      end
+
+      # The total amount of time taken by garbage collection since
+      # profiling was enabled
+      def gc_time
+        if GC.respond_to?(:time)
+          GC.time
+        elsif defined?(GC::Profiler) && GC::Profiler.respond_to?(:total_time)
+          # The 1.9 profiler returns a time in usec
+          GC::Profiler.total_time * 1000000.0
+        end
+      end
+
+      # Assumes collecting_gc?
+      def capture_gc_time
+        # Skip this if we are already in this segment
+        return if !scope_stack.empty? && scope_stack.last.name == "GC/cumulative"
+        num_calls = gc_collections - @last_gc_count
+        elapsed = (gc_time - @last_gc_timestamp).to_f
+        @last_gc_timestamp = gc_time
+        @last_gc_count = gc_collections
+        
+        if defined?(GC::Profiler)
+          GC::Profiler.clear
+          @last_gc_timestamp = 0
+        end
+        
+        if num_calls > 0
+          # µs to seconds
+          elapsed = elapsed / 1000000.0
+          # Allocate the GC time to a scope as if the GC just ended
+          # right now.
+          time = Time.now.to_f
+          gc_scope = push_scope("GC/cumulative", time - elapsed)
+          # GC stats are collected into a blamed metric which allows
+          # us to show the stats controller by controller
+          gc_stats = NewRelic::Agent.get_stats(gc_scope.name, true)
+          gc_stats.record_multiple_data_points(elapsed, num_calls)
+          pop_scope(gc_scope, elapsed, time)
+        end
+      end
+
+      def scope_stack
+        Thread::current[:newrelic_scope_stack] ||= []
+      end
+
+    end
+  end
+end
+end

data/lib/new_relic/agent/transaction_sample_builder.rb

@@ -0,0 +1,101 @@
+require 'new_relic/collection_helper'
+require 'new_relic/transaction_sample'
+require 'new_relic/control'
+require 'new_relic/agent/instrumentation/metric_frame'
+module NewRelic
+  module Agent
+    # a builder is created with every sampled transaction, to dynamically
+    # generate the sampled data.  It is a thread-local object, and is not
+    # accessed by any other thread so no need for synchronization.
+    class TransactionSampleBuilder
+      attr_reader :current_segment, :sample
+
+      include NewRelic::CollectionHelper
+
+      def initialize(time=Time.now)
+        @sample = NewRelic::TransactionSample.new(time.to_f)
+        @sample_start = time.to_f
+        @current_segment = @sample.root_segment
+      end
+
+      def sample_id
+        @sample.sample_id
+      end
+      def ignored?
+        @ignore || @sample.params[:path].nil?
+      end
+      def ignore_transaction
+        @ignore = true
+      end
+      def trace_entry(metric_name, time)
+        segment = @sample.create_segment(time.to_f - @sample_start, metric_name)
+        @current_segment.add_called_segment(segment)
+        @current_segment = segment
+      end
+
+      def trace_exit(metric_name, time)
+        if metric_name != @current_segment.metric_name
+          fail "unbalanced entry/exit: #{metric_name} != #{@current_segment.metric_name}"
+        end
+        @current_segment.end_trace(time.to_f - @sample_start)
+        @current_segment = @current_segment.parent_segment
+      end
+
+      def finish_trace(time)
+        # This should never get called twice, but in a rare case that we can't reproduce in house it does.
+        # log forensics and return gracefully
+        if @sample.frozen?
+          log = NewRelic::Control.instance.log
+          log.error "Unexpected double-freeze of Transaction Trace Object: \n#{@sample.to_s}"
+          return
+        end
+        @sample.root_segment.end_trace(time.to_f - @sample_start)
+        @sample.params[:custom_params] = normalize_params(NewRelic::Agent::Instrumentation::MetricFrame.custom_parameters)
+        @sample.freeze
+        @current_segment = nil
+      end
+
+      def scope_depth
+        depth = -1        # have to account for the root
+        current = @current_segment
+
+        while(current)
+          depth += 1
+          current = current.parent_segment
+        end
+
+        depth
+      end
+
+      def freeze
+        @sample.freeze unless sample.frozen?
+      end
+
+      def set_profile(profile)
+        @sample.profile = profile
+      end
+
+      def set_transaction_info(path, uri, params)
+        @sample.params[:path] = path
+
+        if NewRelic::Control.instance.capture_params
+          params = normalize_params params
+
+          @sample.params[:request_params].merge!(params)
+          @sample.params[:request_params].delete :controller
+          @sample.params[:request_params].delete :action
+        end
+        @sample.params[:uri] ||= uri || params[:uri]
+      end
+
+      def set_transaction_cpu_time(cpu_time)
+        @sample.params[:cpu_time] = cpu_time
+      end
+
+      def sample
+        @sample
+      end
+
+    end
+  end
+end

data/lib/new_relic/agent/transaction_sampler.rb

@@ -0,0 +1,378 @@
+require 'new_relic/agent'
+require 'new_relic/control'
+require 'new_relic/agent/transaction_sample_builder'
+module NewRelic
+  module Agent
+
+    class TransactionSampler
+
+      # Module defining methods stubbed out when the agent is disabled
+      module Shim #:nodoc:
+        def notice_first_scope_push(*args); end
+        def notice_push_scope(*args); end
+        def notice_pop_scope(*args); end
+        def notice_scope_empty(*args); end
+      end
+
+      BUILDER_KEY = :transaction_sample_builder
+
+      attr_accessor :stack_trace_threshold, :random_sampling, :sampling_rate
+      attr_reader :samples, :last_sample, :disabled
+
+      def initialize
+        # @samples is an array of recent samples up to @max_samples in
+        # size - it's only used by developer mode
+        @samples = []
+        @max_samples = 100
+
+        # @harvest_count is a count of harvests used for random
+        # sampling - we pull 1 @random_sample in every @sampling_rate harvests
+        @harvest_count = 0
+        @random_sample = nil
+        @sampling_rate = 10
+
+        # @segment_limit and @stack_trace_threshold come from the
+        # configuration file, with built-in defaults that should
+        # suffice for most customers
+        config = NewRelic::Control.instance
+        sampler_config = config.fetch('transaction_tracer', {})
+        @segment_limit = sampler_config.fetch('limit_segments', 4000)
+        @stack_trace_threshold = sampler_config.fetch('stack_trace_threshold', 0.500).to_f
+
+        # This lock is used to synchronize access to the @last_sample
+        # and related variables. It can become necessary on JRuby or
+        # any 'honest-to-god'-multithreaded system
+        @samples_lock = Mutex.new
+      end
+
+      # Returns the current sample id, delegated from `builder`
+      def current_sample_id
+        b=builder
+        b and b.sample_id
+      end
+
+      # Enable the transaction sampler - this also registers it with
+      # the statistics engine.
+      def enable
+        @disabled = false
+        NewRelic::Agent.instance.stats_engine.transaction_sampler = self
+      end
+
+      # Disable the transaction sampler - this also deregisters it
+      # with the statistics engine.
+      def disable
+        @disabled = true
+        NewRelic::Agent.instance.stats_engine.remove_transaction_sampler(self)
+      end
+
+      # Set with an integer value n, this takes one in every n
+      # harvested samples. It also resets the harvest count to a
+      # random integer between 0 and (n-1)
+      def sampling_rate=(val)
+        @sampling_rate = val.to_i
+        @harvest_count = rand(val.to_i).to_i
+      end
+
+
+      # Creates a new transaction sample builder, unless the
+      # transaction sampler is disabled. Takes a time parameter for
+      # the start of the transaction sample
+      def notice_first_scope_push(time)
+        start_builder(time.to_f) unless disabled
+      end
+
+      # This delegates to the builder to create a new open transaction
+      # segment for the specified scope, beginning at the optionally
+      # specified time.
+      #
+      # Note that in developer mode, this captures a stacktrace for
+      # the beginning of each segment, which can be fairly slow
+      def notice_push_scope(scope, time=Time.now)
+        return unless builder
+
+        builder.trace_entry(scope, time.to_f)
+
+        capture_segment_trace if NewRelic::Control.instance.developer_mode?
+      end
+
+      # in developer mode, capture the stack trace with the segment.
+      # this is cpu and memory expensive and therefore should not be
+      # turned on in production mode
+      def capture_segment_trace
+        return unless NewRelic::Control.instance.developer_mode?
+        segment = builder.current_segment
+        if segment
+          # Strip stack frames off the top that match /new_relic/agent/
+          trace = caller
+          while trace.first =~/\/lib\/new_relic\/agent\//
+            trace.shift
+          end
+
+          trace = trace[0..39] if trace.length > 40
+          segment[:backtrace] = trace
+        end
+      end
+
+      # Defaults to zero, otherwise delegated to the transaction
+      # sample builder
+      def scope_depth
+        return 0 unless builder
+
+        builder.scope_depth
+      end
+
+      # Informs the transaction sample builder about the end of a
+      # traced scope
+      def notice_pop_scope(scope, time = Time.now)
+        return unless builder
+        raise "frozen already???" if builder.sample.frozen?
+        builder.trace_exit(scope, time.to_f)
+      end
+
+      # This is called when we are done with the transaction.  We've
+      # unwound the stack to the top level. It also clears the
+      # transaction sample builder so that it won't continue to have
+      # scopes appended to it.
+      #
+      # It sets various instance variables to the finished sample,
+      # depending on which settings are active. See `store_sample`
+      def notice_scope_empty(time=Time.now)
+
+        last_builder = builder
+        return unless last_builder
+
+        last_builder.finish_trace(time.to_f)
+        clear_builder
+        return if last_builder.ignored?
+
+        @samples_lock.synchronize do
+          # NB this instance variable may be used elsewhere, it's not
+          # just a side effect
+          @last_sample = last_builder.sample
+          store_sample(@last_sample)
+        end
+      end
+
+      # Samples can be stored in three places: the random sample
+      # variable, when random sampling is active, the developer mode
+      # @samples array, and the @slowest_sample variable if it is
+      # slower than the current occupant of that slot
+      def store_sample(sample)
+        store_random_sample(sample)
+        store_sample_for_developer_mode(sample)
+        store_slowest_sample(sample)
+      end
+
+      # Only active when random sampling is true - this is very rarely
+      # used. Always store the most recent sample so that random
+      # sampling can pick a few of the samples to store, upon harvest
+      def store_random_sample(sample)
+        if @random_sampling
+          @random_sample = sample
+        end
+      end
+
+      # Samples take up a ton of memory, so we only store a lot of
+      # them in developer mode - we truncate to @max_samples
+      def store_sample_for_developer_mode(sample)
+        return unless NewRelic::Control.instance.developer_mode?
+        @samples = [] unless @samples
+        @samples << sample
+        truncate_samples
+      end
+
+      # Sets @slowest_sample to the passed in sample if it is slower
+      # than the current sample in @slowest_sample
+      def store_slowest_sample(sample)
+        @slowest_sample = sample if slowest_sample?(@slowest_sample, sample)
+      end
+
+      # Checks to see if the old sample exists, or if it's duration is
+      # less than the new sample
+      def slowest_sample?(old_sample, new_sample)
+        old_sample.nil? || (new_sample.duration > old_sample.duration)
+      end
+
+      # Smashes the @samples array down to the length of @max_samples
+      # by taking the last @max_samples elements of the array
+      def truncate_samples
+        if @samples.length > @max_samples
+          @samples = @samples[-@max_samples..-1]
+        end
+      end
+
+      # Delegates to the builder to store the path, uri, and
+      # parameters if the sampler is active
+      def notice_transaction(path, uri=nil, params={})
+        builder.set_transaction_info(path, uri, params) if !disabled && builder
+      end
+
+      # Tells the builder to ignore a transaction, if we are currently
+      # creating one. Only causes the sample to be ignored upon end of
+      # the transaction, and does not change the metrics gathered
+      # outside of the sampler
+      def ignore_transaction
+        builder.ignore_transaction if builder
+      end
+
+      # For developer mode profiling support - delegates to the builder
+      def notice_profile(profile)
+        builder.set_profile(profile) if builder
+      end
+
+      # Sets the CPU time used by a transaction, delegates to the builder
+      def notice_transaction_cpu_time(cpu_time)
+        builder.set_transaction_cpu_time(cpu_time) if builder
+      end
+
+      MAX_DATA_LENGTH = 16384
+      # This method is used to record metadata into the currently
+      # active segment like a sql query, memcache key, or Net::HTTP uri
+      #
+      # duration is seconds, float value.
+      def notice_extra_data(message, duration, key, config=nil, config_key=nil)
+        return unless builder
+        segment = builder.current_segment
+        if segment
+          segment[key] = truncate_message(append_new_message(segment[key], message))
+          segment[config_key] = config if config_key
+          append_backtrace(segment, duration)
+        end
+      end
+
+      private :notice_extra_data
+
+      # Truncates the message to `MAX_DATA_LENGTH` if needed, and
+      # appends an ellipsis because it makes the trucation clearer in
+      # the UI
+      def truncate_message(message)
+        if message.length > (MAX_DATA_LENGTH - 4)
+          message[0..MAX_DATA_LENGTH - 4] + '...'
+        else
+          message
+        end
+      end
+
+      # Allows the addition of multiple pieces of metadata to one
+      # segment - i.e. traced method calls multiple sql queries
+      def append_new_message(old_message, message)
+        if old_message
+          old_message + ";\n" + message
+        else
+          message
+        end
+      end
+
+      # Appends a backtrace to a segment if that segment took longer
+      # than the specified duration
+      def append_backtrace(segment, duration)
+        segment[:backtrace] = caller.join("\n") if duration >= @stack_trace_threshold
+      end
+
+      # some statements (particularly INSERTS with large BLOBS
+      # may be very large; we should trim them to a maximum usable length
+      # config is the driver configuration for the connection
+      # duration is seconds, float value.
+      def notice_sql(sql, config, duration)
+        if NewRelic::Agent.is_sql_recorded?
+          notice_extra_data(sql, duration, :sql, config, :connection_config)
+        end
+      end
+
+      # Adds non-sql metadata to a segment - generally the memcached
+      # key
+      #
+      # duration is seconds, float value.
+      def notice_nosql(key, duration)
+        notice_extra_data(key, duration, :key)
+      end
+
+      # Every 1/n harvests, adds the most recent sample to the harvest
+      # array if it exists. Makes sure that the random sample is not
+      # also the slowest sample for this harvest by `uniq!`ing the
+      # result array
+      #
+      # random sampling is very, very seldom used
+      def add_random_sample_to(result)
+        return unless @random_sampling && @sampling_rate && @sampling_rate.to_i > 0
+        @harvest_count += 1
+        if (@harvest_count.to_i % @sampling_rate.to_i) == 0
+          result << @random_sample if @random_sample
+        end
+        result.uniq!
+        nil # don't assume this method returns anything
+      end
+
+      # Returns an array of slow samples, with either one or two
+      # elements - one element unless random sampling is enabled. The
+      # sample returned will be the slowest sample among those
+      # available during this harvest
+      def add_samples_to(result, slow_threshold)
+        if @slowest_sample && @slowest_sample.duration >= slow_threshold
+          result << @slowest_sample
+        end
+        result.compact!
+        result = result.sort_by { |x| x.duration }
+        result = result[-1..-1] || []
+        add_random_sample_to(result)
+        result
+      end
+
+      # get the set of collected samples, merging into previous samples,
+      # and clear the collected sample list. Truncates samples to a
+      # specified @segment_limit to save memory and bandwith
+      # transmitting samples to the server.
+      def harvest(previous = [], slow_threshold = 2.0)
+        return [] if disabled
+        result = Array(previous)
+        @samples_lock.synchronize do
+          result = add_samples_to(result, slow_threshold)
+          # clear previous transaction samples
+          @slowest_sample = nil
+          @random_sample = nil
+          @last_sample = nil
+        end
+        # Truncate the samples at 2100 segments. The UI will clamp them at 2000 segments anyway.
+        # This will save us memory and bandwidth.
+        result.each { |sample| sample.truncate(@segment_limit) }
+        result
+      end
+
+      # reset samples without rebooting the web server
+      def reset!
+        @samples = []
+        @last_sample = nil
+        @random_sample = nil
+        @slowest_sample = nil
+      end
+
+      private
+
+      # Checks to see if the transaction sampler is disabled, if
+      # transaction trace recording is disabled by a thread local, or
+      # if execution is untraced - if so it clears the transaction
+      # sample builder from the thread local, otherwise it generates a
+      # new transaction sample builder with the stated time as a
+      # starting point and saves it in the thread local variable
+      def start_builder(time=nil)
+        if disabled || !NewRelic::Agent.is_transaction_traced? || !NewRelic::Agent.is_execution_traced?
+          clear_builder
+        else
+          Thread::current[BUILDER_KEY] ||= TransactionSampleBuilder.new(time)
+        end
+      end
+
+      # The current thread-local transaction sample builder
+      def builder
+        Thread::current[BUILDER_KEY]
+      end
+
+      # Sets the thread local variable storing the transaction sample
+      # builder to nil to clear it
+      def clear_builder
+        Thread::current[BUILDER_KEY] = nil
+      end
+
+    end
+  end
+end