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

A consulta recursiva do Mongodb não está funcionando conforme o esperado com $graphLookup


Você pode usar $graphLookup e outros operadores de matriz úteis,
  • $match filtro que os registros tenham apenas sponsor é ""
  • $graphLookup para obter registros filho e número de profundidade em depthField level
  • $unwind desconstruir downline 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 downline matriz
  • $addFields agora encontre os filhos de nível aninhado e aloque para seu nível,
    • $reduce para iterar o loop de downline 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 downline de prev array e return, mescla objetos atuais com downline array usando $mergeObjects e concatenar com current array de let usando $concatArrays
  • $addFields para retornar apenas presentChild array porque exigimos apenas esse array processado
db.collection.aggregate([
  { $match: { sponsor: "" } },
  {
    $graphLookup: {
      from: "collection",
      startWith: "$_id",
      connectFromField: "_id",
      connectToField: "sponsor",
      depthField: "level",
      as: "downline"
    }
  },
  {
    $unwind: {
      path: "$downline",
      preserveNullAndEmptyArrays: true
    }
  },
  { $sort: { "downline.level": -1 } },
  {
    $group: {
      _id: "$_id",
      sponsor: { $first: "$sponsor" },
      companyname: { $first: "$companyname" },
      downline: { $push: "$downline" }
    }
  },
  {
    $addFields: {
      downline: {
        $reduce: {
          input: "$downline",
          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",
                          {
                            downline: {
                              $filter: {
                                input: "$$prev",
                                as: "e",
                                cond: { $eq: ["$$e.sponsor", "$$this._id"] }
                              }
                            }
                          }
                        ]
                      }
                    ]
                  ]
                }
              }
            }
          }
        }
      }
    }
  },
  { $addFields: { downline: "$downline.presentChild" } }
])

Playground