MongoDB聚合：将数组属性展平为根数组

Question

我正在开发一个使用$graphLookUp 递归查找cmets 线程的功能，我几乎拥有它。 （虽然有点复杂，但它确实有效！）

我需要的最后一步是： 不要将嵌套的 posteriorThread 作为根数组 ($$ROOT) 的属性，而是将其合并到根本身。

• 聚合代码：

const posteriorThread = await Comment.aggregate([
  {
    $match: {
      _id: post.threadDescendant
    }
  },

  {
    $graphLookup: {
      from: 'baseposts',
      startWith: '$threadDescendant',
      connectFromField: 'threadDescendant',
      connectToField: '_id',
      as: 'posteriorThread'
    }
  },

  {
    $unwind: '$posteriorThread'
  },

  {
    $sort: { 'posteriorThread.depth': 1 }
  },

  {
    $group: {
      _id: '$_id',
      posteriorThread: { $push: '$posteriorThread' },
      root: { $first: '$$ROOT' }
    }
  },

  {
    $project: {
      'root.posteriorThread': 0
    }
  },

  {
    $replaceRoot: {
      newRoot: {
        $mergeObjects: [
          {
            posteriorThread: '$posteriorThread'
          },
          '$root'
        ]
      }
    }
  }
]);

• 电流输出

 OUTPUT: posteriorThread
[
  {
    _id: '5f7eab40575e6fc56ee07604',
    onModel: 'BasePost',
    depth: 1,
    user: '5f5da45245c07cc06e51b09f',
    text: 'thread 0',
    isThread: true,
    threadDescendant: '5f7eabad575e6fc56ee07607',
    posteriorThread: [
      {
        _id: '5f7eabad575e6fc56ee07607',
        onModel: 'Comment',
        depth: 2,
        user: '5f5da45245c07cc06e51b09f',
        text: 'thread 1',
        isThread: true,
        threadDescendant: '5f7eac82575e6fc56ee07609'
      },
      {
        _id: '5f7eac82575e6fc56ee07609',
        onModel: 'Comment',
        depth: 3,
        user: '5f5da45245c07cc06e51b09f',
        text: 'thread 2',
        isThread: true
      }
    ]
  }
];

• 期望的输出

OUTPUT: posteriorThread 
[
  {
    _id: '5f7eab40575e6fc56ee07604',
    onModel: 'BasePost',
    depth: 1,
    user: '5f5da45245c07cc06e51b09f',
    text: 'thread 0',
    isThread: true,
    threadDescendant: '5f7eabad575e6fc56ee07607'
  },
  {
    _id: '5f7eabad575e6fc56ee07607',
    onModel: 'Comment',
    depth: 2,
    user: '5f5da45245c07cc06e51b09f',
    text: 'thread 1',
    isThread: true,
    threadDescendant: '5f7eac82575e6fc56ee07609'
  },
  {
    _id: '5f7eac82575e6fc56ee07609',
    onModel: 'Comment',
    depth: 3,
    user: '5f5da45245c07cc06e51b09f',
    text: 'thread 2',
    isThread: true
  }
];

我可以在常规 js 中聚合后完成此操作，但我更愿意在聚合中完成所有操作。 需要替换的部分是 mergeObjects 位并替换为其他内容或 group 聚合并采取不同的策略，但我不确定该放置什么。 另外，如果您有任何其他建议来使这个更清洁，我会全力以赴。

提前致谢。

Answer 1

真的很有挑战性。至少对我来说。而且非常有趣的案例。让我们试试我的解决方案。希望它有效..

db.test.aggregate([

    // PREVIOSU STEPS YOU ALREADY DID

    {
        $group: {
            _id: "$_id",
            items: {$push: "$$ROOT"},
            subItems: {$first: "$posteriorThread"}
        }
    },
    {
        $project: {
            "items.posteriorThread": 0
        }
    },
    {
        $addFields: {
            allItems: {
                $concatArrays: ["$items", "$subItems"]
            }
        }
    },
    {
        $group: {
            _id: null,
            mergedItems: {$push: "$allItems"}
        }
    },
    {
        $unwind: "$mergedItems"
    },
    {
        $unwind: "$mergedItems"
    },
    {
        $replaceRoot: {
          newRoot: "$mergedItems"
        }
    }

])

Answer 2

感谢@Sunil K Samanta 引导我朝着正确的方向前进。这不是最漂亮的解决方案，但它确实给了我正确的解决方案。

const posteriorThread = await Comment.aggregate([
  {
    $match: {
      _id: post.threadDescendant
    }
  },

  {
    $graphLookup: {
      from: 'baseposts',
      startWith: '$threadDescendant',
      connectFromField: 'threadDescendant',
      connectToField: '_id',
      as: 'posteriorThread'
    }
  },

  {
    $unwind: '$posteriorThread'
  },

  {
    $sort: { 'posteriorThread.depth': 1 }
  },

  {
    $group: {
      _id: '$_id',
      items: { $push: '$$ROOT.posteriorThread' },
      root: { $push: '$$ROOT' },
    },
  },

  {
    $project: {
      items: 1,
      root: { $slice: ['$$ROOT.root', 0, 1] },
    },
  },

  {
    $project: {
      'root.posteriorThread': 0,
    },
  },

  {
    $addFields: {
      allItems: {
        $concatArrays: ['$root', '$items'],
      },
    },
  },

  {
    $replaceRoot: {
      newRoot: {
        full_posterior: '$$ROOT.allItems',
      },
    },
  },
])
)[0].full_posterior;