BigW Consortium Gitlab

20170830150306_drop_events_for_migration_table.rb 1.29 KB
Newer Older
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
# See http://doc.gitlab.com/ce/development/migration_style_guide.html
# for more information on how to write migrations for GitLab.

class DropEventsForMigrationTable < ActiveRecord::Migration
  include Gitlab::Database::MigrationHelpers

  DOWNTIME = false

  disable_ddl_transaction!

  class Event < ActiveRecord::Base
    include EachBatch
  end

  def up
    transaction do
      drop_table :events_for_migration
    end
  end

  # rubocop: disable Migration/Datetime
  def down
    create_table :events_for_migration do |t|
      t.string :target_type, index: true
      t.integer :target_id, index: true
      t.string :title
      t.text :data
      t.integer :project_id
      t.datetime :created_at, index: true
      t.datetime :updated_at
      t.integer :action, index: true
      t.integer :author_id, index: true

      t.index [:project_id, :id]
    end

    Event.all.each_batch do |relation|
      start_id, stop_id = relation.pluck('MIN(id), MAX(id)').first

      execute <<-EOF.strip_heredoc
      INSERT INTO events_for_migration (target_type, target_id, project_id, created_at, updated_at, action, author_id)
      SELECT target_type, target_id, project_id, created_at, updated_at, action, author_id
      FROM events
      WHERE id BETWEEN #{start_id} AND #{stop_id}
      EOF
    end
  end
end