Add unset_rugged rake task

Adds an unset_rugged rake task that unsets all rugged feature flags.
Also fixes the existing disable_rugged task to have it explicitly
disable feature flags instead of just unsetting them.
This commit is contained in:
John Cai 2019-07-11 09:53:38 -07:00
parent a06107582f
commit ff0654b0b4
1 changed files with 11 additions and 3 deletions

View File

@ -10,14 +10,22 @@ namespace :gitlab do
set_rugged_feature_flags(false)
puts 'All Rugged feature flags were disabled.'
end
task unset_rugged: :environment do
set_rugged_feature_flags(nil)
puts 'All Rugged feature flags were unset.'
end
end
def set_rugged_feature_flags(status)
Gitlab::Git::RuggedImpl::Repository::FEATURE_FLAGS.each do |flag|
if status
Feature.enable(flag)
else
case status
when nil
Feature.get(flag).remove
when true
Feature.enable(flag)
when false
Feature.disable(flag)
end
end
end