appoptics_apm-zearn 4.13.1

data/lib/joboe_metal.rb

@@ -0,0 +1,212 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+module Oboe_metal
+  include_package 'com.tracelytics.joboe'
+  java_import 'com.tracelytics.joboe.LayerUtil'
+  java_import 'com.tracelytics.joboe.SettingsReader'
+  java_import 'com.tracelytics.joboe.Context'
+  java_import 'com.tracelytics.joboe.Event'
+  java_import 'com.tracelytics.agent.Agent'
+
+  class Context
+    class << self
+      def toString
+        getMetadata.toHexString
+      end
+
+      def fromString(xtrace)
+        Context.setMetadata(xtrace)
+      end
+
+      def clear
+        clearMetadata
+      end
+
+      def get
+        getMetadata
+      end
+    end
+  end
+
+  module EventUtil
+    def self.metadataString(evt)
+      evt.getMetadata.toHexString
+    end
+  end
+
+  def UdpReporter
+    Java::ComTracelyticsJoboe
+  end
+
+  module Metadata
+    Java::ComTracelyticsJoboeMetaData
+  end
+
+  module Reporter
+    class << self
+      ##
+      # start
+      #
+      # Start the AppOpticsAPM Reporter
+      #
+      def start
+        return unless AppOpticsAPM.loaded
+
+        if ENV.key?('APPOPTICS_GEM_TEST')
+          AppOpticsAPM.reporter = Java::ComTracelyticsJoboe::ReporterFactory.getInstance.buildTestReporter(false)
+        else
+          AppOpticsAPM.reporter = Java::ComTracelyticsJoboe::ReporterFactory.getInstance.buildUdpReporter
+        end
+
+        begin
+          # Import the tracing mode and sample rate settings
+          # from the Java agent (user configured in
+          # /usr/local/tracelytics/javaagent.json when under JRuby)
+          cfg = LayerUtil.getLocalSampleRate(nil, nil)
+
+          if cfg.hasSampleStartFlag
+            AppOpticsAPM::Config.tracing_mode = :enabled
+          else
+            AppOpticsAPM::Config.tracing_mode = :disabled
+          end
+
+          AppOpticsAPM.sample_rate = cfg.getSampleRate
+          AppOpticsAPM::Config.sample_rate = cfg.sampleRate
+          AppOpticsAPM::Config.sample_source = cfg.sampleRateSourceValue
+        rescue => e
+          AppOpticsAPM.logger.debug "[appoptics_apm/debug] Couldn't retrieve/acces joboe sampleRateCfg"
+          AppOpticsAPM.logger.debug "[appoptics_apm/debug] #{e.message}"
+        end
+
+        # Only report __Init from here if we are not instrumenting a framework.
+        # Otherwise, frameworks will handle reporting __Init after full initialization
+        unless defined?(::Rails) || defined?(::Sinatra) || defined?(::Padrino) || defined?(::Grape)
+          AppOpticsAPM::API.report_init unless ENV.key?('APPOPTICS_GEM_TEST')
+        end
+      end
+
+      ##
+      # restart
+      #
+      # This is a nil method for AppOpticsAPM under Java.  It is maintained only
+      # for compability across interfaces.
+      #
+      def restart
+        AppOpticsAPM.logger.warn "[appoptics_apm/reporter] Reporter.restart isn't supported under JRuby"
+      end
+
+      ##
+      # clear_all_traces
+      #
+      # Truncates the trace output file to zero
+      #
+      def clear_all_traces
+        AppOpticsAPM.reporter.reset if AppOpticsAPM.loaded
+      end
+
+      ##
+      # get_all_traces
+      #
+      # Retrieves all traces written to the trace file
+      #
+      def get_all_traces
+        return [] unless AppOpticsAPM.loaded
+
+        # Joboe TestReporter returns a Java::ComTracelyticsExtEbson::DefaultDocument
+        # document for traces which doesn't correctly support things like has_key? which
+        # raises an unhandled exception on non-existent key (duh).  Here we convert
+        # the Java::ComTracelyticsExtEbson::DefaultDocument doc to a pure array of Ruby
+        # hashes
+        traces = []
+        AppOpticsAPM.reporter.getSentEventsAsBsonDocument.to_a.each do |e|
+          t = {}
+          e.each_pair { |k, v|
+            t[k] = v
+          }
+          traces << t
+        end
+        traces
+      end
+
+      def sendReport(evt)
+        evt.report(AppOpticsAPM.reporter)
+      end
+    end
+  end
+end
+
+module AppOpticsAPM
+  extend AppOpticsAPMBase
+  include Oboe_metal
+
+  class << self
+    def sample?(opts = {})
+      begin
+        # Return false if no-op mode
+        return false unless AppOpticsAPM.loaded
+
+        return true if ENV.key?('APPOPTICS_GEM_TEST')
+
+        # Validation to make Joboe happy.  Assure that we have the KVs and that they
+        # are not empty strings.
+        opts[:xtrace] = nil      if opts[:xtrace].is_a?(String)     && opts[:xtrace].empty?
+
+        opts[:xtrace]     ||= nil
+
+        sr_cfg = Java::ComTracelyticsJoboe::LayerUtil.shouldTraceRequest(APPOPTICS_STR_BLANK, { 'X-Trace' => opts[:xtrace] })
+
+        # Store the returned SampleRateConfig into AppOpticsAPM::Config
+        if sr_cfg
+          begin
+            AppOpticsAPM::Config.sample_rate = sr_cfg.sampleRate
+            AppOpticsAPM::Config.sample_source = sr_cfg.sampleRateSourceValue
+            # If we fail here, we do so quietly.  This was we don't spam logs
+            # on every request
+          end
+        else
+          AppOpticsAPM.sample_rate = -1
+          AppOpticsAPM.sample_source = -1
+        end
+
+        sr_cfg ? true : false
+      rescue => e
+        AppOpticsAPM.logger.debug "[appoptics_apm/debug] #{e.message}"
+        false
+      end
+    end
+
+    def set_tracing_mode(_mode)
+      AppOpticsAPM.logger.warn '[appoptics_apm/joboe] When using JRuby set the tracing mode in /usr/local/tracelytics/javaagent.json instead'
+    end
+
+    def set_sample_rate(_rate)
+      # N/A
+    end
+  end
+end
+
+# Assure that the Joboe Java Agent was loaded via premain
+case Java::ComTracelyticsAgent::Agent.getStatus
+  when Java::ComTracelyticsAgent::Agent::AgentStatus::INITIALIZED_SUCCESSFUL
+    AppOpticsAPM.loaded = true
+
+  when Java::ComTracelyticsAgent::Agent::AgentStatus::INITIALIZED_FAILED
+    AppOpticsAPM.loaded = false
+    $stderr.puts '=============================================================='
+    $stderr.puts 'AppOpticsAPM Java Agent not initialized properly.'
+    $stderr.puts 'Possibly misconfigured?  Going into no-op mode.'
+    $stderr.puts 'http://docs.appoptics.solarwinds.com/Instrumentation/other-instrumentation-modules.html#jruby'
+    $stderr.puts '=============================================================='
+
+  when Java::ComTracelyticsAgent::Agent::AgentStatus::UNINITIALIZED
+    AppOpticsAPM.loaded = false
+    $stderr.puts '=============================================================='
+    $stderr.puts 'AppOpticsAPM Java Agent not loaded. Going into no-op mode.'
+    $stderr.puts 'To preload the AppOpticsAPM java agent see:'
+    $stderr.puts 'http://docs.appoptics.solarwinds.com/Instrumentation/other-instrumentation-modules.html#jruby'
+    $stderr.puts '=============================================================='
+
+  else
+    AppOpticsAPM.loaded = false
+end

data/lib/oboe.rb

@@ -0,0 +1,7 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+# This module is provided for backward compatibility.
+# It simply redirects to the appoptics_apm module which will
+# also load backward compatibility support.
+require 'appoptics_apm'

data/lib/oboe_metal.rb

@@ -0,0 +1,172 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+require 'thread'
+
+# Disable docs and Camelcase warns since we're implementing
+# an interface here.  See OboeBase for details.
+# rubocop:disable Style/Documentation, Naming/MethodName
+module AppOpticsAPM
+  extend AppOpticsAPMBase
+  include Oboe_metal
+
+  class Reporter
+    class << self
+      ##
+      # start
+      #
+      # Start the AppOpticsAPM Reporter
+      #
+      def start
+        AppOpticsAPM.loaded = false unless AppOpticsAPM::OboeInitOptions.instance.service_key_ok?
+        return unless AppOpticsAPM.loaded
+
+        begin
+          options = AppOpticsAPM::OboeInitOptions.instance.array_for_oboe # creates an array with the options in the right order
+
+          AppOpticsAPM.reporter = Oboe_metal::Reporter.new(*options)
+
+          # Only report __Init from here if we are not instrumenting a framework.
+          # Otherwise, frameworks will handle reporting __Init after full initialization
+          unless defined?(::Rails) || defined?(::Sinatra) || defined?(::Padrino) || defined?(::Grape)
+            AppOpticsAPM::API.report_init
+          end
+
+        rescue => e
+          $stderr.puts e.message
+          raise
+        end
+      end
+      alias :restart :start
+
+      ##
+      # sendReport
+      #
+      # Send the report for the given event
+      #
+      def sendReport(evt)
+        AppOpticsAPM.reporter.sendReport(evt)
+      end
+
+      ##
+      # sendStatus
+      #
+      # Send the report for the given event
+      #
+      def sendStatus(evt, context = nil)
+        AppOpticsAPM.reporter.sendStatus(evt, context)
+      end
+
+      ##
+      # clear_all_traces
+      #
+      # Truncates the trace output file to zero
+      #
+      def clear_all_traces
+        File.truncate(AppOpticsAPM::OboeInitOptions.instance.host, 0)
+      end
+
+      ##
+      # get_all_traces
+      #
+      # Retrieves all traces written to the trace file
+      #
+      def get_all_traces
+        io = File.open(AppOpticsAPM::OboeInitOptions.instance.host, 'r')
+        contents = io.readlines(nil)
+        io.close
+
+        return contents if contents.empty?
+
+        traces = []
+
+        # We use Gem.loaded_spec because older versions of the bson
+        # gem didn't even have a version embedded in the gem.  If the
+        # gem isn't in the bundle, it should rightfully error out
+        # anyways.
+        #
+        if Gem.loaded_specs['bson'] && Gem.loaded_specs['bson'].version.to_s < '4.0'
+          s = StringIO.new(contents[0])
+
+          until s.eof?
+            traces << if ::BSON.respond_to? :read_bson_document
+                        BSON.read_bson_document(s)
+                      else
+                        BSON::Document.from_bson(s)
+                      end
+          end
+        else
+          bbb = ::BSON::ByteBuffer.new(contents[0])
+          until bbb.length == 0
+            traces << Hash.from_bson(bbb)
+          end
+        end
+
+        traces
+      end
+    end
+  end
+
+  module EventUtil
+    def self.metadataString(evt)
+      evt.metadataString
+    end
+  end
+
+  class << self
+    # def sample?(opts = {})
+    #   # Return false if no-op mode
+    #   return false unless AppOpticsAPM.loaded
+    #
+    #   # Assure defaults since SWIG enforces Strings
+    #   xtrace  = opts[:xtrace]     ? opts[:xtrace].to_s.strip       : APPOPTICS_STR_BLANK
+    #
+    #   # the first arg has changed to be the service name, blank means to use the default (from the service key)
+    #   rv = AppOpticsAPM::Context.sampleRequest(APPOPTICS_STR_BLANK, xtrace)
+    #
+    #   if rv == 0
+    #     AppOpticsAPM.sample_rate = -1
+    #     AppOpticsAPM.sample_source = -1
+    #     false
+    #   else
+    #     # liboboe version > 1.3.1 returning a bit masked integer with SampleRate and
+    #     # source embedded
+    #     AppOpticsAPM.sample_rate = (rv & SAMPLE_RATE_MASK)
+    #     AppOpticsAPM.sample_source = (rv & SAMPLE_SOURCE_MASK) >> 24
+    #     true
+    #   end
+    # rescue StandardError => e
+    #   AppOpticsAPM.logger.debug "[oboe/error] sample? error: #{e.inspect}"
+    #   false
+    # end
+
+    # def set_tracing_mode(mode)
+    #   return unless AppOpticsAPM.loaded
+    #
+    #   value = mode.to_sym
+    #
+    #   case value
+    #   when :disabled, :never
+    #     AppOpticsAPM::Context.setTracingMode(APPOPTICS_TRACE_DISABLED)
+    #
+    #   when :enabled, :always
+    #     AppOpticsAPM::Context.setTracingMode(APPOPTICS_TRACE_ENABLED)
+    #
+    #   else
+    #     AppOpticsAPM.logger.fatal "[oboe/error] Invalid tracing mode set: #{mode}"
+    #     AppOpticsAPM::Context.setTracingMode(APPOPTICS_TRACE_DISABLED)
+    #   end
+    # end
+
+    def set_sample_rate(rate)
+      return unless AppOpticsAPM.loaded
+
+      # Update liboboe with the new SampleRate value
+      AppOpticsAPM::Context.setDefaultSampleRate(rate.to_i)
+    end
+  end
+end
+# rubocop:enable Style/Documentation
+
+AppOpticsAPM.loaded = true
+AppOpticsAPM.config_lock = Mutex.new

data/lib/rails/generators/appoptics_apm/install_generator.rb

@@ -0,0 +1,47 @@
+# Copyright (c) 2016 SolarWinds, LLC.
+# All rights reserved.
+
+module AppOpticsAPM
+  class InstallGenerator < ::Rails::Generators::Base
+    source_root File.join(File.dirname(__FILE__), 'templates')
+    desc "Copies a AppOpticsAPM gem initializer file to your application."
+
+    @namespace = "appoptics_apm:install"
+
+    def copy_initializer
+      # Set defaults
+      @verbose = 'false'
+
+      print_header
+      print_footer
+
+      template "appoptics_apm_initializer.rb", "config/initializers/appoptics_apm.rb"
+    end
+
+    private
+
+    # rubocop:disable Metrics/MethodLength
+    def print_header
+      say ""
+      say shell.set_color "Welcome to the AppOpticsAPM Ruby instrumentation setup.", :green, :bold
+      say ""
+      say shell.set_color "Documentation Links", :magenta
+      say "-------------------"
+      say ""
+      say "AppOpticsAPM Installation Overview:"
+      say "http://docs.appoptics.solarwinds.com/AppOpticsAPM/install-instrumentation.html"
+      say ""
+      say "More information on instrumenting Ruby applications can be found here:"
+      say "http://docs.appoptics.solarwinds.com/Instrumentation/ruby.html#installing-ruby-instrumentation"
+    end
+    # rubocop:enable Metrics/MethodLength
+
+    def print_footer
+      say ""
+      say "You can change configuration values in the future by modifying config/initializers/appoptics_apm.rb"
+      say ""
+      say "Thanks! Creating the AppOpticsAPM initializer..."
+      say ""
+    end
+  end
+end