db/post_migrate/20180204200836_change_author_id_to_not_null_in_todos.rb


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

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

  class Todo < ActiveRecord::Base
    self.table_name = 'todos'
    include EachBatch
  end

  BATCH_SIZE = 1000

  DOWNTIME = false

  disable_ddl_transaction!

  def up
    Todo.where(author_id: nil).each_batch(of: BATCH_SIZE) do |batch|
      batch.delete_all
    end

    change_column_null :todos, :author_id, false
  end

  def down
    change_column_null :todos, :author_id, true
  end
end