如何将MongoDB集合文档聚合为嵌套层级结构？

我正在开发一个评论存储应用，目前需要把MongoDB Posts 集合中的评论文档，聚合整理成带有嵌套回复的层级结构。下面是我的集合样本数据：

// Posts集合样本数据
{ "_id" : "1", "text" : "hey" }
{ "_id" : "2", "text" : "hello", "replyTo" : "1" }
{ "_id" : "3", "text" : "What's up?", "replyTo" : "2" }
{ "_id" : "4", "text" : "How are you", "replyTo" : "1" }
{ "_id" : "5", "text" : "Knock, knock!" }

我期望得到的结果是把顶级评论（没有replyTo字段的）作为父节点，所有关联的回复嵌套到对应的replies数组中，类似这样的格式：

[
  {
    "_id": "1",
    "text": "hey",
    "replies": [
      {
        "_id": "2",
        "text": "hello",
        "replies": [
          { "_id": "3", "text": "What's up?" }
        ]
      },
      { "_id": "4", "text": "How are you" }
    ]
  },
  { "_id": "5", "text": "Knock, knock!" }
]

解决方案：使用MongoDB聚合管道实现嵌套回复 #

可以通过MongoDB的聚合管道来实现这个需求，核心思路是先构建评论的父子关系，再递归嵌套子回复。下面是完整的聚合命令：

db.Posts.aggregate([
  // 第一步：将所有文档按replyTo分组，建立父评论到子评论的映射
  {
    $group: {
      _id: { $ifNull: ["$replyTo", "root"] },
      children: { $push: "$$ROOT" }
    }
  },
  // 第二步：将分组后的结果转换为键值对，方便后续递归查询
  {
    $group: {
      _id: null,
      commentsMap: {
        $arrayToObject: {
          $map: {
            input: "$$ROOT",
            as: "item",
            in: { k: "$$item._id", v: "$$item.children" }
          }
        }
      }
    }
  },
  // 第三步：递归构建嵌套回复结构
  {
    $project: {
      nestedComments: {
        $function: {
          body: function(commentsMap) {
            // 定义递归函数，为每个评论添加replies
            function buildTree(commentId) {
              const comments = commentsMap[commentId] || [];
              return comments.map(comment => ({
                ...comment,
                replies: buildTree(comment._id)
              }));
            }
            // 返回顶级评论（replyTo为null的，对应分组时的"root"键）
            return buildTree("root");
          },
          args: ["$commentsMap"],
          lang: "js"
        }
      }
    }
  },
  // 第四步：展开嵌套评论数组，得到最终结果
  { $unwind: "$nestedComments" },
  { $replaceRoot: { newRoot: "$nestedComments" } }
])

步骤解释 #

• $group阶段1：把所有评论按replyTo字段分组，如果没有replyTo（顶级评论）就归到root组，同时把同组的评论放到children数组里，这样就建立了父ID到子评论的映射。
• $group阶段2：把上一步的分组结果转换成一个对象，键是父ID，值是对应的子评论数组，方便后续快速查找。
• $project阶段：使用$function自定义JavaScript函数，递归遍历评论映射，为每个评论添加replies字段，把它的子评论嵌套进去，最后返回顶级评论的树状结构。
• 最后两步：把嵌套的顶级评论数组展开，替换成根文档，得到我们想要的层级化结果。

如果你的MongoDB版本不支持$function（需要4.4及以上版本），也可以用$graphLookup来实现，不过递归的逻辑会稍微复杂一点，上面的方案是最直观易懂的。

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