Prepare rake task for storage rollback

We are keeping compatibility with existing scheduled jobs.

1 files changed, 8 insertions, 2 deletions

diff --git a/app/workers/storage_migrator_worker.rb b/app/workers/storage_migrator_worker.rb
index fa76fbac55c..714c4eed5b2 100644
--- a/app/workers/storage_migrator_worker.rb
+++ b/app/workers/storage_migrator_worker.rb
@@ -3,8 +3,14 @@
 class StorageMigratorWorker
   include ApplicationWorker
 
-  def perform(start, finish)
+  # @param [Integer] start initial ID of the batch
+  # @param [Integer] finish last ID of the batch
+  # @param [String] operation the operation to be performed: ['migrate', 'rollback']
+  def perform(start, finish, operation = :migrate)
+    # when scheduling a job symbols are converted to string, we need to convert back
+    operation = operation.to_sym if operation
+
     migrator = Gitlab::HashedStorage::Migrator.new
-    migrator.bulk_migrate(start, finish)
+    migrator.bulk_migrate(start: start, finish: finish, operation: operation)
   end
 end