sql >> Databasteknik >  >> NoSQL >> MongoDB

Flera sammanfogningsvillkor med operatorn $lookup

Vi kan göra flera anslutningsvillkor med $lookup aggregeringspipeline-operatör i version 3.6 och nyare.

Vi måste tilldela fältens värden till variabel med hjälp av let valfritt fält; du kommer sedan åt dessa variabler i pipeline fältsteg där du anger pipelinen som ska köras på samlingarna.

Observera att i $match steg använder vi $expr utvärderingsfrågeoperator för att jämföra fältens värde.

Det sista steget i pipelinen är $replaceRoot aggregeringspipelinestadiet där vi helt enkelt slår samman $lookup resultat med en del av $$ROOT dokument med $mergeObjects operatör.

db.collection2.aggregate([
       {
          $lookup: {
             from: "collection1",
             let: {
                firstUser: "$user1",
                secondUser: "$user2"
             },
             pipeline: [
                {
                   $match: {
                      $expr: {
                         $and: [
                            {
                               $eq: [
                                  "$user1",
                                  "$$firstUser"
                               ]
                            },
                            {
                               $eq: [
                                  "$user2",
                                  "$$secondUser"
                               ]
                            }
                         ]
                      }
                   }
                }
             ],
             as: "result"
          }
       },
       {
          $replaceRoot: {
             newRoot: {
                $mergeObjects:[
                   {
                      $arrayElemAt: [
                         "$result",
                         0
                      ]
                   },
                   {
                      percent1: "$$ROOT.percent1"
                   }
                ]
             }
          }
       }
    ]
)

Denna pipeline ger något som ser ut så här:

{
    "_id" : ObjectId("59e1ad7d36f42d8960c06022"),
    "user1" : 1,
    "user2" : 2,
    "percent" : 0.3,
    "percent1" : 0.56
}

Om du inte använder version 3.6+, kan du först gå med genom att använda ett av dina fält, låt säga "user1" och sedan avveckla arrayen av det matchande dokumentet med $unwind aggregeringsrörledningsoperatör. Nästa steg i pipelinen är $redact steg där du filtrerar bort de dokument där värdet på "user2" från samlingen "joined" och indatadokumentet inte är lika med $$KEEP och $$PRUNE systemvariabler. Du kan sedan omforma ditt dokument i $project skede.

db.collection1.aggregate([
    { "$lookup": { 
        "from": "collection2", 
        "localField": "user1", 
        "foreignField": "user1", 
        "as": "collection2_doc"
    }}, 
    { "$unwind": "$collection2_doc" },
    { "$redact": { 
        "$cond": [
            { "$eq": [ "$user2", "$collection2_doc.user2" ] }, 
            "$$KEEP", 
            "$$PRUNE"
        ]
    }}, 
    { "$project": { 
        "user1": 1, 
        "user2": 1, 
        "percent1": "$percent", 
        "percent2": "$collection2_doc.percent"
    }}
])

som producerar:

{
    "_id" : ObjectId("572daa87cc52a841bb292beb"),
    "user1" : 1,
    "user2" : 2,
    "percent1" : 0.56,
    "percent2" : 0.3
}

Om dokumenten i dina samlingar har samma struktur och du upplever att du utför denna operation ofta, bör du överväga att slå samman de två samlingarna till en eller infoga dokumenten i dessa samlingar i en ny samling.

db.collection3.insertMany(
    db.collection1.find({}, {"_id": 0})
    .toArray()
    .concat(db.collection2.find({}, {"_id": 0}).toArray())
)

Sedan $group dina dokument efter "användare1" och "användare2"

db.collection3.aggregate([
    { "$group": {
        "_id": { "user1": "$user1", "user2": "$user2" }, 
        "percent": { "$push": "$percent" }
    }}
])

vilket ger:

{ "_id" : { "user1" : 1, "user2" : 2 }, "percent" : [ 0.56, 0.3 ] }


  1. MongoDB Tutorial för 2022 – Lär dig vad är MongoDB?

  2. Grunderna för att distribuera en MongoDB Replica Set och Shards med Puppet

  3. Lagra redis hget-värden i en variabel i nodejs

  4. Dynamisk sortering med Redis