如何在MongoDB中更新嵌套文档：避免重复新增数组元素

我将单个用户的聊天详情存储到MongoDB文档的userChats数组中，文档结构如下：

{
    "userId": "Ldu9N8DTIjfUApMlLUhuwzxZszt1",
    "userName": "Fake Star",
    "userChats": [
        {
            "chatId": "Ldu9N8DTIjfUApMlLUhuwzxZszt1Jaz5seHjL0Nct0IdSgixRW9wM8s2",
            "sender": "Ldu9N8DTIjfUApMlLUhuwzxZszt1",
            "receiver": "Jaz5seHjL0Nct0IdSgixRW9wM8s2",
            "receiverPic": "receiverPic",
            "lastMessage": "lastMessagelastMessagelastMessage",
            "timeStamp": 100
        },
        {
            "chatId": "Ldu9N8DTIjfUApMlLUhuwzxZszt1Jaz5seHjL0Nct0IdSgixRW9wM8s2",
            "sender": "Ldu9N8DTIjfUApMlLUhuwzxZszt1",
            "receiver": "Jaz5seHjL0Nct0IdSgixRW9wM8s2",
            "receiverPic": "receiverPic",
            "lastMessage": "lastMessagelastMessagelastMessage",
            "timeStamp": 1000
        }
    ]
}

我用Kotlin编写了操作MongoDB的代码：

override suspend fun insertChatDetail(chatDetail: ChatInfo) {

    val chatExists = userCollection.countDocuments(
        Document("_id", chatDetail.sender)
            .append("userChats.chatId", chatDetail.chatId)
    ) > 0

    if (chatExists) {
        println("debug Chat detail already exist, updating existing one")
        // Update the existing chat document with new information
        val updateDocument = Document("\$set", Document("userChats.$.chatId.timeStamp", chatDetail.timeStamp))
        userCollection.updateOne(Document("_id", chatDetail.sender), updateDocument)
    } else {
        println("debug Chat detail doesn't exist, creating new one")
        // Add the new chat to the userChats array
        val updateDocument = Document("\$push", Document("userChats", chatDetail))
        userCollection.updateOne(Document("_id", chatDetail.sender), updateDocument)
    }
}

遇到的问题：第一次插入timeStamp为100的记录，第二次插入同chatId但timeStamp为1000的记录时，代码新增了一条数组元素，而非将原记录的timeStamp从100更新为1000，请问如何修改代码实现预期效果？

问题根源 #

1. 更新条件缺失：原更新操作仅匹配用户_id，未指定要更新的数组元素（匹配chatId），导致$占位符无法定位到目标数组元素，更新操作无效，后续判断逻辑出错。
2. 更新路径错误：原代码中的userChats.$.chatId.timeStamp路径错误，timeStamp是chatId的同级字段，正确路径应为userChats.$.timeStamp。

修改后的代码 #

override suspend fun insertChatDetail(chatDetail: ChatInfo) {
    // 构建查询条件：匹配用户ID，且userChats数组中存在指定chatId的元素
    val query = Document("_id", chatDetail.sender)
        .append("userChats.chatId", chatDetail.chatId)
    
    // 尝试更新匹配的数组元素，可同步更新lastMessage等其他字段
    val updateResult = userCollection.updateOne(
        query,
        Document("\$set", Document(
            "userChats.$.timeStamp", chatDetail.timeStamp,
            "userChats.$.lastMessage", chatDetail.lastMessage
        ))
    )

    if (updateResult.modifiedCount == 0) {
        // 无匹配元素，执行新聊天插入
        println("debug Chat detail doesn't exist, creating new one")
        userCollection.updateOne(
            Document("_id", chatDetail.sender),
            Document("\$push", Document("userChats", chatDetail))
        )
    } else {
        println("debug Chat detail updated successfully")
    }
}

关键说明 #

• 精准定位数组元素：查询条件同时包含_id和userChats.chatId，确保$占位符准确指向数组中对应chatId的元素。
• 简化逻辑：先尝试更新，通过modifiedCount判断是否有元素被修改，无修改则执行插入，避免先查询再判断的冗余操作，减少数据库请求次数。
• 灵活更新字段：在$set中可添加其他需要同步刷新的字段（如lastMessage），一次更新完成多字段修改。

内容的提问来源于stack exchange，提问作者Harpreet Singh.8052