summaryrefslogtreecommitdiff
path: root/db/migrate/20161018024215_migrate_labels_priority.rb
blob: 3e2540c134cdd0b2a27ca517fabc72b586978177 (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
class MigrateLabelsPriority < ActiveRecord::Migration[4.2]
  include Gitlab::Database::MigrationHelpers

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

  disable_ddl_transaction!

  def up
    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
  end

  def down
    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
  end
end