appoptics_apm-zearn 4.13.1

data/lib/appoptics_apm/support/transaction_settings.rb

@@ -0,0 +1,219 @@
+# Copyright (c) 2018 SolarWinds, LLC.
+# All rights reserved.
+#
+
+AO_TRACING_ENABLED = 1
+AO_TRACING_DISABLED = 0
+AO_TRACING_UNSET = -1
+
+AO_TRACING_DECISIONS_OK = 0
+
+OBOE_SETTINGS_UNSET = -1
+
+module AppOpticsAPM
+  ##
+  # This module helps with setting up the transaction filters and applying them
+  #
+  class TransactionSettings
+
+    attr_accessor :do_sample, :do_metrics
+    attr_reader   :auth_msg, :do_propagate, :status_msg, :type, :source, :rate, :xtrace
+                  #, :status
+
+    def initialize(url = '', xtrace = nil, options = nil)
+      @do_metrics = false
+      @do_sample = false
+      @do_propagate = true
+      @xtrace = xtrace
+      tracing_mode = AO_TRACING_ENABLED
+
+      if AppOpticsAPM::Context.isValid
+        @do_sample = AppOpticsAPM.tracing?
+        return
+      end
+
+      if url && asset?(url)
+        @do_propagate = false
+        return
+      end
+
+      if tracing_mode_disabled? && !tracing_enabled?(url) ||
+        tracing_disabled?(url)
+
+        tracing_mode = AO_TRACING_DISABLED
+      end
+
+      args = [@xtrace]
+      args << tracing_mode
+      args << (AppOpticsAPM::Config[:sample_rate] || OBOE_SETTINGS_UNSET)
+
+      if options && (options.options || options.signature)
+        args << (options.trigger_trace ? 1 : 0)
+        args << (trigger_tracing_mode_disabled? ? 0 : 1)
+        args << options.options
+        args << options.signature
+        args << options.timestamp
+      end
+
+      metrics, sample, @rate, @source, @bucket_rate, @bucket_cap, @type, @auth, @status_msg, @auth_msg, @status =
+        AppOpticsAPM::Context.getDecisions(*args)
+
+      if @status > AO_TRACING_DECISIONS_OK
+        AppOpticsAPM.logger.warn "[appoptics-apm/sample] Problem getting the sampling decisions: #{@status_msg} code: #{@status}"
+      end
+
+      @do_metrics = metrics > 0
+      @do_sample = sample > 0
+    end
+
+    def to_s
+      "do_propagate: #{do_propagate}, do_sample: #{do_sample}, do_metrics: #{do_metrics} rate: #{rate}, source: #{source}"
+    end
+
+    def add_kvs(kvs)
+      kvs[:SampleRate] = @rate
+      kvs[:SampleSource] = @source
+    end
+
+    def triggered_trace?
+      @type == 1
+    end
+
+    def auth_ok?
+      # @auth is undefined if initialize is called with an existing context
+      !@auth || @auth < 1
+    end
+
+    private
+
+    ##
+    # check the config setting for :tracing_mode
+    def tracing_mode_disabled?
+      AppOpticsAPM::Config[:tracing_mode] &&
+        [:disabled, :never].include?(AppOpticsAPM::Config[:tracing_mode])
+    end
+
+    ##
+    # tracing_enabled?
+    #
+    # Given a path, this method determines whether it matches any of the
+    # regexps to exclude it from metrics and traces
+    #
+    def tracing_enabled?(url)
+      return false unless AppOpticsAPM::Config[:url_enabled_regexps].is_a? Array
+      # once we only support Ruby >= 2.4.0 use `match?` instead of `=~`
+      return AppOpticsAPM::Config[:url_enabled_regexps].any? { |regex| regex =~ url }
+    rescue => e
+      AppOpticsAPM.logger.warn "[AppOpticsAPM/filter] Could not apply :enabled filter to path. #{e.inspect}"
+      true
+    end
+
+    ##
+    # tracing_disabled?
+    #
+    # Given a path, this method determines whether it matches any of the
+    # regexps to exclude it from metrics and traces
+    #
+    def tracing_disabled?(url)
+      return false unless AppOpticsAPM::Config[:url_disabled_regexps].is_a? Array
+      # once we only support Ruby >= 2.4.0 use `match?` instead of `=~`
+      return AppOpticsAPM::Config[:url_disabled_regexps].any? { |regex| regex =~ url }
+    rescue => e
+      AppOpticsAPM.logger.warn "[AppOpticsAPM/filter] Could not apply :disabled filter to path. #{e.inspect}"
+      false
+    end
+
+    def trigger_tracing_mode_disabled?
+      AppOpticsAPM::Config[:trigger_tracing_mode] &&
+        AppOpticsAPM::Config[:trigger_tracing_mode] == :disabled
+    end
+
+    ##
+    # asset?
+    #
+    # Given a path, this method determines whether it is a static asset
+    #
+    def asset?(path)
+      return false unless AppOpticsAPM::Config[:dnt_compiled]
+      # once we only support Ruby >= 2.4.0 use `match?` instead of `=~`
+      return AppOpticsAPM::Config[:dnt_compiled] =~ path
+    rescue => e
+      AppOpticsAPM.logger.warn "[AppOpticsAPM/filter] Could not apply do-not-trace filter to path. #{e.inspect}"
+      false
+    end
+
+    public
+
+    class << self
+
+      def asset?(path)
+        return false unless AppOpticsAPM::Config[:dnt_compiled]
+        # once we only support Ruby >= 2.4.0 use `match?` instead of `=~`
+        return AppOpticsAPM::Config[:dnt_compiled] =~ path
+      rescue => e
+        AppOpticsAPM.logger.warn "[AppOpticsAPM/filter] Could not apply do-not-trace filter to path. #{e.inspect}"
+        false
+      end
+
+
+      def compile_url_settings(settings)
+        if !settings.is_a?(Array) || settings.empty?
+          reset_url_regexps
+          return
+        end
+
+        # `tracing: disabled` is the default
+        disabled = settings.select { |v| !v.has_key?(:tracing) || v[:tracing] == :disabled }
+        enabled = settings.select { |v| v[:tracing] == :enabled }
+
+        AppOpticsAPM::Config[:url_enabled_regexps] = compile_regexp(enabled)
+        AppOpticsAPM::Config[:url_disabled_regexps] = compile_regexp(disabled)
+      end
+
+      def compile_regexp(settings)
+        regexp_regexp     = compile_url_settings_regexp(settings)
+        extensions_regexp = compile_url_settings_extensions(settings)
+
+        regexps = [regexp_regexp, extensions_regexp].flatten.compact
+
+        regexps.empty? ? nil : regexps
+      end
+
+      def compile_url_settings_regexp(value)
+        regexps = value.select do |v|
+          v.key?(:regexp) &&
+            !(v[:regexp].is_a?(String) && v[:regexp].empty?) &&
+            !(v[:regexp].is_a?(Regexp) && v[:regexp].inspect == '//')
+        end
+
+        regexps.map! do |v|
+          begin
+            v[:regexp].is_a?(String) ? Regexp.new(v[:regexp], v[:opts]) : Regexp.new(v[:regexp])
+          rescue
+            AppOpticsAPM.logger.warn "[appoptics_apm/config] Problem compiling transaction_settings item #{v}, will ignore."
+            nil
+          end
+        end
+        regexps.keep_if { |v| !v.nil?}
+        regexps.empty? ? nil : regexps
+      end
+
+      def compile_url_settings_extensions(value)
+        extensions = value.select do |v|
+          v.key?(:extensions) &&
+            v[:extensions].is_a?(Array) &&
+            !v[:extensions].empty?
+        end
+        extensions = extensions.map { |v| v[:extensions] }.flatten
+        extensions.keep_if { |v| v.is_a?(String)}
+
+        extensions.empty? ? nil : Regexp.new("(#{Regexp.union(extensions).source})(\\?.+){0,1}$")
+      end
+
+      def reset_url_regexps
+        AppOpticsAPM::Config[:url_enabled_regexps] = nil
+        AppOpticsAPM::Config[:url_disabled_regexps] = nil
+      end
+    end
+  end
+end

data/lib/appoptics_apm/support/x_trace_options.rb

@@ -0,0 +1,110 @@
+# Copyright (c) 2019 SolarWinds, LLC.
+# All rights reserved.
+
+
+module AppOpticsAPM
+  class XTraceOptions
+
+    attr_reader :options, :signature, :trigger_trace, :timestamp
+    attr_reader :pd_keys, :custom_kvs, :ignored # used in tests
+    ##
+    # Params:
+    # +options+ : An X-Trace-Options @options string
+    # +signature+ : hmac signature to pass on for verification
+    #
+    # populates:
+    # - @force_trace (true|false)
+    # - @app_id (as given by Pingdom)
+    # - @probe_id (as given by Pingdom)
+    # - @loc (2 characters given by Pingdom)
+    # - @custom_kvs (hash)
+    # - @ignored (array)
+    #
+    # split it up by ';' separator
+    # kv assignment by '='
+    # currently valid keys:
+    # - force_trace (valid: 0,1) unless we use just a kv
+    # - application_id (format defined by pingdom (no validation))
+    # - probe_id
+    # - custom_* (';=' not allowed in key), value (validate max. length)
+    # - ts (unix timestamp)
+    # - other keys will be reported in the response options as ignored
+
+    def initialize(options, signature = nil)
+      @options = options.dup
+      @signature = signature.dup
+      @trigger_trace = false
+      @custom_kvs = {}
+      @pd_keys = nil
+      @ignored = []
+      @timestamp = 0
+
+      options&.split(/;+/)&.each do |val|
+        k = val.split('=', 2)
+
+        next unless k[0] # it can be nil, eg when the header starts with ';'
+
+        k[0]&.strip!
+        case k[0]
+        when 'trigger-trace'
+          if k[1]
+            @ignored << 'trigger-trace'
+          else
+            @trigger_trace = true
+          end
+        when 'pd-keys'
+          if @pd_keys
+            AppOpticsAPM.logger.info "[appoptics_apm/x-trace-options] Duplicate key: #{k[0]}"
+          else
+            @pd_keys = k[1].strip
+          end
+        when /^custom-[^\s]*$/
+          if @custom_kvs[k[0]]
+            AppOpticsAPM.logger.info "[appoptics_apm/x-trace-options] Duplicate key: #{k[0]}"
+          else
+            @custom_kvs[k[0]] = k[1].strip
+          end
+        when 'ts'
+          if @timestamp > 0
+            AppOpticsAPM.logger.info "[appoptics_apm/x-trace-options] Duplicate key: #{k[0]}"
+          else
+            @timestamp = k[1].to_i
+          end
+        else
+          @ignored << k[0]
+        end
+      end
+      unless @ignored.empty?
+        msg = "[appoptics_apm/x-trace-options] Some keys were ignored: #{@ignored.join(',' )}"
+        AppOpticsAPM.logger.info(msg)
+      end
+    end
+
+    def add_kvs(kvs, settings)
+      return unless settings.auth_ok?
+
+      @custom_kvs.each { |k,v| kvs[k] = v } unless @custom_kvs.empty?
+      kvs['PDKeys'] = @pd_keys if @pd_keys
+      kvs['TriggeredTrace'] = true if settings.triggered_trace?
+    end
+
+    def add_response_header(headers, settings)
+      return unless options
+
+      response = []
+      response << "auth=#{settings.auth_msg}" if @signature
+      if settings.auth_ok?
+        if @trigger_trace
+          trigger_msg = settings.xtrace && settings.type == 0 ? 'ignored' : settings.status_msg
+        else
+          trigger_msg = 'not-requested'
+        end
+        response << "trigger-trace=#{trigger_msg}"
+        response << "ignored=#{@ignored.join(',')}" unless @ignored.empty?
+      end
+
+      headers['X-Trace-Options-Response'] = response.join(';')
+    end
+
+  end
+end

data/lib/appoptics_apm/support_report.rb

@@ -0,0 +1,119 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+require 'rbconfig'
+require 'logger'
+
+module AppOpticsAPM
+  ##
+  # This module is used to debug problematic setups and/or environments.
+  # Depending on the environment, output may be to stdout or the framework
+  # log file (e.g. log/production.log)
+
+  ##
+  # yesno
+  #
+  # Utility method to translate value/nil to "yes"/"no" strings
+  def self.yesno(x)
+    x ? 'yes' : 'no'
+  end
+
+  # rubocop:disable Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity, Metrics/AbcSize
+  def self.support_report
+    @logger_level = AppOpticsAPM.logger.level
+    AppOpticsAPM.logger.level = ::Logger::DEBUG
+
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn '* BEGIN AppOpticsAPM Support Report'
+    AppOpticsAPM.logger.warn '*   Please email the output of this report to technicalsupport@solarwinds.com'
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn "Ruby: #{RUBY_DESCRIPTION}"
+    AppOpticsAPM.logger.warn "$0: #{$0}"
+    AppOpticsAPM.logger.warn "$1: #{$1}" unless $1.nil?
+    AppOpticsAPM.logger.warn "$2: #{$2}" unless $2.nil?
+    AppOpticsAPM.logger.warn "$3: #{$3}" unless $3.nil?
+    AppOpticsAPM.logger.warn "$4: #{$4}" unless $4.nil?
+    AppOpticsAPM.logger.warn "AppOpticsAPM.loaded == #{AppOpticsAPM.loaded}"
+
+    using_jruby = defined?(JRUBY_VERSION)
+    AppOpticsAPM.logger.warn "Using JRuby?: #{yesno(using_jruby)}"
+    if using_jruby
+      AppOpticsAPM.logger.warn "Joboe Agent Status: #{Java::ComTracelyticsAgent::Agent.getStatus}"
+    end
+
+    on_heroku = AppOpticsAPM.heroku?
+    AppOpticsAPM.logger.warn "On Heroku?: #{yesno(on_heroku)}"
+    if on_heroku
+      AppOpticsAPM.logger.warn "APPOPTICS_URL: #{ENV['APPOPTICS_URL']}"
+    end
+
+    AppOpticsAPM.logger.warn "AppOpticsAPM::Ruby defined?: #{yesno(defined?(AppOpticsAPM::Ruby))}"
+    AppOpticsAPM.logger.warn "AppOpticsAPM.reporter: #{AppOpticsAPM.reporter}"
+
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn '* Frameworks'
+    AppOpticsAPM.logger.warn '********************************************************'
+
+    using_rails = defined?(::Rails)
+    AppOpticsAPM.logger.warn "Using Rails?: #{yesno(using_rails)}"
+    if using_rails
+      AppOpticsAPM.logger.warn "AppOpticsAPM::Rails loaded?: #{yesno(defined?(AppOpticsAPM::Rails))}"
+      if defined?(AppOpticsAPM::Rack)
+        AppOpticsAPM.logger.warn "AppOpticsAPM::Rack middleware loaded?: #{yesno(::Rails.configuration.middleware.include? AppOpticsAPM::Rack)}"
+      end
+    end
+
+    using_sinatra = defined?(::Sinatra)
+    AppOpticsAPM.logger.warn "Using Sinatra?: #{yesno(using_sinatra)}"
+
+    using_padrino = defined?(::Padrino)
+    AppOpticsAPM.logger.warn "Using Padrino?: #{yesno(using_padrino)}"
+
+    using_grape = defined?(::Grape)
+    AppOpticsAPM.logger.warn "Using Grape?: #{yesno(using_grape)}"
+
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn '* ActiveRecord Adapter'
+    AppOpticsAPM.logger.warn '********************************************************'
+    if defined?(::ActiveRecord)
+      if defined?(::ActiveRecord::Base.connection.adapter_name)
+        AppOpticsAPM.logger.warn "ActiveRecord adapter: #{::ActiveRecord::Base.connection.adapter_name}"
+      end
+    else
+      AppOpticsAPM.logger.warn 'No ActiveRecord'
+    end
+
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn '* AppOpticsAPM::Config Values'
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM::Config.print_config
+
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn '* OS, Platform + Env'
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn "host_os: " + RbConfig::CONFIG['host_os']
+    AppOpticsAPM.logger.warn "sitearch: " + RbConfig::CONFIG['sitearch']
+    AppOpticsAPM.logger.warn "arch: " + RbConfig::CONFIG['arch']
+    AppOpticsAPM.logger.warn RUBY_PLATFORM
+    AppOpticsAPM.logger.warn "RACK_ENV: #{ENV['RACK_ENV']}"
+    AppOpticsAPM.logger.warn "RAILS_ENV: #{ENV['RAILS_ENV']}" if using_rails
+
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn '* Raw __Init KVs'
+    AppOpticsAPM.logger.warn '********************************************************'
+    platform_info = AppOpticsAPM::Util.build_init_report
+    platform_info.each { |k,v|
+      AppOpticsAPM.logger.warn "#{k}: #{v}"
+    }
+
+    AppOpticsAPM.logger.warn '********************************************************'
+    AppOpticsAPM.logger.warn '* END AppOpticsAPM Support Report'
+    AppOpticsAPM.logger.warn '*   Support Email: technicalsupport@solarwinds.com'
+    AppOpticsAPM.logger.warn '*   Github: https://github.com/librato/ruby-appoptics'
+    AppOpticsAPM.logger.warn '********************************************************'
+
+    AppOpticsAPM.logger.level = @logger_level
+    nil
+  end
+  # rubocop:enable Metrics/CyclomaticComplexity, Metrics/PerceivedComplexity, Metrics/AbcSize
+end

data/lib/appoptics_apm/test.rb

@@ -0,0 +1,95 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+module AppOpticsAPM
+  module Test
+    class << self
+      ##
+      # load_extras
+      #
+      # This method simply loads all the extras needed to run
+      # tests such as models, jobs etc...
+      #
+      def load_extras
+        # If we're using the libraries gemfile (with sidekiq and resque)
+        if AppOpticsAPM::Test.gemfile?(:libraries)
+          # Load all of the test workers
+          pattern = File.join(File.dirname(__FILE__), '../../test/jobs/**/', '*.rb')
+          Dir.glob(pattern) do |f|
+            AppOpticsAPM.logger.debug "[appoptics_apm/test] Loading test job file: #{File.basename(f)}"
+            require f
+          end
+        end
+      end
+
+      ##
+      # gemfile?
+      #
+      # Method used to determine under which gemfile we're running.
+      # Pass <tt>name</tt> as the gemfile name only (without the
+      # .gemfile extension)
+      #
+      # returns true or fase depending on result
+      #
+      def gemfile?(name)
+        File.basename(ENV['BUNDLE_GEMFILE']) == (name.to_s + '.gemfile')
+      end
+
+      ##
+      # gemfile
+      #
+      # Used to determine under which gemfile we are running.  This
+      # method will return the name of the active gemfile
+      #
+      def gemfile
+        File.basename(ENV['BUNDLE_GEMFILE']).split('.').first
+      end
+
+      ##
+      # set_postgresql_env
+      #
+      # Used to set the postgresql specific DATABASE_URL env based
+      # on various conditions
+      def set_postgresql_env
+        if ENV.key?('TRAVIS_PSQL_PASS')
+          ENV['DATABASE_URL'] = "postgresql://postgres:#{ENV['TRAVIS_PSQL_PASS']}@127.0.0.1:5432/travis_ci_test"
+        elsif ENV.key?('POSTGRES_USER')
+          port = ENV.key?('POSTGRES_PORT') ? ENV['POSTGRES_PORT'] : 5432
+          ENV['DATABASE_URL'] = "postgresql://#{ENV['POSTGRES_PASSWORD']}:#{ENV['POSTGRES_USER']}@#{ENV['POSTGRES_HOST']}:#{port}/travis_ci_test"
+        else
+          ENV['DATABASE_URL'] = 'postgresql://postgres@127.0.0.1:5432/travis_ci_test'
+        end
+      end
+      ##
+      # set_mysql_env
+      #
+      # Used to set the mysql specific DATABASE_URL env based
+      # on various conditions
+      def set_mysql_env
+        if ENV.key?('TRAVIS_MYSQL_PASS')
+          ENV['DATABASE_URL'] = "mysql://root:#{ENV['TRAVIS_MYSQL_PASS']}@127.0.0.1:3306/travis_ci_test"
+        elsif ENV.key?('DOCKER_MYSQL_PASS')
+          port = ENV.key?('MYSQL_PORT') ? ENV['MYSQL_PORT'] : 3306
+          ENV['DATABASE_URL'] = "mysql://root:#{ENV['DOCKER_MYSQL_PASS']}@#{ENV['MYSQL_HOST']}:#{port}/travis_ci_test"
+        else
+          ENV['DATABASE_URL'] = 'mysql://root@127.0.0.1:3306/travis_ci_test'
+        end
+      end
+
+      ##
+      # set_mysql2_env
+      #
+      # Used to set the mysql specific DATABASE_URL env based
+      # on various conditions
+      def set_mysql2_env
+        if ENV.key?('TRAVIS_MYSQL_PASS')
+          ENV['DATABASE_URL'] = "mysql2://root:#{ENV['TRAVIS_MYSQL_PASS']}@127.0.0.1:3306/travis_ci_test"
+        elsif ENV.key?('DOCKER_MYSQL_PASS')
+          ENV['DATABASE_URL'] = "mysql2://root:#{ENV['DOCKER_MYSQL_PASS']}@#{ENV['MYSQL_HOST']}:3306/travis_ci_test"
+        else
+          ENV['DATABASE_URL'] = 'mysql2://root@127.0.0.1:3306/travis_ci_test'
+        end
+      end
+    end
+  end
+end

data/lib/appoptics_apm/thread_local.rb

@@ -0,0 +1,26 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+module AppOpticsAPM
+  ##
+  # Provides thread local storage for AppOpticsAPM.
+  #
+  # Example usage:
+  # module AppOpticsAPMBase
+  #   extend AppOpticsAPM::ThreadLocal
+  #   thread_local :layer_op
+  # end
+  module ThreadLocal
+    def thread_local(name)
+      key = "__#{self}_#{name}__".intern
+
+      define_method(name) do
+        Thread.current[key]
+      end
+
+      define_method(name.to_s + '=') do |value|
+        Thread.current[key] = value
+      end
+    end
+  end
+end