summaryrefslogtreecommitdiff
path: root/db/post_migrate/20180119121225_remove_redundant_pipeline_stages.rb
blob: 89ed422ea1c1b0fb5d91ec0ae23a1e331390f672 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class RemoveRedundantPipelineStages < ActiveRecord::Migration
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = false

  def up
    redundant_stages_ids = <<~SQL
      SELECT id FROM ci_stages WHERE (pipeline_id, name) IN (
        SELECT pipeline_id, name FROM ci_stages
          GROUP BY pipeline_id, name HAVING COUNT(*) > 1
      )
    SQL

    execute <<~SQL
      UPDATE ci_builds SET stage_id = NULL WHERE stage_id IN (#{redundant_stages_ids})
    SQL

    if Gitlab::Database.postgresql?
      execute <<~SQL
        DELETE FROM ci_stages WHERE id IN (#{redundant_stages_ids})
      SQL
    else # We can't modify a table we are selecting from on MySQL
      execute <<~SQL
        DELETE a FROM ci_stages AS a, ci_stages AS b
          WHERE a.pipeline_id = b.pipeline_id AND a.name = b.name
            AND a.id <> b.id
      SQL
    end
  end

  def down
    # noop
  end
end