32 lines
1.1 KiB
Ruby
32 lines
1.1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
namespace :gitlab do
|
||
|
|
namespace :db do
|
||
|
|
namespace :truncate_legacy_tables do
|
||
|
|
desc "GitLab | DB | Truncate CI Tables on Main"
|
||
|
|
task :main, [:min_batch_size] => [:environment, 'gitlab:db:validate_config'] do |_t, args|
|
||
|
|
args.with_defaults(min_batch_size: 5)
|
||
|
|
Gitlab::Database::TablesTruncate.new(
|
||
|
|
database_name: 'main',
|
||
|
|
min_batch_size: args.min_batch_size.to_i,
|
||
|
|
logger: Logger.new($stdout),
|
||
|
|
dry_run: ENV['DRY_RUN'] == 'true',
|
||
|
|
until_table: ENV['UNTIL_TABLE']
|
||
|
|
).execute
|
||
|
|
end
|
||
|
|
|
||
|
|
desc "GitLab | DB | Truncate Main Tables on CI"
|
||
|
|
task :ci, [:min_batch_size] => [:environment, 'gitlab:db:validate_config'] do |_t, args|
|
||
|
|
args.with_defaults(min_batch_size: 5)
|
||
|
|
Gitlab::Database::TablesTruncate.new(
|
||
|
|
database_name: 'ci',
|
||
|
|
min_batch_size: args.min_batch_size.to_i,
|
||
|
|
logger: Logger.new($stdout),
|
||
|
|
dry_run: ENV['DRY_RUN'] == 'true',
|
||
|
|
until_table: ENV['UNTIL_TABLE']
|
||
|
|
).execute
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|