summaryrefslogtreecommitdiff
path: root/lib/gitlab/metrics/subscribers/active_record.rb
blob: d947c128ce22249a882d0816878bcb7b8bca23b5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
module Gitlab
  module Metrics
    module Subscribers
      # Class for tracking raw SQL queries.
      #
      # Queries are obfuscated before being logged to ensure no private data is
      # exposed via InfluxDB/Grafana.
      class ActiveRecord < ActiveSupport::Subscriber
        attach_to :active_record

        SERIES = 'sql_queries'

        def sql(event)
          return unless current_transaction

          values = values_for(event)
          tags   = tags_for(event)

          current_transaction.add_metric(SERIES, values, tags)
        end

        private

        def values_for(event)
          { duration: event.duration }
        end

        def tags_for(event)
          sql  = ObfuscatedSQL.new(event.payload[:sql]).to_s
          tags = { sql: sql }

          file, line = Metrics.last_relative_application_frame

          if file and line
            tags[:file] = file
            tags[:line] = line
          end

          tags
        end

        def current_transaction
          Transaction.current
        end
      end
    end
  end
end