MongoDB实现数组元素存在则删除、不存在则添加的方法

嘿，这个需求我之前刚好碰到过，MongoDB 4.2及以上版本提供的聚合式更新能力，能完美解决这种「数组值开关」的场景——不用额外写判断逻辑，一条命令就能实现目标值存在则删除，不存在则添加的效果。

具体实现命令（Mongo Shell示例） #

假设你的集合名为forumComments，我们针对指定_id的文档操作：

db.forumComments.findOneAndUpdate(
  // 定位目标文档的查询条件
  { _id: ObjectId("5aebf141a805cd28433c414c") },
  // 用聚合管道实现条件更新逻辑
  [
    {
      $set: {
        usersLike: {
          $cond: {
            // 判断目标值是否在usersLike数组中
            if: { $in: ["1", "$usersLike"] },
            // 如果存在：过滤掉目标值，生成新数组
            then: { $filter: { input: "$usersLike", cond: { $ne: ["$$this", "1"] } } },
            // 如果不存在：把目标值追加到原数组末尾
            else: { $concatArrays: ["$usersLike", ["1"]] }
          }
        }
      }
    }
  ],
  // 返回更新后的文档（默认返回更新前的，所以需要显式指定）
  { returnDocument: "after" }
)

逻辑拆解 #

• 查询条件：这里用_id精准定位文档，你也可以根据业务需求换成forumId等其他条件
• 聚合式更新：MongoDB 4.2允许用聚合管道作为更新参数，让我们能基于文档当前的值做动态判断：
  • $cond：核心的条件判断函数，相当于代码里的if-else
  • $in：检查目标值（这里是"1"）是否存在于usersLike数组中
  • $filter：当值存在时，过滤掉数组中等于目标值的元素
  • $concatArrays：当值不存在时，把目标值和原数组合并，实现追加效果
• 返回设置：returnDocument: "after"会让命令直接返回更新后的文档，方便你验证结果

效果验证 #

• 原文档：{ "_id" : ObjectId("5aebf141a805cd28433c414c"), ..., "usersLike" : [ "1","2" ], ... }
• 执行命令后：返回的文档中usersLike会变成["2"]
• 再执行一次命令：因为"1"不在数组里，会被添加回去，usersLike变成["2", "1"]

如果是用Node.js、Python等编程语言的MongoDB驱动，语法只需稍作调整，核心的聚合管道逻辑是完全一致的。

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