Implement build stage_id reference migration clean up

2 files changed, 22 insertions, 1 deletions

diff --git a/db/migrate/20170710083355_build_stage_id_ref_migration_cleanup.rb b/db/migrate/20170710083355_build_stage_id_ref_migration_cleanup.rb
new file mode 100644
index 00000000000..be8efb140d9
--- /dev/null
+++ b/db/migrate/20170710083355_build_stage_id_ref_migration_cleanup.rb
@@ -0,0 +1,21 @@
+require Rails.root.join('db', 'post_migrate', '20170628080858_migrate_stage_id_reference_in_background')
+
+class BuildStageIdRefMigrationCleanup < ActiveRecord::Migration
+  include Gitlab::Database::MigrationHelpers
+
+  DOWNTIME = false
+
+  disable_ddl_transaction!
+
+  ##
+  # `MigrateStageIdReferenceInBackground` background migration cleanup.
+  #
+  def up
+    Gitlab::BackgroundMigration
+      .steal(MigrateStageIdReferenceInBackground::MIGRATION)
+  end
+
+  def down
+    # noop
+  end
+end
diff --git a/db/schema.rb b/db/schema.rb
index 5264fc99557..44bc0109e66 100644
--- a/db/schema.rb
+++ b/db/schema.rb
@@ -11,7 +11,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 20170707184244) do
+ActiveRecord::Schema.define(version: 20170711145558) do
 
   # These are extensions that must be enabled in order to support this database
   enable_extension "plpgsql"