如何在MongoDB聚合管道中将对象数组转换为嵌套对象

Question

我目前正在做一个我的数据库管理系统是 MongoDB 的项目。 我正在编写一个聚合管道，其中有几个阶段。 我目前正在某个特定阶段苦苦挣扎，我想获得以下输出。 MongoDB 有如此多的运算符表达式，我对使用哪一个来实现这一点感到困惑。 我有一个名为 Styles 和 Sku 的集合，如下所示：

// Schema for styles
const styles = mongoose.Schema({
  id: Number,
  productId: Number,
  name: String,
  sale_price: String,
  original_price: String,
  default_price: Boolean,
}, {
  strict: false,
});

// Schema for skus
const skus = mongoose.Schema({
  id: Number,
  styleId: Number,
  size: String,
  quantity: String,
}, {
  strict: false,
});

每种样式都可以有多个 SKU，一对多关系。 在我的聚合管道中，我使用 $lookup 查找该特定样式的所有 SKU，并在样式文档中添加一个名为 SKU 的新字段。 在 $lookup 之后我得到了这样的结果。

{
            "style_id": 1,
            "name": "Forest Green & Black",
            "original_price": "140",
            "sale_price": "0",
            "default?": true,
             "skus": [
                { 
                   "id": 37,
                   "styleId": 1,
                   "size": "XS",
                   "quantity": 16 
                },
                { 
                   "id": 38,
                   "styleId": 1,
                   "size": "S",
                   "quantity": 8
                } 
              ]  

}

这是预期的 $lookup 返回匹配的数组。 但我希望我的样式文档看起来像这样。

{
            "style_id": 1,
            "name": "Forest Green & Black",
            "original_price": "140",
            "sale_price": "0",
            "default?": true,
             "skus": {
                  "37": {
                    "styleId": 1,
                   "size": "XS",
                   "quantity": 16 
                  },
                  "38": {
                   "styleId": 1,
                   "size": "S",
                   "quantity": 8
                  }
               }
                    

}

有人可以知道如何在聚合管道中构造上述数据吗？ 任何帮助将不胜感激。 提前致谢。

Answer 1

您可以使用$map或$reduce将数组中的对象转换为表单

[
 {
  k: <key1>
  v: <value1>
 },
 {
  k: <key2>
  v: <value2>
 }
]

然后使用$arrayToObject将数组转换为一个对象，如：

{ 
    <key1>: <value1>,
    <key2>: <value2> 
}

总之，这可能看起来像：

{$addFields: {
    skus: {
      $arrayToObject: {
        $map: {
          input: "$skus",
          in: {
            k: {$toString: "$$this.id"},
            v: "$$this"
          }
        }
      }
    }
}}

操场