Rails与PostgreSQL：如何实现外键范围内的列自动递增？

当然可以实现！这种按外键分组的自增字段（比如每个门店的独立订单编号）在实际业务中非常常见，下面我一步步教你用Rails迁移和模型回调来搞定这个需求～

1. 给订单表添加自增字段的迁移 #

首先我们需要给orders表新增一个字段，比如store_order_number，用来存储每个门店下的自增序号。

先生成迁移文件：

rails generate migration AddStoreOrderNumberToOrders store_order_number:integer

然后打开生成的迁移文件，补充联合索引（提升查询效率，同时保证同门店内编号唯一）：

class AddStoreOrderNumberToOrders < ActiveRecord::Migration[7.0]
  def change
    add_column :orders, :store_order_number, :integer
    # 添加联合唯一索引，确保同一个门店下不会出现重复的订单编号
    add_index :orders, [:store_id, :store_order_number], unique: true
  end
end

运行迁移生效：

rails db:migrate

2. 在模型中添加自动赋值的回调 #

接下来在Order模型里添加before_create回调，在创建订单时自动计算当前门店的最大序号，再加1赋值给新订单：

class Order < ApplicationRecord
  belongs_to :store

  before_create :set_store_order_number

  private

  def set_store_order_number
    # 找到当前门店已有的最大订单编号，没有的话默认0，加1就是新编号
    max_number = Order.where(store_id: store_id).maximum(:store_order_number) || 0
    self.store_order_number = max_number + 1
  end
end

3. 处理并发写入的问题（可选但关键） #

上面的逻辑在单线程环境下没问题，但如果多个请求同时给同一个门店创建订单，可能会出现编号重复的情况。我们可以用数据库事务+行锁来避免这个问题：

修改set_store_order_number方法：

def set_store_order_number
  Order.transaction do
    # 锁定当前门店的所有订单记录，防止并发写入时的竞态条件
    Order.where(store_id: store_id).lock(true)
    max_number = Order.where(store_id: store_id).maximum(:store_order_number) || 0
    self.store_order_number = max_number + 1
  end
end

这样就能保证同一时间只有一个请求能修改该门店的订单编号，避免重复。

4. 给历史数据补全序号（如果需要） #

如果你的订单表已经有历史数据，需要给它们补全store_order_number，可以打开Rails控制台执行这段代码：

Store.find_each do |store|
  # 按订单创建顺序（id排序）给每个门店的订单分配序号
  store.orders.order(:id).each_with_index do |order, index|
    order.update!(store_order_number: index + 1)
  end
end

这样就能让历史订单也符合按门店自增的规则啦～

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