MongoDB
 sql >> Base de données >  >> NoSQL >> MongoDB

La requête récursive Mongodb ne fonctionne pas comme prévu avec $ graphLookup

Vous pouvez utiliser $graphLookup et d'autres opérateurs de tableaux utiles,

  • $match filtrer que les enregistrements n'ont que sponsor est ""
  • $graphLookup pour obtenir les enregistrements enfants et le numéro de profondeur dans depthField level
  • $unwind déconstruire downline tableau et permet de ne pas supprimer les enfants vides
  • $sort par champ de niveau de profondeur level par ordre décroissant
  • $group par id champ et reconstruire downline tableau
  • $addFields maintenant trouver les enfants du niveau imbriqué et allouer à son niveau,
    • $reduce pour itérer la boucle de downline tableau.
    • initialiser le champ par défaut level la valeur par défaut est -1, presentChild est [], prevChild est [] pour les besoins des conditions
    • $let pour initialiser les champs :
      • prev selon la condition si les deux level sont égaux puis renvoient prevChild sinon retourner presentChild
      • current selon la condition si les deux level sont égaux puis renvoient presentChild sinon []
    • in pour retourner level champ et prevChild champ à partir de champs initialisés
      • presentChild $filter downline de prev array et return, fusionne les objets actuels avec downline tableau utilisant $mergeObjects et concat avec current tableau de let utilisant $concatArrays
  • $addFields pour renvoyer uniquement presentChild tableau car nous n'avions besoin que de ce tableau traité
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" } }
])

Aire de jeux