Class: Sentry::Rails::Tracing::ActiveRecordSubscriber

Inherits:
AbstractSubscriber show all
Defined in:
lib/sentry/rails/tracing/active_record_subscriber.rb

Constant Summary collapse

EVENT_NAMES = 
["sql.active_record"].freeze
SPAN_PREFIX = 
"db."
SPAN_ORIGIN = 
"auto.db.rails"
EXCLUDED_EVENTS = 
["SCHEMA", "TRANSACTION"].freeze
SUPPORT_SOURCE_LOCATION = 
ActiveSupport::BacktraceCleaner.method_defined?(:clean_frame)

Class Method Summary collapse

Methods inherited from AbstractSubscriber

record_on_current_span, #subscribe_to_event, unsubscribe!

Class Method Details

.subscribe! ⇒ Object 



26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
 
# File 'lib/sentry/rails/tracing/active_record_subscriber.rb', line 26

def subscribe!
  record_query_source = SUPPORT_SOURCE_LOCATION && Sentry.configuration.rails.enable_db_query_source
  query_source_threshold = Sentry.configuration.rails.db_query_source_threshold_ms

  subscribe_to_event(EVENT_NAMES) do |event_name, duration, payload|
    next if EXCLUDED_EVENTS.include? payload[:name]

    record_on_current_span(
      op: SPAN_PREFIX + event_name,
      origin: SPAN_ORIGIN,
      start_timestamp: payload[START_TIMESTAMP_NAME],
      description: payload[:sql],
      duration: duration
    ) do |span|
      span.set_tag(:cached, true) if payload.fetch(:cached, false) # cached key is only set for hits in the QueryCache, from Rails 5.1

      connection = payload[:connection]

      if payload[:connection_id]
        span.set_data(:connection_id, payload[:connection_id])

        # we fallback to the base connection on rails < 6.0.0 since the payload doesn't have it
        connection ||= ActiveRecord::Base.connection_pool.connections.find { |conn| conn.object_id == payload[:connection_id] }
      end

      next unless connection

      db_config =
        if connection.pool.respond_to?(:db_config)
          connection.pool.db_config.configuration_hash
        elsif connection.pool.respond_to?(:spec)
          connection.pool.spec.config
        end

      next unless db_config

      span.set_data(Span::DataConventions::DB_SYSTEM, db_config[:adapter]) if db_config[:adapter]
      span.set_data(Span::DataConventions::DB_NAME, db_config[:database]) if db_config[:database]
      span.set_data(Span::DataConventions::SERVER_ADDRESS, db_config[:host]) if db_config[:host]
      span.set_data(Span::DataConventions::SERVER_PORT, db_config[:port]) if db_config[:port]
      span.set_data(Span::DataConventions::SERVER_SOCKET_ADDRESS, db_config[:socket]) if db_config[:socket]

      # both duration and query_source_threshold are in ms
      if record_query_source && duration >= query_source_threshold
        backtrace_line = Backtrace.source_location(&backtrace_cleaner)

        if backtrace_line
          span.set_data(Span::DataConventions::FILEPATH, backtrace_line.file) if backtrace_line.file
          span.set_data(Span::DataConventions::LINENO, backtrace_line.number) if backtrace_line.number
          span.set_data(Span::DataConventions::FUNCTION, backtrace_line.method) if backtrace_line.method
          # Only JRuby has namespace in the backtrace
          span.set_data(Span::DataConventions::NAMESPACE, backtrace_line.module_name) if backtrace_line.module_name
        end
      end
    end
  end
end