diff options
Diffstat (limited to 'db/post_migrate')
-rw-r--r-- | db/post_migrate/20170526185921_migrate_build_stage_reference.rb | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/db/post_migrate/20170526185921_migrate_build_stage_reference.rb b/db/post_migrate/20170526185921_migrate_build_stage_reference.rb index 3c30dfd6dde..aa50b72839d 100644 --- a/db/post_migrate/20170526185921_migrate_build_stage_reference.rb +++ b/db/post_migrate/20170526185921_migrate_build_stage_reference.rb @@ -6,9 +6,11 @@ class MigrateBuildStageReference < ActiveRecord::Migration def up disable_statement_timeout - stage_id = Arel.sql('(SELECT id FROM ci_stages ' \ - 'WHERE ci_stages.pipeline_id = ci_builds.commit_id ' \ - 'AND ci_stages.name = ci_builds.stage)') + stage_id = Arel.sql(<<-SQL.strip_heredoc + (SELECT id FROM ci_stages + WHERE ci_stages.pipeline_id = ci_builds.commit_id + AND ci_stages.name = ci_builds.stage) + SQL update_column_in_batches(:ci_builds, :stage_id, stage_id) do |table, query| query.where(table[:stage_id].eq(nil)) |