gitlab-org--gitlab-foss/db/migrate/20180608091413_add_group_to_todos.rb

37 lines
902 B
Ruby
Raw Normal View History

class AddGroupToTodos < ActiveRecord::Migration[4.2]
include Gitlab::Database::MigrationHelpers
DOWNTIME = false
disable_ddl_transaction!
2018-08-06 05:05:44 -04:00
class Todo < ActiveRecord::Base
self.table_name = 'todos'
include ::EachBatch
end
def up
2018-08-06 05:05:44 -04:00
add_column(:todos, :group_id, :integer) unless group_id_exists?
add_concurrent_foreign_key :todos, :namespaces, column: :group_id, on_delete: :cascade
add_concurrent_index :todos, :group_id
change_column_null :todos, :project_id, true
end
def down
2018-08-06 05:05:44 -04:00
remove_foreign_key_without_error(:todos, column: :group_id)
remove_concurrent_index(:todos, :group_id)
remove_column(:todos, :group_id) if group_id_exists?
2018-08-06 05:05:44 -04:00
Todo.where(project_id: nil).each_batch { |batch| batch.delete_all }
change_column_null :todos, :project_id, false
end
private
def group_id_exists?
column_exists?(:todos, :group_id)
end
end