diff options
Diffstat (limited to 'lib/gitlab/background_migration/batching_strategies/primary_key_batching_strategy.rb')
-rw-r--r-- | lib/gitlab/background_migration/batching_strategies/primary_key_batching_strategy.rb | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/lib/gitlab/background_migration/batching_strategies/primary_key_batching_strategy.rb b/lib/gitlab/background_migration/batching_strategies/primary_key_batching_strategy.rb index 80693728e86..09700438d47 100644 --- a/lib/gitlab/background_migration/batching_strategies/primary_key_batching_strategy.rb +++ b/lib/gitlab/background_migration/batching_strategies/primary_key_batching_strategy.rb @@ -17,8 +17,9 @@ module Gitlab # column_name - The column to batch over # batch_min_value - The minimum value which the next batch will start at # batch_size - The size of the next batch - def next_batch(table_name, column_name, batch_min_value:, batch_size:) - model_class = define_batchable_model(table_name) + # job_arguments - The migration job arguments + def next_batch(table_name, column_name, batch_min_value:, batch_size:, job_arguments:) + model_class = define_batchable_model(table_name, connection: ActiveRecord::Base.connection) quoted_column_name = model_class.connection.quote_column_name(column_name) relation = model_class.where("#{quoted_column_name} >= ?", batch_min_value) |