Bạn có thể sử dụng $ graphLookup và các toán tử mảng hữu ích khác,
-
$match
lọc mà các bản ghi chỉ cóparent_id
lànull
-
$graphLookup
để lấy các bản ghi con và số chiều sâu ở cấp độ DeepFieldlevel
-
$unwind
giải cấu trúcchildren
mảng và cho phép không xóa phần con trống -
$sort
theo trường cấp độ sâulevel
theo thứ tự giảm dần -
$group
bởiid
trường và tạo lạichildren
mảng
db.collection.aggregate([
{ $match: { parent_id: null } },
{
$graphLookup: {
from: "collection",
startWith: "$id",
connectFromField: "id",
connectToField: "parent_id",
depthField: "level",
as: "children"
}
},
{
$unwind: {
path: "$children",
preserveNullAndEmptyArrays: true
}
},
{ $sort: { "children.level": -1 } },
{
$group: {
_id: "$id",
parent_id: { $first: "$parent_id" },
name: { $first: "$name" },
type: { $first: "$type" },
category: { $first: 1 },
children: { $push: "$children" }
}
},
-
$addFields
bây giờ tìm các cấp độ con lồng nhau và phân bổ cho cấp độ của nó,- $ giảm để lặp lại vòng lặp của
children
mảng. - khởi tạo trường mặc định
level
giá trị mặc định là -1,presentChild
là [],prevChild
là [] cho mục đích điều kiện -
$let
để khởi tạo các trường:-
prev
theo điều kiện nếu cảlevel
bằng nhau thì trả vềprevChild
nếu không thì trả vềpresentChild
-
current
theo điều kiện nếu cảlevel
bằng nhau thì trả vềpresentChild
nếu không thì []
-
-
in
để trả vềlevel
trường vàprevChild
trường từ các trường đã khởi tạo-
presentChild
$filter
children
từprev
mảng và trả về, hợp nhất các đối tượng hiện tại vớichildren
mảng sử dụng$mergeObjects
và nối vớicurrent
mảng let sử dụng$concatArrays
-
- $ giảm để lặp lại vòng lặp của
-
$addFields
để chỉ trả lạipresentChild
mảng vì chúng tôi chỉ yêu cầu mảng đã xử lý đó
{
$addFields: {
children: {
$reduce: {
input: "$children",
initialValue: { level: -1, presentChild: [], prevChild: [] },
in: {
$let: {
vars: {
prev: {
$cond: [
{ $eq: ["$$value.level", "$$this.level"] },
"$$value.prevChild",
"$$value.presentChild"
]
},
current: {
$cond: [{ $eq: ["$$value.level", "$$this.level"] }, "$$value.presentChild", []]
}
},
in: {
level: "$$this.level",
prevChild: "$$prev",
presentChild: {
$concatArrays: [
"$$current",
[
{
$mergeObjects: [
"$$this",
{
children: {
$filter: {
input: "$$prev",
as: "e",
cond: { $eq: ["$$e.parent_id", "$$this.id"] }
}
}
}
]
}
]
]
}
}
}
}
}
}
}
},
{
$addFields: {
id: "$_id",
children: "$children.presentChild"
}
}
])