Contents

require 'agent/am_objectholder'
require 'agent/trackers/database_tracker'

module ManageEngine
  module Instrumentation
    class ActiveRecordSQL
      
      def present?
        defined?(::ActiveRecord::Base) && defined?(::ActiveSupport::Notifications)
      end
      
      def name
        'ActiveRecord'
      end
      
      def instrument
        @obj = ManageEngine::APMObjectHolder.instance
        @obj.log.info "Instrumenting ActiveRecord"
        
        ActiveSupport::Notifications.subscribe('sql.active_record') do |name, start, finish, id, payload|
          begin 
            if @obj.config.agent_enabled && @obj.config.sql_capture && payload[:name] != "SCHEMA" # Dropping internal schema related queries
              tracker = Thread.current[:apminsight]
              if tracker != nil
                dbTracker = ManageEngine::Tracker::DatabaseTracker.new(payload[:name], start.to_f * 1000)
                dbTracker.sql(payload[:sql])
                dbTracker.params(payload[:binds])
                dbTracker.finish(finish.to_f * 1000)
                
                if dbTracker.duration >= @obj.config.sql_trace_t.to_f
                  dbTracker.sqlBacktrace(caller(10))
                end
                
                exception = payload[:exception_object]
                if exception != nil
                  dbTracker.setError(exception)
                end
                
                tracker.addChild(dbTracker)
              end
            end
          rescue Exception => e
            @obj.log.logException("Error processing #{name} payload", e)
          end
        end #subscribe
      end #def instrument
      
    end #class ActiveRecordSQL
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
site24x7_apminsight-1.5.3	lib/agent/server/instrument/active_record.rb
site24x7_apminsight-1.5.2	lib/agent/server/instrument/active_record.rb
site24x7_apminsight-1.5.1	lib/agent/server/instrument/active_record.rb
site24x7_apminsight-1.5	lib/agent/server/instrument/active_record.rb