oneapm_rpm 1.3.7 → 1.4.0

data/lib/one_apm/support/http_clients/thrift_tracer.rb

@@ -0,0 +1,64 @@
+module OneApm
+  module Support
+    module HTTPClients
+      module ThriftTracer
+        def self.included clazz
+          clazz.class_eval do 
+
+            def is_cross_app=(is_cross_app)
+              @is_cross_app = is_cross_app
+            end
+
+            def is_cross_app?
+              @is_cross_app
+            end
+
+            def write_field_stop_with_oneapm
+              is_write_fielded =  false
+              begin
+                state = OneApm::TransactionState.tl_get
+                message = OneApm::Agent::CrossAppTracingMessage.request_data(state)
+                if message
+                  write_field_begin('oneapm_message', ::Thrift::Types::STRING, -32768)
+                  is_write_fielded = true
+                  write_string("####{message}###")
+                end
+              rescue => e
+                 OneApm::Manager.logger.error "Thrift write_field_stop error: #{e}"
+              ensure
+                write_field_end if is_write_fielded
+              end
+              write_field_stop_without_oneapm
+            end
+            alias :write_field_stop_without_oneapm :write_field_stop
+            alias :write_field_stop :write_field_stop_with_oneapm
+            
+            def read_field_begin_with_oneapm
+              name, type, id = read_field_begin_without_oneapm
+              begin
+                is_cross_app = true if type == ::Thrift::Types::STRING && id == -32768
+              rescue => e
+                 OneApm::Manager.logger.error "Thrift read_field_begin error: #{e}"
+              end
+              [name, type, id]
+             end
+            alias :read_field_begin_without_oneapm :read_field_begin
+            alias :read_field_begin :read_field_begin_with_oneapm
+
+            def read_string_with_oneapm
+              string = read_string_without_oneapm
+              if string && is_cross_app?
+                if /###(?<message>.*)###/ =~ string
+                  OneApm::Agent::CrossAppTracingMessage.process_request(message)
+                end
+              end
+              string
+            end
+            alias :read_string_without_oneapm :read_string
+            alias :read_string :read_string_with_oneapm
+          end
+        end
+      end
+    end
+  end
+end

data/lib/one_apm/support/http_clients/typhoeus_wrappers.rb

@@ -47,7 +47,11 @@ module OneApm
         end
 
         def host
-          self['host'] || self['Host'] || @uri.host
+          @uri.host
+        end
+
+        def port
+          @uri.port
         end
 
         def method
@@ -67,6 +71,22 @@ module OneApm
         def uri
           @uri
         end
+
+        def query
+          @uri.query.nil?? {} : CGI.parse(@uri.query)
+        end
+
+        def body
+          @request.options[:body]
+        end
+
+        def post_params
+          body.nil?? {} :  CGI.parse(body)
+        end
+
+        def params
+          post_params.merge(query)
+        end
       end
     end
   end

data/lib/one_apm/support/http_clients/uri_util.rb

@@ -29,6 +29,7 @@ module OneApm
             address.normalize!
             URI.parse(address.to_s)
           else
+            'http://'
             URI.parse(url)
           end
         end

data/lib/one_apm/support/ip.rb

@@ -0,0 +1,15 @@
+# encoding: utf-8
+module OneApm
+  module Agent
+    module IP
+
+      def self.get
+        ipinfo = `/sbin/ifconfig | grep 'inet ' | grep  -v '127.0.0.1' | awk '{print $2}'`.split("\n")[0].sub(/addr:|地址:|\s+/, '') 
+        ipinfo.split("\n")[0].sub(/addr:|地址:|\s+/, '') 
+      rescue => e
+        nil
+      end
+
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine.rb

@@ -0,0 +1,39 @@
+# encoding: utf-8
+require 'one_apm/support/rename_rules_engine/external_rename_rule'
+module OneApm
+  module Support
+    class RenameRulesEngine
+
+      class << self
+
+        def create_or_update_external_rules(connect_response)
+          unpacked_rules = unpack_rules(connect_response)
+          if unpacked_rules.nil?
+            nil
+          elsif current_external_rename_rules.nil?
+            OneApm::Manager.agent.external_rename_rules = ExternalRenameRule.new(unpacked_rules)
+          elsif need_update?(unpacked_rules)
+            OneApm::Manager.agent.external_rename_rules = ExternalRenameRule.new(unpacked_rules)
+            OneApm::Manager.logger.debug "Updated RenameRules from version : #{unpacked_rules['hash_value']}"
+          end
+        end
+
+        # There are two kinds of response
+        # First one from connect data, another one from event_loop
+        def unpack_rules(connect_response)
+          return nil if connect_response.nil?
+          connect_response['btm_value'].nil? ? connect_response : connect_response['btm_value']
+        end
+
+        def need_update?(unpacked_rules)
+          current_external_rename_rules.hash_value.to_i < unpacked_rules['hash_value'].to_i
+        end
+
+        def current_external_rename_rules
+          OneApm::Manager.agent.external_rename_rules
+        end
+
+      end
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine/external_rename_rule.rb

@@ -0,0 +1,84 @@
+# encoding: utf-8
+require 'one_apm/support/rename_rules_engine/match_expression/engine'
+require 'one_apm/support/rename_rules_engine/rename/engine'
+
+module OneApm
+  module Support
+    class RenameRulesEngine
+      class ExternalRenameRule
+        OA_RENAME_SEPARATOR = '/'.freeze
+
+        attr_accessor :hash_value, :rules
+
+        def initialize(hash={})
+          @hash_value = hash['hash_value']
+          rules  = hash['rules'] || []
+          rules  = rules.select { |r| r['type'] == "remote" }
+          rules  = rules.sort_by { |r| r['priority'] }
+          rules  = rules.map { |r| Rule.new(r) }
+          rules  = rules.reverse
+          @rules = rules
+        end
+
+        def rename(request)
+          selected_rule = select_rule(request)
+          OneApm::Manager.logger.debug "request", request.inspect
+          OneApm::Manager.logger.debug "selected_rule", selected_rule.inspect
+          res = if selected_rule.nil?
+                  default_rename(request)
+                else
+                  selected_rule.apply(request)
+                end
+          "External/HTTP/#{request.host}:#{request.port}/#{res}"
+        end
+
+        def default_rename(request)
+          path = request.uri.path.split(OA_RENAME_SEPARATOR)
+          path.delete('')
+          if path.size <= 1
+            "#{path.first}/"
+          else
+            "#{path[0..1].join(OA_RENAME_SEPARATOR)}"
+          end
+        end
+
+        def select_rule(request)
+          OneApm::Manager.logger.debug "@rules", @rules.inspect
+          @rules.select { |r| r.enable?&&r.should_rename?(request) }.first
+        end
+
+      end
+
+      class Rule
+        OA_RENAME_SEPARATOR = '/'.freeze
+        attr_accessor :replacement, :state, :ignore, :priority, :match_expressions, :rename_rules
+
+        def initialize(hash)
+          @replacement = hash['replacement']
+          @state = hash['state'].to_i
+          @ignore = hash['ignore']
+          @priority = hash['priority'].to_i
+          @match_expressions = hash['match_expression'].map { |match_expression| MatchExpression::Engine.create_match_expression(match_expression) }
+          @rename_rules = hash['rename'].map { |rename| Rename::Engine.create_rename_rules(rename) }
+        end
+
+        def should_rename?(request)
+          match_expressions.map do |match_expression|
+            match_expression.match?(request)
+          end.all?
+        end
+
+        def enable?
+          !ignore && state == 0
+        end
+
+        def apply(request)
+          rename_rules.map do |rename_rule|
+            rename_rule.segment(request)
+          end.join(OA_RENAME_SEPARATOR)
+        end
+      end
+
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine/match_expression/base.rb

@@ -0,0 +1,46 @@
+# encoding: utf-8
+
+module OneApm
+  module Support
+    class RenameRulesEngine
+      module MatchExpression
+
+        class Base
+          attr_accessor :is_not, :match_value, :operate, :state
+
+          def initialize(expression)
+            @is_not = expression[:is_not]
+            @match_value = expression[:match_value]
+            @operate = expression[:operate]
+            @state = expression[:state]==0 ? true : false
+          end
+
+          def type_match?(type, expression, sample)
+            case type
+              when 'equals'
+                sample.eql?(expression)
+              when 'endwith'
+                sample.end_with?(expression)
+              when 'startswith'
+                sample.start_with?(expression)
+              when 'contains'
+                sample.include?(expression)
+              when 'in'
+                expression = expression.split(',')
+                expression.include?(sample)
+              when 'existence', 'isnotempty'
+                !(sample.nil? || sample.to_s.empty?)
+              else
+                false
+            end
+          end
+
+          def match?(request)
+            false
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine/match_expression/engine.rb

@@ -0,0 +1,35 @@
+# encoding: utf-8
+
+
+require 'one_apm/support/rename_rules_engine/match_expression/split_object'
+require 'one_apm/support/rename_rules_engine/match_expression/url'
+require 'one_apm/support/rename_rules_engine/match_expression/method'
+require 'one_apm/support/dotted_hash'
+
+module OneApm
+  module Support
+    class RenameRulesEngine
+      module MatchExpression
+        module Engine
+
+          extend self
+
+          def create_match_expression(expression)
+            expression = DottedHash.new(expression)
+            case expression[:name].downcase
+              when 'splitobject'
+                SplitObject.new(expression)
+              when 'method'
+                Method.new(expression)
+              when 'url'
+                Url.new(expression)
+              else
+                Base.new(expression)
+            end
+          end
+        end
+        
+      end
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine/match_expression/method.rb

@@ -0,0 +1,20 @@
+require 'one_apm/support/rename_rules_engine/match_expression/base'
+
+module OneApm
+  module Support
+    class RenameRulesEngine
+
+        module MatchExpression
+          class Method < Base
+
+            def match?(request)
+              return true if !state
+              match_result = type_match?(operate.downcase,match_value.to_s,request.method.to_s)
+              is_not ? !match_result : match_result
+            end
+          end
+
+      end
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine/match_expression/split_object.rb

@@ -0,0 +1,19 @@
+require 'one_apm/support/rename_rules_engine/match_expression/base'
+module OneApm
+  module Support
+    class RenameRulesEngine
+      module MatchExpression
+        class SplitObject < Base
+
+          def match?(request)
+            return true if match_value.downcase=='all'
+            return true if !state
+            match_result = type_match?(operate.downcase, match_value, "#{request.uri.host}:#{request.port}")
+            is_not ? !match_result : match_result
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine/match_expression/url.rb

@@ -0,0 +1,19 @@
+require 'one_apm/support/rename_rules_engine/match_expression/base'
+
+module OneApm
+  module Support
+    class RenameRulesEngine
+
+      module MatchExpression
+        class Url < Base
+          def match?(request)
+            return true if !state
+            match_result = type_match?(operate.downcase, match_value, request.uri.path)
+            is_not ? (!match_result) : (match_result)
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine/rename/base.rb

@@ -0,0 +1,25 @@
+# encoding: utf-8
+
+module OneApm
+  module Support
+    class RenameRulesEngine
+      module Rename
+        class Base
+
+          attr_accessor :state,:value
+
+          def initialize(rename_hash)
+            @value = rename_hash[:value]
+            @state = rename_hash[:state]==0 ? true : false
+          end
+
+
+          def segment(request)
+            ''
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/one_apm/support/rename_rules_engine/rename/engine.rb

@@ -0,0 +1,36 @@
+# encoding: utf-8
+
+
+require 'one_apm/support/rename_rules_engine/rename/base'
+require 'one_apm/support/rename_rules_engine/rename/header'
+require 'one_apm/support/rename_rules_engine/rename/parameter'
+require 'one_apm/support/rename_rules_engine/rename/segment'
+require 'one_apm/support/dotted_hash'
+
+module OneApm
+  module Support
+    class RenameRulesEngine
+      module Rename
+        module Engine
+
+          extend self
+
+          def create_rename_rules(rename_hash)
+            rename_hash = DottedHash.new(rename_hash)
+            case rename_hash[:type].downcase
+              when 'segments'
+                Segment.new(rename_hash)
+              when 'parameter'
+                Parameter.new(rename_hash)
+              when 'header'
+                Header.new(rename_hash)
+              else
+                Base.new(rename_hash)
+            end
+          end
+        end
+
+      end
+    end
+  end
+end