使用MongoDB Go驱动NewReplaceOneModel丢失数据的原因及方案咨询

问题场景 #

第一种写法（执行后出现数据丢失）：

updateModels := make([]mongo.WriteModel, 0, len(documents))
for _, doc := range documents{
    filter := bson.D{primitive.E{Key: "fieldA", Value: fieldA}, primitive.E{Key: "fieldB", Value: fieldB}, primitive.E{Key: "_id", Value: doc.ID}}
    replace := bson.D{
        primitive.E{
            Key:   "column",
            Value: doc.Column,
        }, primitive.E{
            Key:   "content",
            Value: doc.Content,
        }}
    model := mongo.NewReplaceOneModel().SetFilter(filter).SetReplacement(replace)
    updateModels = append(updateModels, model)
}
if _, err = m.collection.BulkWrite(ctx, updateModels); err != nil {
    return
}

第二种写法（执行正常，无数据丢失）：

updateModels := make([]mongo.WriteModel, 0, len(documents))
for _, doc := range documents{
    filter := bson.D{primitive.E{Key: "fieldA", Value: fieldA}, primitive.E{Key: "fieldB", Value: fieldB}, primitive.E{Key: "_id", Value: doc.ID}}
    replace := bson.D{
                    primitive.E{Key: "_id", Value: doc.ID},
        primitive.E{Key: "fieldA", Value: fieldA},
        primitive.E{Key: "fieldB", Value: fieldB},
        primitive.E{
            Key:   "column",
            Value: doc.Column,
        }, primitive.E{
            Key:   "content",
            Value: doc.Content,
        }}
    model := mongo.NewReplaceOneModel().SetFilter(filter).SetReplacement(replace)
    updateModels = append(updateModels, model)
}
if _, err = m.collection.BulkWrite(ctx, updateModels); err != nil {
    return
}

1. 使用NewReplaceOneModel是否必须携带所有字段？ #

不是必须，但要明确：NewReplaceOneModel对应的是MongoDB的replaceOne操作，会用你传入的替换文档完全覆盖匹配到的原文档。如果替换文档中不包含原文档的某些字段，这些字段会直接从数据库文档中被移除。只有当你明确需要删除原文档中未列出的字段时，才适合只传部分字段。

2. 第一种写法导致数据丢失的原因是什么？ #

第一种写法里，替换文档只包含column和content两个字段，执行替换操作后，原文档中的_id、fieldA、fieldB等其他字段都会被删除，最终数据库里的文档只剩下这两个字段，所以看起来像是“数据丢失”。

第二种写法把所有需要保留的字段都放进了替换文档，相当于用包含全部字段的新文档覆盖原文档，因此原字段都被保留，不会出现丢失。

3. 改用NewUpdateOneModel是否是更优选择？ #

如果你的需求只是更新文档中的部分字段，而非完全替换整个文档结构，那NewUpdateOneModel确实是更优选择。

NewUpdateOneModel对应MongoDB的updateOne操作，使用$set等更新操作符仅修改指定字段，不会影响原文档中未被更新的字段，代码更简洁，也避免了因遗漏字段导致的数据丢失。示例写法如下：

updateModels := make([]mongo.WriteModel, 0, len(documents))
for _, doc := range documents{
    filter := bson.D{primitive.E{Key: "fieldA", Value: fieldA}, primitive.E{Key: "fieldB", Value: fieldB}, primitive.E{Key: "_id", Value: doc.ID}}
    // 使用$set仅更新需要修改的字段
    update := bson.D{
        primitive.E{
            Key: "$set",
            Value: bson.D{
                primitive.E{Key: "column", Value: doc.Column},
                primitive.E{Key: "content", Value: doc.Content},
            },
        },
    }
    model := mongo.NewUpdateOneModel().SetFilter(filter).SetUpdate(update)
    updateModels = append(updateModels, model)
}
if _, err = m.collection.BulkWrite(ctx, updateModels); err != nil {
    return
}

只有当你需要彻底替换整个文档的结构（比如大幅调整字段组成）时，NewReplaceOneModel才是合适的选择。

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