summaryrefslogtreecommitdiff
path: root/db/migrate/20161014173530_create_label_priorities.rb
blob: f9d94ebdc7081a276e14dd782f5d601f1fdff4a5 (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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
class CreateLabelPriorities < ActiveRecord::Migration
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = true
  DOWNTIME_REASON = 'Prioritezed labels will not work as expected until this migration is complete.'

  disable_ddl_transaction!

  def up
    create_table :label_priorities do |t|
      t.references :project, foreign_key: { on_delete: :cascade }, null: false
      t.references :label, foreign_key: { on_delete: :cascade }, null: false
      t.integer :priority, null: false

      t.timestamps null: false
    end

    execute <<-EOF.strip_heredoc
      INSERT INTO label_priorities (project_id, label_id, priority, created_at, updated_at)
      SELECT labels.project_id, labels.id, labels.priority, NOW(), NOW()
      FROM labels
      WHERE labels.project_id IS NOT NULL
        AND labels.priority IS NOT NULL;
    EOF

    add_concurrent_index :label_priorities, [:project_id, :label_id], unique: true
    add_concurrent_index :label_priorities, :priority

    remove_column :labels, :priority
  end

  def down
    add_column :labels, :priority, :integer

    if Gitlab::Database.mysql?
      execute <<-EOF.strip_heredoc
        UPDATE labels
          INNER JOIN label_priorities ON labels.id = label_priorities.label_id AND labels.project_id = label_priorities.project_id
        SET labels.priority = label_priorities.priority;
      EOF
    else
      execute <<-EOF.strip_heredoc
        UPDATE labels
        SET priority = label_priorities.priority
        FROM label_priorities
        WHERE labels.id = label_priorities.label_id
          AND labels.project_id = label_priorities.project_id;
      EOF
    end

    add_concurrent_index :labels, :priority

    drop_table :label_priorities
  end
end