1 files changed, 61 insertions, 0 deletions
diff --git a/lib/gitlab/background_migration/migrate_to_hashed_storage.rb b/lib/gitlab/background_migration/migrate_to_hashed_storage.rb
new file mode 100644
index 00000000000..4054db4fb87
--- /dev/null
+++ b/lib/gitlab/background_migration/migrate_to_hashed_storage.rb
@@ -0,0 +1,61 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module BackgroundMigration
+    # Background migration to move any legacy project to Hashed Storage
+    class MigrateToHashedStorage
+      def perform
+        batch_size = helper.batch_size
+        legacy_projects_count = Project.with_unmigrated_storage.count
+
+        if storage_migrator.rollback_pending?
+          logger.warn(
+            migrator: 'MigrateToHashedStorage',
+            message: 'Aborting an storage rollback operation currently in progress'
+          )
+
+          storage_migrator.abort_rollback!
+        end
+
+        if legacy_projects_count == 0
+          logger.info(
+            migrator: 'MigrateToHashedStorage',
+            message: 'There are no projects requiring migration to Hashed Storage'
+          )
+
+          return
+        end
+
+        logger.info(
+          migrator: 'MigrateToHashedStorage',
+          message: "Enqueuing migration of #{legacy_projects_count} projects in batches of #{batch_size}"
+        )
+
+        helper.project_id_batches_migration do |start, finish|
+          storage_migrator.bulk_schedule_migration(start: start, finish: finish)
+
+          logger.info(
+            migrator: 'MigrateToHashedStorage',
+            message: "Enqueuing migration of projects in batches of #{batch_size} from ID=#{start} to ID=#{finish}",
+            batch_from: start,
+            batch_to: finish
+          )
+        end
+      end
+
+      private
+
+      def helper
+        Gitlab::HashedStorage::RakeHelper
+      end
+
+      def storage_migrator
+        @storage_migrator ||= Gitlab::HashedStorage::Migrator.new
+      end
+
+      def logger
+        @logger ||= ::Gitlab::BackgroundMigration::Logger.build
+      end
+    end
+  end
+end