gitlab-org--gitlab-foss/lib/gitlab/rugged_instrumentation.rb
Stan Hu 291df05e43 Add Rugged calls to performance bar
This will help diagnose the source of excessive I/O from Rugged
calls. To implement this, we need to obtain the full list of arguments
sent to each request method.
2019-07-23 21:38:05 -07:00

47 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