tingyun_rpm 1.1.4.2 → 1.2.0

data/lib/ting_yun/logger.rb

@@ -2,6 +2,7 @@
 # This file is distributed under Ting Yun's license terms.
 
 module TingYun
+  # the basic Logger
   module Logger
 
   end

data/lib/ting_yun/logger/agent_logger.rb

@@ -1,25 +1,25 @@
 # encoding: utf-8
 # This file is distributed under Ting Yun's license terms.
-
 require 'thread'
 require 'logger'
 require 'ting_yun/logger/log_once'
 require 'ting_yun/logger/memory_logger'
 require 'ting_yun/support/hostname'
 require 'ting_yun/logger/null_logger'
-
+require 'ting_yun/logger/create_logger_helper'
 
 module TingYun
   module Logger
     class AgentLogger
-      include LogOnce
+      include ::TingYun::Logger::LogOnce
+      include ::TingYun::Logger::CreateLoggerHelper
 
       def initialize(root = "", override_logger=nil)
         @already_logged_lock = Mutex.new
         clear_already_logged
         create_log(root, override_logger)
-        set_log_level!
-        set_log_format!
+        set_log_level
+        set_log_format
 
         gather_startup_logs
       end
@@ -64,9 +64,6 @@ module TingYun
         end
       end
 
-      def log_formatter=(formatter)
-        @log.formatter = formatter
-      end
 
       private
 
@@ -102,67 +99,14 @@ module TingYun
         nil
       end
 
-      def check_log_file
-        unless File.exist? @file_path
-          begin
-            @log = ::Logger.new(@file_path)
-            set_log_format!
-          rescue => e
-            @log = ::Logger.new(STDOUT)
-            warn("check_log_file:  Failed creating logger for file #{@file_path}, using standard out for logging.", e)
-          end
-        end
-      end
-
-      def create_log(root, override_logger)
-        if !override_logger.nil?
-          @log = override_logger
-        elsif ::TingYun::Agent.config[:'nbs.agent_enabled'] == false
-          create_null_logger
-        else
-          if wants_stdout?
-            @log = ::Logger.new(STDOUT)
-          else
-            create_log_to_file(root)
-          end
-        end
-      end
-
-      def create_log_to_file(root)
-        path = find_or_create_file_path(::TingYun::Agent.config[:agent_log_file_path], root)
-        if path.nil?
-          @log = ::Logger.new(STDOUT)
-          warn("Error creating log directory #{::TingYun::Agent.config[:agent_log_file_path]}, using standard out for logging.")
-        else
-          @file_path = "#{path}/#{::TingYun::Agent.config[:agent_log_file_name]}"
-          begin
-            @log = ::Logger.new(@file_path)
-          rescue => e
-            @log = ::Logger.new(STDOUT)
-            warn("Failed creating logger for file #{@file_path}, using standard out for logging.", e)
-          end
-        end
-      end
-
-      def create_null_logger
-        @log = ::TingYun::Logger::NullLogger.new
-      end
 
       def wants_stdout?
         ::TingYun::Agent.config[:agent_log_file_name].upcase == "STDOUT"
       end
 
-      def find_or_create_file_path(path_setting, root)
-        for abs_path in [File.expand_path(path_setting),
-                         File.expand_path(File.join(root, path_setting))] do
-          if File.directory?(abs_path) || (Dir.mkdir(abs_path) rescue nil)
-            return abs_path[%r{^(.*?)/?$}]
-          end
-        end
-        nil
-      end
 
-      def set_log_level!
+
+      def set_log_level
         @log.level = AgentLogger.log_level_for(::TingYun::Agent.config[:agent_log_level])
       end
 
@@ -178,11 +122,11 @@ module TingYun
         LOG_LEVELS.fetch(level.to_s.downcase, ::Logger::INFO)
       end
 
-      def set_log_format!
-        @hostname = TingYun::Support::Hostname.get
-        @prefix = wants_stdout? ? '** [TingYun]' : ''
+      def set_log_format
+        hostname = TingYun::Support::Hostname.get
+        prefix = wants_stdout? ? '** [TingYun]' : ''
         @log.formatter = Proc.new do |severity, timestamp, progname, msg|
-          "#{@prefix}[#{timestamp.strftime("%m/%d/%y %H:%M:%S %z")} #{@hostname} (#{$$})] #{severity} : #{msg}\n"
+          "#{prefix}[#{timestamp.strftime("%m/%d/%y %H:%M:%S %z")} #{hostname} (#{$$})] #{severity} : #{msg}\n"
         end
       end
 

data/lib/ting_yun/logger/create_logger_helper.rb

@@ -0,0 +1,72 @@
+# encoding: utf-8
+
+module TingYun
+  module Logger
+    module CreateLoggerHelper
+
+
+
+      def create_log(root, override_logger)
+        if !override_logger.nil?
+          @log = override_logger
+        elsif ::TingYun::Agent.config[:'nbs.agent_enabled'] == false
+          create_null_logger
+        else
+          if wants_stdout?
+            @log = ::Logger.new(STDOUT)
+          else
+            create_log_to_file(root)
+          end
+        end
+      end
+
+
+      def check_log_file
+        unless File.exist? @file_path
+          begin
+            @log = ::Logger.new(@file_path)
+            set_log_format
+          rescue => e
+            @log = ::Logger.new(STDOUT)
+            warn("check_log_file:  Failed creating logger for file #{@file_path}, using standard out for logging.", e)
+          end
+        end
+      end
+
+
+
+      def create_log_to_file(root)
+        path = find_or_create_file_path(::TingYun::Agent.config[:agent_log_file_path], root)
+        unless path
+          @log = ::Logger.new(STDOUT)
+          warn("Error creating log directory #{::TingYun::Agent.config[:agent_log_file_path]}, using standard out for logging.")
+        else
+          @file_path = "#{path}/#{::TingYun::Agent.config[:agent_log_file_name]}"
+          begin
+            @log = ::Logger.new(@file_path)
+          rescue => e
+            @log = ::Logger.new(STDOUT)
+            warn("Failed creating logger for file #{@file_path}, using standard out for logging.", e)
+          end
+        end
+      end
+
+      def create_null_logger
+        @log = ::TingYun::Logger::NullLogger.new
+      end
+
+
+      def find_or_create_file_path(path_setting, root)
+        for abs_path in [File.expand_path(path_setting),
+                         File.expand_path(File.join(root, path_setting))] do
+          if File.directory?(abs_path) || (FileUtils.mkdir_p(abs_path) rescue nil)
+            return abs_path[%r{^(.*?)/?$}]
+          end
+        end
+        nil
+      end
+
+
+    end
+  end
+end

data/lib/ting_yun/metrics/metric_data.rb

@@ -1,7 +1,7 @@
 # encoding: utf-8
 # This file is distributed under Ting Yun's license terms.
 
-require 'ting_yun/support/coerce'
+
 
 module TingYun
   module Metrics
@@ -16,8 +16,8 @@ module TingYun
 
       def initialize(metric_spec, stats, metric_id)
         @metric_spec = metric_spec
-        self.stats = stats
-        self.metric_id = metric_id
+        @stats = stats
+        @metric_id = metric_id
       end
 
       def eql?(o)
@@ -45,42 +45,20 @@ module TingYun
         "#<MetricData metric_spec:#{metric_spec.inspect}, stats:#{stats.inspect}, metric_id:#{metric_id.inspect}>"
       end
 
-      include TingYun::Support::Coerce
 
       def to_collector_array(encoder=nil)
-        stat_key = metric_id || stats_has_parent?
-        [ stat_key,metrics(stat_key)]
+        stat_key = metric_id || to_hash
+        [stat_key, metrics(stat_key)]
       end
 
-      def stats_has_parent?
-        hash =  { 'name' => metric_spec.name }
-        hash['calleeId'] = metric_spec.calleeId unless metric_spec.calleeId.nil?
-        hash['calleeName'] = metric_spec.calleeName unless metric_spec.calleeName.nil?
-        unless metric_spec.scope.empty?
-          hash['parent'] = metric_spec.scope
-        end
-
-        return hash
+      def to_hash
+        metric_spec.to_hash
       end
 
       def metrics(stat_key)
-
-        metrics = []
-
-        metrics << int(stats.call_count, stat_key)
-        if stats.max_call_time != 0.0 #apedx
-          metrics << float(stats.total_call_time, stat_key)
-          metrics << float(stats.total_exclusive_time, stat_key)
-          metrics << float(stats.max_call_time, stat_key)
-        end
-
-        if stats.min_call_time !=0.0 #
-          metrics << float(stats.min_call_time, stat_key)
-          metrics << float(stats.sum_of_squares, stat_key)
-        end
-
-        metrics
+        stats.metrics(stat_key)
       end
+
     end
   end
 end

data/lib/ting_yun/metrics/metric_spec.rb

@@ -82,6 +82,17 @@ module TingYun
         return (self.scope || '') <=> (o.scope || '')
       end
 
+      def to_hash
+        hash =  { 'name' => name }
+        hash['calleeId'] = calleeId if calleeId
+        hash['calleeName'] = calleeName if calleeName
+        unless scope.empty?
+          hash['parent'] = scope
+        end
+
+        return hash
+      end
+
 
 
     end

data/lib/ting_yun/metrics/stats.rb

@@ -1,5 +1,6 @@
 # encoding: utf-8
 # This file is distributed under Ting Yun's license terms.
+require 'ting_yun/support/coerce'
 
 module TingYun
   module Metrics
@@ -78,6 +79,10 @@ module TingYun
         }.to_json(*_)
       end
 
+
+
+
+
       def record(value=nil, aux=nil, &blk)
         if blk
           yield self
@@ -108,7 +113,6 @@ module TingYun
       end
 
 
-
       # record a single data point into the statistical gatherer.  The gatherer
       # will aggregate all data points collected over a specified period and upload
       # its data to the TingYun server
@@ -142,6 +146,25 @@ module TingYun
             )
       end
 
+      include TingYun::Support::Coerce
+
+      def metrics(stat_key)
+        metrics = []
+
+        metrics << int(call_count, stat_key)
+        if max_call_time != 0.0 #apedx
+          metrics << float(total_call_time, stat_key)
+          metrics << float(total_exclusive_time, stat_key)
+          metrics << float(max_call_time, stat_key)
+        end
+
+        if min_call_time !=0.0 #
+          metrics << float(min_call_time, stat_key)
+          metrics << float(sum_of_squares, stat_key)
+        end
+
+        metrics
+      end
 
       protected
 

data/lib/ting_yun/middleware/agent_middleware.rb

@@ -0,0 +1,28 @@
+# encoding: utf-8
+require 'ting_yun/instrumentation/middleware_proxy'
+
+module TingYun
+  class AgentMiddleware
+
+    include TingYun::Instrumentation::MiddlewareTracing
+
+    attr_reader :transaction_options, :category, :target
+
+    def initialize(app)
+      @app = app
+      @category = :middleware
+      @target   = self
+      @transaction_options = {
+          :transaction_name => build_transaction_name
+      }
+    end
+
+    def build_transaction_name
+      prefix = ::TingYun::Instrumentation::Support::TransactionNamer.prefix_for_category(nil, @category)
+      "#{prefix}#{self.class.name}/call"
+    end
+
+
+
+  end
+end

data/lib/ting_yun/middleware/browser_monitoring.rb

@@ -0,0 +1,111 @@
+# encoding: utf-8
+require 'ting_yun/middleware/agent_middleware'
+require 'ting_yun/instrumentation/support/javascript_instrumentor'
+
+
+module TingYun
+  class BrowserMonitoring < AgentMiddleware
+
+    CONTENT_TYPE        = 'Content-Type'.freeze
+    TEXT_HTML           = 'text/html'.freeze
+    CONTENT_DISPOSITION = 'Content-Disposition'.freeze
+    ATTACHMENT          = 'attachment'.freeze
+
+    SCAN_LIMIT          = 64_000
+
+    TITLE_END           = '</title>'.freeze
+    TITLE_END_CAPITAL   = '</TITLE>'.freeze
+    HEAD_END            = '<head>'.freeze
+    HEAD_END_CAPITAL    = '<HEAD>'.freeze
+
+    GT                  = '>'.freeze
+
+
+
+
+    def traced_call(env)
+      result = @app.call(env)   # [status, headers, response]
+
+      js_to_inject = TingYun::Instrumentation::Support::JavascriptInstrument.browser_timing_header
+
+      if (js_to_inject != '') && should_instrument?(env, result[0], result[1])
+        response_string = auto_instrument_source(result[2], js_to_inject)
+
+        env[ALREADY_INSTRUMENTED_KEY] = true
+        if response_string
+          response = Rack::Response.new(response_string, result[0], result[1])
+          response.finish
+        else
+          result
+        end
+      else
+        result
+      end
+    end
+
+    ALREADY_INSTRUMENTED_KEY = "tingyun.browser_monitoring_already_instrumented"
+
+    def should_instrument?(env, status, headers)
+      status == 200 &&
+          is_ajax?(env) &&
+          !env[ALREADY_INSTRUMENTED_KEY] &&
+          is_html?(headers) &&
+          !is_attachment?(headers)
+    end
+
+    def is_html?(headers)
+      headers[CONTENT_TYPE] && headers[CONTENT_TYPE].include?(TEXT_HTML)
+    end
+
+    def is_ajax?(env)
+      env["HTTP_X_REQUESTED_WITH"].nil?
+    end
+
+    def is_attachment?(headers)
+      headers[CONTENT_DISPOSITION] && headers[CONTENT_DISPOSITION].include?(ATTACHMENT)
+    end
+
+    def auto_instrument_source(response, js_to_inject)
+      source = gather_source(response)
+      close_old_response(response)
+      return nil unless source
+
+      beginning_of_source = source[0..SCAN_LIMIT]
+      insertion_index = find_tag_end(beginning_of_source)
+
+      if insertion_index
+        source = source[0...insertion_index] <<
+            js_to_inject <<
+            source[insertion_index..-1]
+      else
+        TingYun::Agent.logger.debug "Skipping RUM instrumentation. Could not properly determine location to inject script."
+      end
+
+      source
+    rescue => e
+      TingYun::Agent.logger.debug "Skipping RUM instrumentation on exception.", e
+      nil
+    end
+
+    def gather_source(response)
+      source = nil
+      response.each {|fragment| source ? (source << fragment.to_s) : (source = fragment.to_s)}
+      source
+    end
+
+    def close_old_response(response)
+      if response.respond_to?(:close)
+        response.close
+      end
+    end
+
+    def find_tag_end(beginning_of_source)
+      tag_end = beginning_of_source.index(TITLE_END) ||
+          beginning_of_source.index(HEAD_END) ||
+          beginning_of_source.index(TITLE_END_CAPITAL) ||
+          beginning_of_source.index(HEAD_END_CAPITAL)
+
+      beginning_of_source.index(GT, tag_end) + 1 if tag_end
+    end
+  end
+end