File: query_recorder.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (21 lines) | stat: -rw-r--r-- 423 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 
# frozen_string_literal: true

class QueryRecorder
  attr_reader :log

  def initialize(&block)
    @log = []

    ActiveRecord::Base.connection.unprepared_statement do
      ActiveSupport::Notifications.subscribed(method(:callback), 'sql.active_record', &block)
    end
  end

  def callback(_name, _start, _finish, _message_id, values)
    @log << values[:sql]
  end

  def self.log(&block)
    new(&block).log
  end
end