如何通过GORM向PostgreSQL批量Upsert并返回新增记录ID？

核心思路：自定义RETURNING子句 #

GORM默认的RETURNING仅支持直接指定字段，但可以通过自定义Clause绕过限制，无需原生SQL即可在单次请求内完成操作。

步骤1：定义自定义Returning子句 #

先实现一个包含CASE WHEN xmax = 0 THEN id ELSE NULL END逻辑的自定义Clause：

type CustomReturning struct{}

func (c CustomReturning) Name() string {
    return "RETURNING"
}

func (c CustomReturning) Build(builder clause.Builder) {
    builder.WriteString("CASE WHEN xmax = 0 THEN id ELSE NULL END")
}

步骤2：结合Create与OnConflict使用 #

在批量Create操作中，通过Clauses方法注入自定义子句，同时配置Upsert规则：

var records []YourModel
// 假设records是待批量插入的模型实例

var returnedIDs []*int64 // 接收返回结果：新增记录的ID有值，更新记录的ID为nil

db.Clauses(
    clause.OnConflict{
        Columns:   []clause.Column{{Name: "your_unique_column"}}, // 指定冲突判断的唯一字段
        DoUpdates: clause.AssignmentColumns([]string{"col1", "col2"}), // 冲突时需要更新的字段
    },
    CustomReturning{},
).Create(&records).Scan(&returnedIDs)

// 过滤出非nil的ID，即为新增记录的ID
var newIDs []int64
for _, id := range returnedIDs {
    if id != nil {
        newIDs = append(newIDs, *id)
    }
}

方案优势 #

这种方式既保留了GORM的模型映射能力，又通过自定义Clause实现了复杂的返回逻辑，全程只需一次数据库请求，无需额外查询或提前获取现有记录。

补充说明 #

你提到的提前获取现有记录再区分的方案效率较低，而上述方法直接利用PostgreSQL的xmax字段（标识是否发生更新操作）判断记录状态，返回结果可直接区分新增与更新，过滤后即可得到目标ID。

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