Rails中upsert_all方法同时处理插入与更新时的使用问题

你遇到的问题核心是：给新记录传入了nil的id值，违反了Postgres主键非空约束。其实upsert_all的正确用法里，新记录根本不需要指定id字段——Postgres会自动帮你生成自增ID，而已有记录则需要通过id来匹配，触发更新逻辑。

下面是具体的正确实现方式：

1. 构造正确的数据集 #

把需要处理的记录统一构造为哈希数组，区分两种场景的字段设置：

• 对于已有记录：哈希中包含id（主键）以及需要更新的字段（比如name、updated_at等）；
• 对于新记录：哈希中只包含需要插入的字段（不要包含id键），让Postgres自动生成主键ID。

举个简单示例：

records = [
  # 已有记录：带id，用于匹配更新
  { id: 1, name: "Updated Product", updated_at: Time.current },
  # 新记录：不带id，插入时自动生成主键
  { name: "New Product", created_at: Time.current, updated_at: Time.current }
]

Product.upsert_all(records, unique_by: :id)

2. 关键参数说明 #

• unique_by: :id：指定用主键id来检测冲突——当插入的记录id已存在时，执行更新操作；不存在则执行插入。
• 绝对不要给新记录设置id: nil，直接省略这个键即可，Postgres会通过内置序列自动生成自增ID。

3. 批量处理ActiveRecord实例的技巧 #

如果你的数据是从ActiveRecord模型实例中获取的（既有已保存的旧实例，又有新建的实例），可以这样批量构造数据集：

# 已有记录：从数据库查询得到的实例
existing_products = Product.where(category: "electronics")
# 新记录：新建的未保存实例
new_products = [Product.new(name: "Wireless Headphones"), Product.new(name: "Portable Charger")]

# 构造upsert专用的哈希数组
upsert_records = existing_products.map do |product|
  # 先给旧实例赋值需要更新的字段
  product.assign_attributes(stock_count: product.stock_count + 10)
  # 只保留需要的字段：id + 要更新的字段
  product.attributes.slice("id", "name", "stock_count", "updated_at")
end + new_products.map do |product|
  # 新实例只保留插入需要的字段，跳过id
  product.attributes.slice("name", "stock_count", "created_at", "updated_at")
end

Product.upsert_all(upsert_records, unique_by: :id)

为什么之前的方式会报错？ #

当你给新记录设置id: nil时，upsert_all会把nil作为值写入SQL的INSERT语句，而Postgres的主键id字段默认带有NOT NULL约束，自然就触发了PG::NotNullViolation错误。省略id键的话，SQL语句里不会包含id字段，Postgres就会自动调用序列填充这个值。

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