45 lines
1.3 KiB
Ruby
45 lines
1.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# Inspired by https://github.com/peek/peek-pg/blob/master/lib/peek/views/pg.rb
|
|
# PEEK_DB_CLIENT is a constant set in config/initializers/peek.rb
|
|
module Gitlab
|
|
module PerformanceBar
|
|
module PeekQueryTracker
|
|
def sorted_queries
|
|
PEEK_DB_CLIENT.query_details
|
|
.sort { |a, b| b[:duration] <=> a[:duration] }
|
|
end
|
|
|
|
def results
|
|
super.merge(queries: sorted_queries)
|
|
end
|
|
|
|
private
|
|
|
|
def setup_subscribers
|
|
super
|
|
|
|
# Reset each counter when a new request starts
|
|
before_request do
|
|
PEEK_DB_CLIENT.query_details = []
|
|
end
|
|
|
|
subscribe('sql.active_record') do |_, start, finish, _, data|
|
|
if Gitlab::SafeRequestStore.store[:peek_enabled]
|
|
unless data[:cached]
|
|
backtrace = Gitlab::Profiler.clean_backtrace(caller)
|
|
track_query(data[:sql].strip, data[:binds], backtrace, start, finish)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def track_query(raw_query, bindings, backtrace, start, finish)
|
|
duration = (finish - start) * 1000.0
|
|
query_info = { duration: duration.round(3), sql: raw_query, backtrace: backtrace }
|
|
|
|
PEEK_DB_CLIENT.query_details << query_info
|
|
end
|
|
end
|
|
end
|
|
end
|