20161018024215_migrate_labels_priority.rb 1.13 KB
Newer Older
1
class MigrateLabelsPriority < ActiveRecord::Migration[4.2]
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
  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