在使用MongoDB时，有时我们需要根据不同的键来更新多条记录，这个问题可能会让人感到困惑。不过，幸好在MongoDB中，我们可以使用Bulk Write操作来实现这个目标。Bulk Write是一种批量写入操作，可以在单个操作中执行多个更新、插入或删除操作。在这篇文章中，我将向大家介绍如何使用Bulk Write来根据Mongo中的不同键更新多条记录。

问题内容

如果我有类似于以下内容的内容...

collection.insertmany(context.todo(), []interface{}{
   bson.m{ "_id" : 1, "member" : "abc123", "status" : "p" },
   bson.m{ "_id" : 2, "member" : "xyz123", "status" : "a" },
   bson.m{ "_id" : 3, "member" : "lmn123", "status" : "p" },
   bson.m{ "_id" : 4, "member" : "pqr123", "status" : "d" },
   bson.m{ "_id" : 5, "member" : "ijk123", "status" : "p" },
   bson.m{ "_id" : 6, "member" : "cde123", "status" : "a" },
} )

是否可以在一个 insertmany 查询中应用以下更新？

[{"_id" : "1", "status" : "P0-A0"},
 {"_id" : "2", "status" : "P0-A1"},
 {"_id" : "3", "status" : "P0-A2"},
 {"_id" : "4", "status" : "P0-A3"},
 {"_id" : "5", "status" : "P0-A4"},
 {"_id" : "6", "status" : "P0-A5"}]

如果是这样，用 golang 会如何完成？

具体来说，使用 collection.updatemany(context.todo(), filter, update)，我的 filter 和 update 会有什么？

感谢您的帮助。

解决方法

你不能用一个来做到这一点collection.updatemany() 调用，因为您无法对不同的匹配文档应用不同的更新文档。您必须多次调用 collection.updatemany()，每个不同的更新文档一次。

如果您想通过一次调用高效地完成此操作，您可以使用 collection.bulkwrite()。您必须准备一个不同的 mongo.writemodel每个文档更新。

它可能是这样的：

wm := []mongo.writemodel{
    mongo.newupdateonemodel().setfilter(bson.m{"_id": "1"}).setupdate(bson.m{"$set": bson.m{"status": "p0-a0"}}),
    mongo.newupdateonemodel().setfilter(bson.m{"_id": "2"}).setupdate(bson.m{"$set": bson.m{"status": "p0-a1"}}),
    mongo.newupdateonemodel().setfilter(bson.m{"_id": "3"}).setupdate(bson.m{"$set": bson.m{"status": "p0-a2"}}),
    mongo.newupdateonemodel().setfilter(bson.m{"_id": "4"}).setupdate(bson.m{"$set": bson.m{"status": "p0-a3"}}),
    mongo.newupdateonemodel().setfilter(bson.m{"_id": "5"}).setupdate(bson.m{"$set": bson.m{"status": "p0-a4"}}),
    mongo.newupdateonemodel().setfilter(bson.m{"_id": "6"}).setupdate(bson.m{"$set": bson.m{"status": "p0-a5"}}),
}

上面的切片文字中有太多“重复”，您可以使用辅助函数捕获它们：

create := func(id, newstatus string) *mongo.updateonemodel {
    return mongo.newupdateonemodel().
        setfilter(bson.m{"_id": id}).
        setupdate(bson.m{"$set": bson.m{"status": newstatus}})
}

wm := []mongo.writemodel{
    create("1", "p0-a0"),
    create("2", "p0-a1"),
    create("3", "p0-a2"),
    create("4", "p0-a3"),
    create("5", "p0-a4"),
    create("6", "p0-a5"),
}

此外，如果更新中有可以轻松定义的逻辑，请使用循环而不是列出所有元素：

var wm []mongo.writemodel
for i := 1; i <= 6; i++ {
    newstatus := fmt.sprintf("p0-a%d", i-1)
    wm = append(wm, mongo.newupdateonemodel().
        setfilter(bson.m{"_id": strconv.itoa(i)}).
        setupdate(bson.m{"$set": bson.m{"status": newstatus}}),
    )
}

您可以通过一次调用执行所有更新，如下所示：

res, err := coll.BulkWrite(ctx, wm)

查看相关内容：mongodb 更新文档数组并替换为替换文档数组