Mongoose 聚合在同一文档中按父 ID 分组

Question

我在 mongoDB 呼叫公司中有一个简单的文档，其架构如下：

const companySchema = mongoose.Schema({
  _id: mongoose.Schema.Types.ObjectId,
  id: { type: String },
  name: { type: String },
  parentId: { type: String, default: null },
  type: { type: Number },
  active: { type: Boolean, default: true },
  number: { type: String },
});

然后我有一个查询来获取所有公司并按 parentId 将它们分组并将所有子公司添加到分支机构属性。

我在某种程度上完成了这项工作，但需要最后的努力才能获得所需的一切。

这是我目前的代码：

Company.aggregate([
    {
      $match: {
        active: true,
      },
    },

    {
      $project: {
        _id: { $cond: ['$parentId', '$parentId', '$id'] },
        name: 1,
        id: '$id',
        parent: '$parent',
        parentId: { $ifNull: ['$parentId', false] },
      },
    },

    { $sort: { _id: 1, parentId: -1 } },
    {
      $group: {
        _id: '$_id',
        name: { $min: { $cond: ['$parentId', '$name', false] } },
        branches: {
          $push: '$$ROOT',
        },
      },
    },
    {
      $project: {
        name: 1,
        branches: '$branches',
      },
    },
  ])

这是我目前得到的输出：

{
  "_id": "29689760",
  "name": "Branch 2",
  "branches": [
    { "name": "Test (Head Office)", "_id": "29689760", "id": "29689760", "parentId": false },
    { "name": "Branch 1", "_id": "29689760", "id": "29694015", "parentId": "29689760" },
    { "name": "Branch 2", "_id": "29689760", "id": "29694016", "parentId": "29689760" },
    { "name": "Branch 3", "_id": "29689760", "id": "29694017", "parentId": "29689760" },
    { "name": "Branch 4", "_id": "29689760", "id": "29694018", "parentId": "29689760" }
  ]
}

我需要的是返回所有关于母公司和分支机构以排除母公司的主要对象，即

{
  "_id": "29689760",
  "name": "Test (Head Office)"",
  "branches": [
//the next line will be missins
    //{ "name": "Test (Head Office)", "_id": "29689760", "id": "29689760", "parentId": false },
    { "name": "Branch 1", "_id": "29689760", "id": "29694015", "parentId": "29689760" },
    { "name": "Branch 2", "_id": "29689760", "id": "29694016", "parentId": "29689760" },
    { "name": "Branch 3", "_id": "29689760", "id": "29694017", "parentId": "29689760" },
    { "name": "Branch 4", "_id": "29689760", "id": "29694018", "parentId": "29689760" }
  ]
}

我尝试将$lookup 添加为聚合的第一个参数，但这会将信息添加到每个分支机构，而不仅仅是母公司。

有什么帮助吗？

Answer 1

当_id 和parentId 中的条件处于$project 和$group 阶段时，您的查询会令人困惑，

但根据你的说法，

我需要的是返回所有关于母公司和分支机构以排除母公司的主要对象

我会在您的查询中建议一个解决方案，您可以在$group 阶段更改此设置，

• push 分支时设置条件，如果parentId 为false 则push root 文档，否则$$REMOVE 排除push

    {
      $group: {
        _id: '$_id',
        name: { $min: { $cond: ['$parentId', '$name', false] } },
        branches: {
          $push: {
            $cond: ['$parentId', '$$ROOT', '$$REMOVE']
          }
        },
      },
    },

---

第二个选项可以使用$lookup 加入并从同一个集合中获取分支，

• $match 你的必填字段和parentId 应该为空
• $sort _id 按升序排列
• $lookup 当前收藏
• $project 显示必填字段

await Company.aggregate([
  {
    $match: {
      active: true,
      parentId: null
    }
  },
  { $sort: { _id: 1 } },
  {
    $lookup: {
      from: 'company', // change your original collection name
      localField: 'id', 
      foreignField: 'parentId',
      as: 'branches'
    }
  },
  {
    $project: {
      _id: 0,
      name: 1,
      branches: {
        id: 1,
        name: 1,
        _id: 1,
        parentId: 1
      }
    }
  }
])

Playground