diff --git a/lib/tasks/gitlab/db/alter_cell_sequences_range.rake b/lib/tasks/gitlab/db/alter_cell_sequences_range.rake index 36304a77af2b39d5bd810120dc7b56408fee4575..3b06bb49c8b1fc3abb86955445fb4ff5452ddd2c 100644 --- a/lib/tasks/gitlab/db/alter_cell_sequences_range.rake +++ b/lib/tasks/gitlab/db/alter_cell_sequences_range.rake @@ -1,19 +1,13 @@ # frozen_string_literal: true -# rubocop:disable Gitlab/AvoidGitlabInstanceChecks -- We want to set sequence limits only on Gitlab instances namespace :gitlab do namespace :db do desc 'Alters sequence limits for cell specific tables' task :alter_cell_sequences_range, [:minval, :maxval] => :environment do |_t, args| - next unless Gitlab.com_except_jh? || Gitlab.dev_or_test_env? - # This is a safety check to ensure this rake does not alter the sequences for the Legacy Cell next if Gitlab.config.cell.database.skip_sequence_alteration - Gitlab::Database::EachDatabase.each_connection do |connection, _database_name| - Gitlab::Database::AlterCellSequencesRange.new(args.minval&.to_i, args.maxval&.to_i, connection).execute - end + Rake::Task['gitlab:db:alter_sequences_range'].invoke(*args) end end end -# rubocop:enable Gitlab/AvoidGitlabInstanceChecks diff --git a/lib/tasks/gitlab/db/alter_sequences_range.rake b/lib/tasks/gitlab/db/alter_sequences_range.rake new file mode 100644 index 0000000000000000000000000000000000000000..20559b81ec4fcc2156c54fcb8a2459d4f78ca68d --- /dev/null +++ b/lib/tasks/gitlab/db/alter_sequences_range.rake @@ -0,0 +1,16 @@ +# frozen_string_literal: true + +# rubocop:disable Gitlab/AvoidGitlabInstanceChecks -- We want to set sequence limits only on Gitlab instances +namespace :gitlab do + namespace :db do + desc 'Alters sequence limits for cell specific tables' + task :alter_sequences_range, [:minval, :maxval] => :environment do |_t, args| + next unless Gitlab.com_except_jh? || Gitlab.dev_or_test_env? + + Gitlab::Database::EachDatabase.each_connection do |connection, _database_name| + Gitlab::Database::AlterCellSequencesRange.new(args.minval&.to_i, args.maxval&.to_i, connection).execute + end + end + end +end +# rubocop:enable Gitlab/AvoidGitlabInstanceChecks diff --git a/scripts/utils.sh b/scripts/utils.sh index efb2118325dab10eb3ea1f3c839e47a6e3b83be1..60da252b71662d3c53d436af1d71c987959c4303 100644 --- a/scripts/utils.sh +++ b/scripts/utils.sh @@ -198,7 +198,7 @@ function setup_db() { fi setup_db_user_only - run_timed_command_with_metric "bundle exec rake db:drop db:create db:schema:load db:migrate gitlab:db:lock_writes" "setup_db" + run_timed_command_with_metric "bundle exec rake db:drop db:create db:schema:load db:migrate gitlab:db:lock_writes 'gitlab:db:alter_sequences_range[100_000_000_001,200_000_000_000]'" "setup_db" setup_db_praefect section_end "setup-db"