给现有表添加enum时如何编写迁移并保留表内容？

要给现有数据表添加PostgreSQL枚举类型（enum）且保留数据，你可以按以下步骤编写迁移：

核心思路 #

PostgreSQL的枚举类型需要先创建自定义类型，再给表添加使用该类型的字段；回滚时要先删除依赖字段，再删除枚举类型（符合Rails指南中关于删除enum需先移除依赖的要求）。整个过程完全可以保留现有数据。

迁移代码示例 #

假设你要给users表添加一个名为user_status的枚举类型，包含active、inactive、pending三个值，并新增status字段：

class AddStatusEnumToUsers < ActiveRecord::Migration[7.0]
  def up
    # 1. 创建PostgreSQL枚举类型
    execute <<-SQL
      CREATE TYPE user_status AS ENUM ('active', 'inactive', 'pending');
    SQL

    # 2. 给现有表添加枚举字段
    # 若现有数据需要先填充值，可先设null: true，填充后再修改约束
    add_column :users, :status, :user_status, null: false, default: 'active'

    # 可选：如果需要根据现有数据填充枚举值（示例：基于原有的active字段）
    # execute <<-SQL
    #   UPDATE users SET status = CASE WHEN active THEN 'active' ELSE 'inactive' END;
    # SQL
  end

  def down
    # 回滚时必须先删除依赖字段，再删除枚举类型
    remove_column :users, :status
    execute <<-SQL
      DROP TYPE user_status;
    SQL
  end
end

关键注意事项 #

• 数据保留：添加字段时可通过设置默认值，或批量更新现有记录来填充枚举值，全程不会丢失原有数据。
• 枚举类型修改：后续若要修改枚举值（新增/删除），需用execute执行SQL，比如新增值：ALTER TYPE user_status ADD VALUE 'suspended';
• 数据库兼容性：此方案仅适用于PostgreSQL，其他数据库（如MySQL）的enum实现逻辑不同，需调整。

内容的提问来源于stack exchange，提问作者Sig