20160416182152_convert_award_note_to_emoji_award.rb 1.46 KB
Newer Older
1
class ConvertAwardNoteToEmojiAward < ActiveRecord::Migration[4.2]
2
  disable_ddl_transaction!
Z.J. van de Weg's avatar
Z.J. van de Weg committed
3

4 5 6 7 8
  def up
    if Gitlab::Database.postgresql?
      migrate_postgresql
    else
      migrate_mysql
9 10
    end
  end
11 12 13 14 15 16 17 18 19 20

  def down
    add_column :notes, :is_award, :boolean

    # This migration does NOT move the awards on notes, if the table is dropped in another migration, these notes will be lost.
    execute "INSERT INTO notes (noteable_type, noteable_id, author_id, note, created_at, updated_at, is_award) (SELECT awardable_type, awardable_id, user_id, name, created_at, updated_at, TRUE FROM award_emoji)"
  end

  def migrate_postgresql
    connection.transaction do
21
      execute 'LOCK notes IN EXCLUSIVE MODE'
22 23 24
      execute "INSERT INTO award_emoji (awardable_type, awardable_id, user_id, name, created_at, updated_at) (SELECT noteable_type, noteable_id, author_id, note, created_at, updated_at FROM notes WHERE is_award = true)"
      execute "DELETE FROM notes WHERE is_award = true"
      remove_column :notes, :is_award, :boolean
25 26 27 28
    end
  end

  def migrate_mysql
29 30
    execute 'LOCK TABLES notes WRITE, award_emoji WRITE;'
    execute 'INSERT INTO award_emoji (awardable_type, awardable_id, user_id, name, created_at, updated_at) (SELECT noteable_type, noteable_id, author_id, note, created_at, updated_at FROM notes WHERE is_award = true);'
31 32
    execute "DELETE FROM notes WHERE is_award = true"
    remove_column :notes, :is_award, :boolean
33 34
  ensure
    execute 'UNLOCK TABLES'
35
  end
36
end