41 lines
1.3 KiB
Ruby
41 lines
1.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Database
|
|
module Migrations
|
|
module Observers
|
|
class QueryDetails < MigrationObserver
|
|
def before
|
|
@file_path = File.join(Instrumentation::RESULT_DIR, 'current-details.json')
|
|
@file = File.open(@file_path, 'wb')
|
|
@writer = Oj::StreamWriter.new(@file, {})
|
|
@writer.push_array
|
|
@subscriber = ActiveSupport::Notifications.subscribe('sql.active_record') do |*args|
|
|
record_sql_event(*args)
|
|
end
|
|
end
|
|
|
|
def after
|
|
ActiveSupport::Notifications.unsubscribe(@subscriber)
|
|
@writer.pop_all
|
|
@writer.flush
|
|
@file.close
|
|
end
|
|
|
|
def record(observation)
|
|
File.rename(@file_path, File.join(Instrumentation::RESULT_DIR, "#{observation.migration}-query-details.json"))
|
|
end
|
|
|
|
def record_sql_event(_name, started, finished, _unique_id, payload)
|
|
@writer.push_value({
|
|
start_time: started.iso8601(6),
|
|
end_time: finished.iso8601(6),
|
|
sql: payload[:sql],
|
|
binds: payload[:type_casted_binds]
|
|
})
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|