lib/gitlab/database/migrations/observers/query_log.rb


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

# frozen_string_literal: true

module Gitlab
  module Database
    module Migrations
      module Observers
        class QueryLog < MigrationObserver
          def before
            @logger_was = ActiveRecord::Base.logger
            @log_file_path = File.join(Instrumentation::RESULT_DIR, 'current.log')
            @logger = Logger.new(@log_file_path)
            ActiveRecord::Base.logger = @logger
          end

          def after
            ActiveRecord::Base.logger = @logger_was
            @logger.close
          end

          def record(observation)
            File.rename(@log_file_path, File.join(Instrumentation::RESULT_DIR, "#{observation.migration}.log"))
          end
        end
      end
    end
  end
end