在使用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 更新文檔數組并替換為替換文檔數組
