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

Question

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

// 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"
}
}
}
}
}}

游乐场