Make sure all stages are migrated after a cleanupbackstage/gb/steal-stages-statuses-migration

1 files changed, 11 insertions, 0 deletions

diff --git a/db/migrate/20170912113435_clean_stages_statuses_migration.rb b/db/migrate/20170912113435_clean_stages_statuses_migration.rb
index 31fd240d3dc..fc091d7894e 100644
--- a/db/migrate/20170912113435_clean_stages_statuses_migration.rb
+++ b/db/migrate/20170912113435_clean_stages_statuses_migration.rb
@@ -5,8 +5,19 @@ class CleanStagesStatusesMigration < ActiveRecord::Migration
 
   disable_ddl_transaction!
 
+  class Stage < ActiveRecord::Base
+    include ::EachBatch
+    self.table_name = 'ci_stages'
+  end
+
   def up
     Gitlab::BackgroundMigration.steal('MigrateStageStatus')
+
+    Stage.where('status IS NULL').each_batch(of: 50) do |batch|
+      range = batch.pluck('MIN(id)', 'MAX(id)').first
+
+      Gitlab::BackgroundMigration::MigrateStageStatus.new.perform(*range)
+    end
   end
 
   def down