MongoDB如何根据数组内用户ID添加contactStatus字段

最简解决方案（推荐） #

直接使用 $addFields 结合 $cond 和 $in 就能实现需求，代码如下：

db.collection.aggregate([
  {
    $addFields: {
      contactStatus: {
        $cond: {
          if: { $in: ["$_id", "$contacted._id"] },
          then: "Contacted",
          else: "Not Contacted"
        }
      }
    }
  }
])

这里 $contacted._id 会自动提取数组中所有元素的 _id 字段生成一个新数组，$in 判断文档自身的 _id 是否存在于这个数组中，最后通过 $cond 将布尔结果转换成你需要的字符串状态。

为什么你的 $reduce 没生效？ #

你之前用 $reduce 全返回 "Not Contacted"，大概率是逻辑写错了。正确的 $reduce 写法应该是遍历数组时记录是否匹配到目标 _id，示例如下：

db.collection.aggregate([
  {
    $addFields: {
      contactStatus: {
        $cond: {
          if: {
            $reduce: {
              input: "$contacted",
              initialValue: false,
              in: { $or: ["$$value", { $eq: ["$$this._id", "$_id"] }] }
            }
          },
          then: "Contacted",
          else: "Not Contacted"
        }
      }
    }
  }
])

这个逻辑里，$reduce 从 false 开始遍历数组，只要有一个元素的 _id 和文档自身 _id 相等，就将结果更新为 true，最后用 $cond 转换为字符串状态。你之前可能是初始值设置错误，或者没有用 $or 保留之前的匹配结果，导致一直返回 false。

关于多状态扩展的问题 #

你提到 $in 返回布尔值不合适，担心以后状态不止两种——其实完全可以通过嵌套 $cond 或者 $switch 扩展多状态逻辑。比如需要区分三种状态的示例：

db.collection.aggregate([
  {
    $addFields: {
      contactStatus: {
        $switch: {
          branches: [
            {
              case: { $in: ["$_id", "$contacted._id"] },
              then: "Contacted"
            },
            {
              case: { $gt: [{ $size: "$contacted" }, 0] },
              then: "Partially Contacted"
            }
          ],
          default: "Not Contacted"
        }
      }
    }
  }
])

这样可以灵活添加更多分支，不用担心状态扩展的问题。

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