MongoDB更新嵌套文档后，如何返回指定ID的成员文档？

嘿，我完全懂你遇到的麻烦——更新了数组里指定UUID的成员角色后，却总是拿到第一个成员，这确实挺让人头疼的！其实问题出在默认的更新操作返回逻辑，还有你选的Distinct方法并不适用于这个场景，我给你两个靠谱的解决方案，都是基于mgo驱动的：

方法一：用FindAndModify原子性更新并返回目标元素 #

这是最推荐的方案，因为它是原子操作，能避免更新和查询之间的竞态问题（比如刚更新完，别的请求又改了这个文档）。它可以在更新完成后直接返回修改后的整个文档，然后你从中提取目标成员就行：

// 先定义你的文档结构，示例：
// type Group struct {
//     ID      bson.ObjectId `bson:"_id"`
//     Members []Member      `bson:"members"`
// }
// type Member struct {
//     ID   string `bson:"id"`
//     Role string `bson:"role"`
// }

// 构建查询条件：找到包含指定UUID成员的文档
query := bson.M{"members.id": targetUUID}
// 构建更新操作：更新匹配到的那个成员的role字段
update := bson.M{"$set": bson.M{"members.$.role": newRole}}

// 配置FindAndModify的选项，关键是ReturnNew设为true，返回更新后的文档
change := mgo.Change{
    Update:    update,
    ReturnNew: true,
}

var updatedGroup Group
err := yourCollection.Find(query).Apply(change, &updatedGroup)
if err != nil {
    // 这里处理错误，比如文档不存在之类的
    log.Fatal(err)
}

// 从更新后的Group里找到我们修改的那个成员
var updatedMember Member
for _, m := range updatedGroup.Members {
    if m.ID == targetUUID {
        updatedMember = m
        break
    }
}

// 现在updatedMember就是你要的更新后的成员啦

方法二：先更新，再查询匹配的数组元素 #

如果你不想用FindAndModify，也可以分两步走：先执行更新，再用投影操作符$只返回匹配的那个数组元素：

// 第一步：执行更新操作
err := yourCollection.Update(query, update)
if err != nil {
    log.Fatal(err)
}

// 第二步：查询并只返回匹配的那个成员
var result bson.M
// 使用Select{"members.$":1}来投影，只返回数组中第一个匹配条件的元素
err = yourCollection.Find(query).Select(bson.M{"members.$": 1}).One(&result)
if err != nil {
    log.Fatal(err)
}

// 提取更新后的成员，注意类型转换
updatedMember := result["members"].([]interface{})[0].(bson.M)

为什么之前的方法行不通？ #

• 普通的Update操作默认不会返回修改后的文档，就算你拿到原始文档，里面的成员也是更新前的状态；
• Distinct()是用来查询某个字段的去重值的，比如查所有成员的role有哪些，根本不是用来获取特定更新后数组元素的工具，所以肯定不符合你的需求。

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