将$facet结果与子文档合并，然后有条件地排除文档

Question

假设在$facet阶段之后，我得到了一个包含两个数组的结果：roomInfo和hotelInfo。

看起来是这样的：

{
  "roomInfo": [
    {
      _id: 'ab1',
      booked: 3
    },
    {
      _id: 'ab2',
      booked: 1
    }
  ],
  "hotelInfo": [
    {
      name: 'Radison Blue',
      roomDetails: [
        {
          _id: 'ab1',
          roomCount: 5
        },
        {
          _id: 'xy1',
          roomCount: 5
        }
      ]
    },
    {
      name: 'Intercontinental',
      roomDetails: [
        {
          _id: 'ab2',
          roomCount: 5
        }
      ]
    }
  ]
};

预期结果

我想要这样的输出：

[
  {
    name: 'Radison Blue',
    roomDetails: [
      {
        _id: 'ab1',
        roomCount: 5,
        booked: 3
      },
      {
        _id: 'xy1',
        roomCount: 5,
        booked: 0
      }
    ]
  },
  {
    name: 'Intercontinental',
    roomDetails: [
      {
        _id: 'ab2',
        roomCount: 5,
        booked: 1
      }
    ]
  }
];

基本上，在匹配roomInfo的ID后，将roomInfo中的预订属性添加到hotelInfo的roomDetails字段中。

如何实现这些用例？

谢谢

Answer 1

基本上，该方法将迭代酒店并相应地匹配每个房间，下面是一个快速工作代码示例：

db.collection.aggregate([
  {
    $unwind: "$hotelInfo"
  },
  {
    $project: {
      name: "$hotelInfo.name",
      "roomDetails": {
        $filter: {
          input: {
            $map: {
              input: "$hotelInfo.roomDetails",
              as: "info",
              in: {
                "$mergeObjects": [
                  "$$info",
                  {
                    "$arrayElemAt": [
                      {
                        $filter: {
                          input: "$roomInfo",
                          as: "room",
                          cond: {
                            $eq: [
                              "$$room._id",
                              "$$info._id"
                            ]
                          }
                        }
                      },
                      0
                    ]
                  }
                ]
              }
            }
          },
          as: "proccessedInfo",
          cond: {
            $ne: [
              "$$proccessedInfo.roomCount",
              "$$proccessedInfo.booked"
            ]
          }
        }
      }
    }
  }
])

蒙哥游乐场

话虽如此，你提到你想在将来打电话。当前的方法似乎不可伸缩，因为这些是“真实数据点”，也就是酒店。如果您的规模有一定的限制（不超过数千家酒店），那就好了。但如果不是这样，我建议你问另一个关于整个管道的问题，这样我们可以调整它以更好地工作。