如何创建支持PostgreSQL GENERATED ALWAYS AS计算列的Rails迁移

要实现数据库层面的持久化计算列（不依赖Rails模型逻辑，支持跨技术栈调用），可以按照Rails版本选择对应实现方式：

方案1：Rails 7.0+ 原生支持写法 #

Rails 7开始内置了虚拟列/生成列的支持，直接在迁移文件中使用virtual方法即可，设置stored: true对应持久化存储的生成列：

class CreateYourTableName < ActiveRecord::Migration[7.0]
  def change
    create_table :your_table_name do |t|
      t.integer :a_column
      t.integer :b_column
      # 定义持久化生成列
      t.virtual :total1, type: :integer, as: "a_column + b_column", stored: true
      t.virtual :total2, type: :integer, as: "a_column - b_column", stored: true
      # 除法逻辑增加除零判断，避免运行时报错，以下为PostgreSQL写法，可根据数据库类型调整
      t.virtual :total3, type: :integer, as: "CASE WHEN b_column != 0 THEN a_column / b_column ELSE 0 END", stored: true
    end
  end
end

方案2：Rails 7 以下版本兼容写法 #

低版本Rails没有内置生成列支持，直接执行原生SQL实现即可，和标准SQL语法完全对齐：

class CreateYourTableName < ActiveRecord::Migration[6.1]
  def up
    execute <<-SQL
      CREATE TABLE your_table_name (
        id serial PRIMARY KEY,
        a_column integer,
        b_column integer,
        total1 integer GENERATED ALWAYS AS (a_column + b_column) STORED,
        total2 integer GENERATED ALWAYS AS (a_column - b_column) STORED,
        total3 integer GENERATED ALWAYS AS (CASE WHEN b_column != 0 THEN a_column / b_column ELSE 0 END) STORED
      );
    SQL
  end

  def down
    drop_table :your_table_name
  end
end

注意事项 #

• 不同数据库对生成列的语法支持略有差异，以上示例为PostgreSQL写法，MySQL 5.7+、SQLite 3.31+均支持同类语法，可根据实际使用的数据库调整细节
• 生成列是数据库层面自动维护的，不允许手动写入，建议在对应的Rails模型中添加只读声明，避免写入时抛出字段不可写错误：

class YourTableName < ApplicationRecord
  attr_readonly :total1, :total2, :total3
end

• 除法逻辑一定要处理b_column为0的边界情况，否则当b_column为空或0时会触发数据库报错。

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