Golang操作MongoDB：插入重复_id时单命令获取对象可行吗？

可以直接通过MongoDB的FindOneAndUpdate命令结合upsert选项实现这个需求，无需先尝试插入再捕获重复键错误查询。

实现原理 #

通过指定_id作为查询条件，使用$setOnInsert定义仅在插入新文档时才生效的字段赋值（避免修改已有文档），开启Upsert选项后，MongoDB会自动完成以下逻辑：

• 如果_id对应的文档不存在：插入新文档并返回该文档
• 如果_id对应的文档已存在：直接返回已有文档

替换后的代码示例 #

import (
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo/options"
)

// 替换YourDocumentType为你的实际文档结构体类型
result := &YourDocumentType{}

// 构建查询条件：匹配目标_id
filter := bson.M{"_id": entry._id}

// 构建更新操作：仅在插入新文档时设置完整字段
update := bson.M{
    "$setOnInsert": entry,
}

// 配置选项：开启upsert，返回操作后的文档（插入的新文档或已存在的文档）
opts := options.FindOneAndUpdate().SetUpsert(true).SetReturnDocument(options.After)

// 执行命令并解码结果
err := s.coll.FindOneAndUpdate(ctx, filter, update, opts).Decode(result)
if err != nil {
    // 处理非重复键类的其他错误
    return nil, err
}

// result即为插入的新文档或已存在的重复文档
return result, nil

关键参数说明 #

• $setOnInsert：仅在插入新文档时执行字段赋值，不会对已存在的文档做任何修改
• SetUpsert(true)：开启自动判断逻辑，不存在则插入，存在则跳过更新
• SetReturnDocument(options.After)：指定返回操作完成后的文档，确保无论插入还是匹配已有文档，都能拿到目标对象

内容的提问来源于stack exchange，提问作者Killian G.