discourse/db/migrate/20240820123402_add_big_int_notifications_id.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

39 lines
1.0 KiB
Ruby
Raw Normal View History

DEV: Migrate notifications#id and related columns to bigint (#28584) * DEV: Migrate notifications#id to bigint (#28444) The `notifications.id` column is the most probable column to run out of values. This is because it is an `int` column that has only 2147483647 values and many notifications are generated on a regular basis in an active community. This commit migrates the column to `bigint`. These migrations do not use `ALTER TABLE ... COLUMN ... TYPE` in order to avoid the `ACCESS EXCLUSIVE` lock on the entire table. Instead, they create a new `bigint` column, copy the values to the new column and then sets the new column as primary key. Related columns (see `user_badges`, `shelved_notifications`) will be migrated in a follow-up commit. * DEV: Fix bigint notifications id migration to deal with public schema (#28538) Follow up to 799a45a291e9f2bd94278f565e58874458768079 * DEV: Migrate shelved_notifications#notification_id to bigint (#28549) DEV: Migrate shelved_notifications#notification_id to bigint The `notifications.id` has been migrated to `bigint` in previous commit 799a45a291e9f2bd94278f565e58874458768079. * DEV: Fix annotations (#28569) Follow-up to ec8ba5a0b9ff8dbefae74d6ec4f08dc13e418c6b * DEV: Migrate user_badges#notification_id to bigint (#28546) The `notifications.id` has been migrated to bigint in previous commit 799a45a291e9f2bd94278f565e58874458768079. This commit migrates one of the related columns, `user_badges.notification_id`, to `bigint`. * DEV: Migrate `User#seen_notification_id` to `bigint` (#28572) `Notification#id` was migrated to `bigint` in 799a45a291e9f2bd94278f565e58874458768079 * DEV: Migrate `Chat::NotificationMention#notification_id` to `bigint` (#28571) `Notification#id` was migrated to `bigint` in 799a45a291e9f2bd94278f565e58874458768079 --------- Co-authored-by: Alan Guo Xiang Tan <gxtan1990@gmail.com>
2024-08-29 23:06:55 +08:00
# frozen_string_literal: true
class AddBigIntNotificationsId < ActiveRecord::Migration[7.0]
def up
# Short-circuit if the table has been migrated already
result =
execute(
"SELECT data_type FROM information_schema.columns WHERE table_name = 'notifications' AND column_name = 'id' LIMIT 1",
)
data_type = result[0]["data_type"]
return if data_type.downcase == "bigint"
# Create new column
execute "ALTER TABLE notifications ADD COLUMN new_id BIGINT NOT NULL DEFAULT 0"
# Mirror new `id` values to `new_id`
execute <<~SQL.squish
CREATE FUNCTION mirror_notifications_id()
RETURNS trigger AS
$$
BEGIN
NEW.new_id = NEW.id;
RETURN NEW;
END;
$$
LANGUAGE plpgsql
SQL
execute <<~SQL.squish
CREATE TRIGGER notifications_new_id_trigger BEFORE INSERT ON notifications
FOR EACH ROW EXECUTE PROCEDURE mirror_notifications_id()
SQL
end
def down
raise ActiveRecord::IrreversibleMigration
end
end