如何为MongoDB的computer集合计算并添加activeSetting属性？

要给每个computer文档添加符合规则的activeSetting字段，可以使用MongoDB 4.2+支持的带聚合管道的更新操作，直接在数据库层面完成关联匹配和字段更新。

核心逻辑 #

1. 关联tag集合，获取当前电脑对应标签的ObjectId（因为computer的tags存的是标签名称，而setting的tags存的是标签ID）。
2. 关联setting集合，筛选出与电脑同类别且共享标签的所有配置。
3. 对筛选出的配置按priority升序排序（优先级数值越小，优先级越高），取第一个配置的_id作为activeSetting。

执行命令 #

db.computer.updateMany(
  {},
  [
    // 关联tag集合，匹配同类别且名称在电脑tags中的标签
    {
      $lookup: {
        from: "tag",
        localField: "tags",
        foreignField: "name",
        let: { computerCategory: "$category" },
        pipeline: [
          { $match: { $expr: { $eq: ["$category", "$$computerCategory"] } } }
        ],
        as: "matchedTags"
      }
    },
    // 提取标签的ObjectId数组
    {
      $addFields: {
        tagIds: { $map: { input: "$matchedTags", as: "tag", in: "$$tag._id" } }
      }
    },
    // 关联setting集合，匹配同类别且共享标签的配置，并按优先级排序
    {
      $lookup: {
        from: "setting",
        let: {
          computerCategory: "$category",
          computerTagIds: "$tagIds"
        },
        pipeline: [
          {
            $match: {
              $expr: {
                $and: [
                  { $eq: ["$category", "$$computerCategory"] },
                  { $gt: [{ $size: { $setIntersection: ["$tags", "$$computerTagIds"] } }, 0] }
                ]
              }
            }
          },
          { $sort: { priority: 1 } }
        ],
        as: "matchedSettings"
      }
    },
    // 设置activeSetting：取优先级最高的配置ID，无匹配则设为null
    {
      $addFields: {
        activeSetting: {
          $cond: {
            if: { $gt: [{ $size: "$matchedSettings" }, 0] },
            then: { $first: "$matchedSettings._id" },
            else: null
          }
        }
      }
    },
    // 清理临时字段
    { $project: { matchedTags: 0, tagIds: 0, matchedSettings: 0 } }
  ]
)

关键细节说明 #

• $setIntersection用于判断setting的标签数组和电脑的标签ID数组是否有交集，确保两者共享至少一个标签。
• 排序规则按priority: 1升序，对应规则中“优先级最高”的要求（示例中priority:1的配置被选中）。
• 如果电脑没有符合条件的配置，activeSetting会被设为null，避免字段缺失。

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