48 lines
1 KiB
Ruby
48 lines
1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module RuggedInstrumentation
|
||
|
|
def self.query_time
|
||
|
|
SafeRequestStore[:rugged_query_time] ||= 0
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.query_time=(duration)
|
||
|
|
SafeRequestStore[:rugged_query_time] = duration
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.query_time_ms
|
||
|
|
(self.query_time * 1000).round(2)
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.query_count
|
||
|
|
SafeRequestStore[:rugged_call_count] ||= 0
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.increment_query_count
|
||
|
|
SafeRequestStore[:rugged_call_count] ||= 0
|
||
|
|
SafeRequestStore[:rugged_call_count] += 1
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.active?
|
||
|
|
SafeRequestStore.active?
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.peek_enabled?
|
||
|
|
SafeRequestStore[:peek_enabled]
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.add_call_details(details)
|
||
|
|
return unless peek_enabled?
|
||
|
|
|
||
|
|
Gitlab::SafeRequestStore[:rugged_call_details] ||= []
|
||
|
|
Gitlab::SafeRequestStore[:rugged_call_details] << details
|
||
|
|
end
|
||
|
|
|
||
|
|
def self.list_call_details
|
||
|
|
return [] unless peek_enabled?
|
||
|
|
|
||
|
|
Gitlab::SafeRequestStore[:rugged_call_details] || []
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|