34 lines
870 B
Ruby
34 lines
870 B
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module Git
|
|
module RuggedImpl
|
|
module UseRugged
|
|
def use_rugged?(repo, feature_key)
|
|
feature = Feature.get(feature_key)
|
|
return feature.enabled? if Feature.persisted?(feature)
|
|
|
|
Gitlab::GitalyClient.can_use_disk?(repo.storage)
|
|
end
|
|
|
|
def wrap_rugged_call(&block)
|
|
Gitlab::GitalyClient::StorageSettings.allow_disk_access do
|
|
start = Gitlab::Metrics::System.monotonic_time
|
|
|
|
result = yield
|
|
|
|
duration = Gitlab::Metrics::System.monotonic_time - start
|
|
|
|
if Gitlab::RuggedInstrumentation.active?
|
|
Gitlab::RuggedInstrumentation.increment_query_count
|
|
Gitlab::RuggedInstrumentation.query_time += duration
|
|
end
|
|
|
|
result
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|