MongoDB如何在单查询中对数组对象执行$set或$inc操作？

你的需求是实现「存在则递增计数，不存在则新增对象」的更新逻辑，不能用覆盖整个数组的$set写法（会丢失已有数据），下面提供两种可行方案：

方案一：分两次查询（简单直观，适合大多数场景） #

先尝试递增已有频道的计数，若未匹配到对应频道，再新增计数对象。

1. 尝试更新已有频道的消息计数：

const updateResult = await db.collection("users").updateOne(
  { userId: userId, "messages.channelId": channelId },
  { $inc: { "messages.$.messageCount": 1 } }
);

2. 若没有匹配到可修改的文档（updateResult.modifiedCount === 0），则新增频道计数对象：

if (updateResult.modifiedCount === 0) {
  await db.collection("users").updateOne(
    { userId: userId },
    { $push: { messages: { channelId: channelId, messageCount: 1 } } },
    { upsert: true } // 若用户文档不存在，自动创建
  );
}

方案二：单聚合管道查询（原子性更强，一次完成） #

利用MongoDB的聚合管道更新功能，在单个查询中完成判断和更新，适合需要严格原子性的场景：

await db.collection("users").updateOne(
  { userId: userId },
  [
    {
      $set: {
        messages: {
          $cond: {
            // 判断数组中是否已存在目标channelId
            if: { $in: [channelId, "$messages.channelId"] },
            // 存在则遍历数组，匹配频道计数+1，其余元素保持不变
            then: {
              $map: {
                input: "$messages",
                as: "msg",
                in: {
                  $cond: {
                    if: { $eq: ["$$msg.channelId", channelId] },
                    then: {
                      channelId: "$$msg.channelId",
                      messageCount: { $add: ["$$msg.messageCount", 1] }
                    },
                    else: "$$msg"
                  }
                }
              }
            },
            // 不存在则将新频道对象追加到数组末尾
            else: {
              $concatArrays: [
                "$messages",
                [{ channelId: channelId, messageCount: 1 }]
              ]
            }
          }
        }
      }
    }
  ],
  { upsert: true } // 若用户文档不存在，自动创建并初始化数组
);

注意事项 #

• 方案一的两次查询虽非严格原子操作，但在Discord消息场景下并发冲突概率极低，且实现简单易维护。
• 方案二通过聚合管道实现了服务器端原子更新，避免了客户端两次查询的间隙问题。
• 绝对不要使用你原先的$set: { messages: [...] }写法，这会直接覆盖整个messages数组，导致历史计数数据丢失。

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