Rails使用upsert_all批量更新数据遇PostgreSQL非空约束问题求助

你遇到的核心问题是upsert_all的机制导致的：它本质是PostgreSQL的INSERT ... ON CONFLICT语句，即使你指定了update_only: :column_a，当冲突发生（记录已存在）时，数据库仍会尝试处理所有字段——如果你的数据里没提供relationship_id，PostgreSQL会默认将其设为null，触发非空约束报错。

方案1：合并现有relationship_id到更新数据 #

先批量查询出目标记录的relationship_id，合并到你的更新数据中，再执行upsert_all：

# 提取需要更新的记录ID
target_ids = data.map { |item| item["id"].to_i }
# 批量获取ID对应的relationship_id，转成哈希便于查找
relationship_map = MyModel.where(id: target_ids).pluck(:id, :relationship_id).to_h

# 合并数据，补充relationship_id字段
updated_data = data.map do |item|
  item.merge("relationship_id" => relationship_map[item["id"].to_i])
end

# 执行批量更新
MyModel.upsert_all(updated_data, update_only: [:column_a])

这种方式逻辑直观，符合Rails的ORM使用习惯，适合数据量中等的场景。

方案2：使用PostgreSQL原生UPDATE FROM语句 #

直接写原生SQL批量更新，完全不触碰relationship_id字段，效率最高：

# 构造VALUES子句的内容
values_clause = data.map { |item| "(#{item["id"]}, #{item["column_a"]})" }.join(", ")

# 执行原生SQL更新
sql = <<~SQL
  UPDATE my_models
  SET column_a = updates.column_a
  FROM (VALUES #{values_clause}) AS updates(id, column_a)
  WHERE my_models.id = updates.id::integer;
SQL

ActiveRecord::Base.connection.execute(sql)

一次SQL操作完成所有更新，避免了额外的查询开销，适合大数据量场景。

方案3：用update_all配合CASE语句 #

正确使用update_all也能实现批量更新，且不会影响其他字段：

# 构造CASE条件语句
case_clause = data.map do |item|
  "WHEN id = #{item["id"]} THEN #{item["column_a"]}"
end.join(" ")

# 批量更新目标记录的column_a
MyModel.where(id: data.map { |d| d["id"] }).update_all(
  "column_a = CASE #{case_clause} END"
)

这种方式也是一次SQL操作，语法简洁，不需要额外查询数据。

• upsert_all：即使指定update_only，它仍会处理INSERT分支的字段完整性检查，缺失relationship_id会导致null值触发约束；
• 错误的update_all用法：如果之前的update_all没有正确构造条件或更新逻辑，可能误操作了字段，导致同样的约束报错。

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