Go+MongoDB：如何用完整结构体更新文档且不修改_id？

给定Go语言+MongoDB技术栈，定义如下Person结构体：

type Person struct {
    ID primitive.ObjectID `bson:"_id" json:"id,omitempty"`
    Name string
    LastName string
    Age int
    Address string
    City string
    FieldX string
    FieldY string
    // ... 其他字段
}

MongoDB已为Name、LastName、Address字段创建唯一索引。业务流程为：

1. 先插入仅包含这三个索引字段的文档，锁定记录避免Kafka重复消息扩散；
2. 后续处理完成后需更新文档，但受会话限制无法获取初始插入的_id。

现有两种方案：

• 方案1：3次DB操作，插入后查询_id，再用_id+完整结构体执行updateOne；
• 方案2：2次DB操作，用索引字段作为过滤条件，手动列出非索引、非_id字段执行updateOne。

疑问：是否存在中间方案，可直接使用完整Person结构体执行updateOne且不修改_id字段？

存在这样的中间方案，核心思路是用索引字段做过滤条件，将完整Person结构体转换为MongoDB的$set更新内容时自动排除_id和索引字段，既不用手动枚举字段，也仅需2次DB操作。

具体实现步骤 #

1. 构建过滤条件：用Name、LastName、Address三个唯一索引字段匹配目标文档；
2. 转换并清理结构体：将完整Person实例转为bson对象，移除_id和三个索引字段，剩余字段作为$set的更新内容；
3. 执行更新：以过滤条件匹配文档，用处理后的更新内容执行updateOne。

Go代码示例 #

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

func updatePerson(col *mongo.Collection, fullPerson Person) error {
    // 1. 用唯一索引字段构建查询过滤器
    filter := bson.M{
        "name":     fullPerson.Name,
        "lastname": fullPerson.LastName,
        "address":  fullPerson.Address,
    }

    // 2. 将完整结构体转为bson.M，排除无需更新的字段
    updateDoc, err := bson.Marshal(fullPerson)
    if err != nil {
        return err
    }
    var updateData bson.M
    if err := bson.Unmarshal(updateDoc, &updateData); err != nil {
        return err
    }
    // 移除_id和索引字段，避免覆盖唯一索引值
    delete(updateData, "_id")
    delete(updateData, "name")
    delete(updateData, "lastname")
    delete(updateData, "address")

    // 3. 构建updateOne的更新参数
    update := bson.M{"$set": updateData}

    // 4. 执行更新操作
    _, err = col.UpdateOne(context.TODO(), filter, update)
    return err
}

优化说明 #

如果觉得手动删除字段不够优雅，可封装通用反射函数，根据指定字段列表自动排除不需要更新的字段，减少重复代码。

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