appoptics_apm 4.0.2

data/lib/joboe_metal.rb

@@ -0,0 +1,214 @@
+# 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 = :always
+          else
+            AppOpticsAPM::Config.tracing_mode = :never
+          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[:layer]  = nil      if opts[:layer].is_a?(String)      && opts[:layer].empty?
+        opts[:xtrace] = nil      if opts[:xtrace].is_a?(String)     && opts[:xtrace].empty?
+
+        opts[:layer]      ||= nil
+        opts[:xtrace]     ||= nil
+
+        sr_cfg = Java::ComTracelyticsJoboe::LayerUtil.shouldTraceRequest(opts[:layer], { '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 '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/README

@@ -0,0 +1,2 @@
+This directory is kept for backward compatibility for applications
+making calls to Oboe modules.

data/lib/oboe/backward_compatibility.rb

@@ -0,0 +1,80 @@
+require 'appoptics_apm/thread_local'
+
+module Oboe
+  extend AppOpticsAPMBase
+  if AppOpticsAPM.loaded
+    include Oboe_metal
+  end
+
+  #
+  # Support for Oboe::API calls
+  #
+  module API
+    include AppOpticsAPM::API
+    extend ::AppOpticsAPM::ThreadLocal
+    thread_local :deprecation_notified
+
+    def self.method_missing(sym, *args, &blk)
+      # Notify of deprecation only once
+      unless @deprecated_notified
+        AppOpticsAPM.logger.warn "[appoptics_apm/warn] Note that Oboe::API has been renamed to AppOpticsAPM::API. (#{sym}:#{args})"
+        AppOpticsAPM.logger.warn '[appoptics_apm/warn] Oboe::API will be deprecated in a future version.'
+        AppOpticsAPM.logger.warn "[appoptics_apm/warn] Caller: #{Kernel.caller[0]}"
+        @deprecated_notified = true
+      end
+      AppOpticsAPM::API.send(sym, *args, &blk)
+    end
+  end
+
+
+  #
+  # Support for Oboe::Config calls
+  #
+  module Config
+    extend ::AppOpticsAPM::ThreadLocal
+    thread_local :deprecation_notified
+
+    def self.method_missing(sym, *args)
+      # Notify of deprecation only once
+      unless @deprecated_notified
+        AppOpticsAPM.logger.warn "[appoptics_apm/warn] Note that Oboe::Config has been renamed to AppOpticsAPM::Config. (#{sym}:#{args})"
+        AppOpticsAPM.logger.warn '[appoptics_apm/warn] Oboe::Config will be deprecated in a future version.'
+        AppOpticsAPM.logger.warn "[appoptics_apm/warn] Caller: #{Kernel.caller[0]}"
+        @deprecated_notified = true
+      end
+      AppOpticsAPM::Config.send(sym, *args)
+    end
+  end
+
+  #
+  # Support for legacy Oboe::Ruby.load calls
+  #
+  module Ruby
+    extend ::AppOpticsAPM::ThreadLocal
+    thread_local :deprecation_notified
+
+    def self.method_missing(sym, *args)
+      # Notify of deprecation only once
+      unless @deprecated_notified
+        AppOpticsAPM.logger.warn "[appoptics_apm/warn] Note that Oboe::Ruby has been renamed to AppOpticsAPM::Ruby. (#{sym}:#{args})"
+        AppOpticsAPM.logger.warn '[appoptics_apm/warn] Oboe::Ruby will be deprecated in a future version.'
+        AppOpticsAPM.logger.warn "[appoptics_apm/warn] Caller: #{Kernel.caller[0]}"
+        @deprecated_notified = true
+      end
+      AppOpticsAPM::Ruby.send(sym, *args)
+    end
+  end
+end
+
+#
+# Support for OboeMethodProfiling
+#
+module OboeMethodProfiling
+  def self.included(klass)
+    klass.extend ClassMethods
+  end
+
+  module ClassMethods
+    include AppOpticsAPMMethodProfiling::ClassMethods
+  end
+end

data/lib/oboe/inst/rack.rb

@@ -0,0 +1,11 @@
+require 'appoptics_apm/inst/rack'
+
+module Oboe
+  class Rack < AppOpticsAPM::Rack
+    # This simply makes Oboe::Rack available (and a clone of AppOpticsAPM::Rack) for
+    # backward compatibility
+    #
+    # Provided for pre-existing apps (sinatra, padrino, grape etc..) that may still
+    # call `use Oboe::Rack`
+  end
+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,187 @@
+# 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, Style/MethodName
+module AppOpticsAPM
+  extend AppOpticsAPMBase
+  include Oboe_metal
+
+  class Reporter
+    class << self
+      ##
+      # start
+      #
+      # Start the AppOpticsAPM Reporter
+      #
+      def start
+        return unless AppOpticsAPM.loaded
+
+        begin
+          protocol = ENV.key?('APPOPTICS_GEM_TEST') ? 'file' :
+                       ENV['TRACELYTICS_REPORTER'] || 'ssl'
+
+          case protocol
+          when 'file'
+            options = "file=#{TRACE_FILE}"
+          when 'udp'
+            options = "addr=#{AppOpticsAPM::Config[:reporter_host]},port=#{AppOpticsAPM::Config[:reporter_port]}"
+          else
+            if ENV['APPOPTICS_SERVICE_KEY'].to_s == ''
+              AppOpticsAPM.logger.warn "[appoptics_apm/warn] APPOPTICS_SERVICE_KEY not set. Cannot submit data."
+              AppOpticsAPM.loaded = false
+              return
+            end
+            # ssl reporter requires the service key passed in as arg "cid"
+            options = "cid=#{ENV['APPOPTICS_SERVICE_KEY']}"
+          end
+
+          AppOpticsAPM.reporter = Oboe_metal::Reporter.new(protocol, 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(TRACE_FILE, 0)
+      end
+
+      ##
+      # get_all_traces
+      #
+      # Retrieves all traces written to the trace file
+      #
+      def get_all_traces
+        io = File.open(TRACE_FILE, 'r')
+        contents = io.readlines(nil)
+
+        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'].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
+      layer   = opts[:layer]      ? opts[:layer].to_s.strip.freeze : APPOPTICS_STR_BLANK
+      xtrace  = opts[:xtrace]     ? opts[:xtrace].to_s.strip       : APPOPTICS_STR_BLANK
+
+      rv = AppOpticsAPM::Context.sampleRequest(layer, 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 :never
+        AppOpticsAPM::Context.setTracingMode(OBOE_TRACE_NEVER)
+
+      when :always
+        AppOpticsAPM::Context.setTracingMode(OBOE_TRACE_ALWAYS)
+
+      else
+        AppOpticsAPM.logger.fatal "[oboe/error] Invalid tracing mode set: #{mode}"
+        AppOpticsAPM::Context.setTracingMode(OBOE_TRACE_NEVER)
+      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,45 @@
+# 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
+
+      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
+
+      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