58 lines
1.8 KiB
Ruby
58 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Git
|
|
module RuggedImpl
|
|
module UseRugged
|
|
def use_rugged?(repo, feature_key)
|
|
return Feature.enabled?(feature_key) if Feature.persisted_name?(feature_key)
|
|
|
|
# Disable Rugged auto-detect(can_use_disk?) when Puma threads>1
|
|
# https://gitlab.com/gitlab-org/gitlab/issues/119326
|
|
return false if running_puma_with_multiple_threads?
|
|
|
|
Gitlab::GitalyClient.can_use_disk?(repo.storage)
|
|
end
|
|
|
|
def execute_rugged_call(method_name, *args)
|
|
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
|
|
start = Gitlab::Metrics::System.monotonic_time
|
|
|
|
result = send(method_name, *args) # rubocop:disable GitlabSecurity/PublicSend
|
|
|
|
duration = Gitlab::Metrics::System.monotonic_time - start
|
|
|
|
if Gitlab::RuggedInstrumentation.active?
|
|
Gitlab::RuggedInstrumentation.increment_query_count
|
|
Gitlab::RuggedInstrumentation.add_query_time(duration)
|
|
|
|
Gitlab::RuggedInstrumentation.add_call_details(
|
|
feature: method_name,
|
|
args: args,
|
|
duration: duration,
|
|
backtrace: Gitlab::BacktraceCleaner.clean_backtrace(caller))
|
|
end
|
|
|
|
result
|
|
end
|
|
end
|
|
|
|
def running_puma_with_multiple_threads?
|
|
return false unless Gitlab::Runtime.puma?
|
|
|
|
::Puma.respond_to?(:cli_config) && ::Puma.cli_config.options[:max_threads] > 1
|
|
end
|
|
|
|
def rugged_feature_keys
|
|
Gitlab::Git::RuggedImpl::Repository::FEATURE_FLAGS
|
|
end
|
|
|
|
def rugged_enabled_through_feature_flag?
|
|
rugged_feature_keys.any? do |feature_key|
|
|
Feature.enabled?(feature_key)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|