discourse/db/migrate/20220325064954_make_some_bo...

20 lines
796 B
Ruby

# frozen_string_literal: true
class MakeSomeBookmarkColumnsNullable < ActiveRecord::Migration[6.1]
def up
change_column_null :bookmarks, :post_id, true
execute "ALTER TABLE bookmarks ADD CONSTRAINT enforce_post_id_or_bookmarkable CHECK (
(post_id IS NOT NULL) OR (bookmarkable_id IS NOT NULL AND bookmarkable_type IS NOT NULL)
)"
end
def down
DB.exec("UPDATE bookmarks SET post_id = bookmarkable_id WHERE bookmarkable_type = 'Post'")
DB.exec(
"UPDATE bookmarks SET post_id = (SELECT id FROM posts WHERE topic_id = bookmarkable_id AND post_number = 1), for_topic = TRUE WHERE bookmarkable_type = 'Topic'",
)
change_column_null :bookmarks, :post_id, false
execute "ALTER TABLE bookmarks DROP CONSTRAINT enforce_post_id_or_bookmarkable"
end
end