54 lines
1.2 KiB
Ruby
54 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Peek
|
|
module Views
|
|
class ActiveRecord < DetailedView
|
|
DEFAULT_THRESHOLDS = {
|
|
calls: 100,
|
|
duration: 3000,
|
|
individual_call: 1000
|
|
}.freeze
|
|
|
|
THRESHOLDS = {
|
|
production: {
|
|
calls: 100,
|
|
duration: 15000,
|
|
individual_call: 5000
|
|
}
|
|
}.freeze
|
|
|
|
def results
|
|
super.merge(calls: detailed_calls)
|
|
end
|
|
|
|
def self.thresholds
|
|
@thresholds ||= THRESHOLDS.fetch(Rails.env.to_sym, DEFAULT_THRESHOLDS)
|
|
end
|
|
|
|
private
|
|
|
|
def detailed_calls
|
|
"#{calls} (#{cached_calls} cached)"
|
|
end
|
|
|
|
def cached_calls
|
|
detail_store.count { |item| item[:cached] == 'cached' }
|
|
end
|
|
|
|
def setup_subscribers
|
|
super
|
|
|
|
subscribe('sql.active_record') do |_, start, finish, _, data|
|
|
if Gitlab::PerformanceBar.enabled_for_request?
|
|
detail_store << {
|
|
duration: finish - start,
|
|
sql: data[:sql].strip,
|
|
backtrace: Gitlab::BacktraceCleaner.clean_backtrace(caller),
|
|
cached: data[:cached] ? 'cached' : ''
|
|
}
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|