MongoDB
 sql >> Base de Dados >  >> NoSQL >> MongoDB

Faça árvore de nós com tabela recursiva com Express e Mongo


Você pode usar $graphLookup e outros operadores de matriz úteis,
  • $match filtro que os registros tenham apenas parent_id é null
  • $graphLookup para obter registros filho e número de profundidade em depthField level
  • $unwind desconstruir children array e permitir não remover filhos vazios
  • $sort por campo de nível de profundidade level em ordem decrescente
  • $group por id campo e reconstruir children matriz
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 agora encontre os filhos de nível aninhado e aloque para seu nível,
    • $reduce para iterar loop de children matriz.
    • inicializar o campo padrão level o valor padrão é -1, presentChild é [], prevChild é [] para fins de condições
    • $let para inicializar campos:
      • prev conforme a condição se ambos level são iguais então retorne prevChild caso contrário, retorne presentChild
      • current conforme a condição se ambos level são iguais então retorne presentChild caso contrário []
    • in para retornar level campo e prevChild campo de campos inicializados
      • presentChild $filter children de prev array e return, mescla objetos atuais com children array usando $mergeObjects e concatenar com current array de let usando $concatArrays
  • $addFields para retornar apenas presentChild array porque exigimos apenas esse array processado
  {
    $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"
    }
  }
])

Playground