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

Comment trouver l'intersection d'ensembles d'ensembles entre les documents d'une même collection dans MongoDB ?

Utiliser le cadre d'agrégation pour obtenir le résultat souhaité. L'agrégation opérateur d'ensemble qui ferait la magie est $setIntersection .

Le pipeline d'agrégation suivant répond à vos attentes :

db.test.aggregate([
    {
        "$match": {
            "_id": { "$in": [1, 3] }
        }
    },
    {
        "$group": {
            "_id": 0,
            "set1": { "$first": "$set" },
            "set2": { "$last": "$set" }
        }
    },
    {
        "$project": { 
            "set1": 1, 
            "set2": 1, 
            "commonToBoth": { "$setIntersection": [ "$set1", "$set2" ] }, 
            "_id": 0 
        }
    }
])

Sortie :

/* 0 */
{
    "result" : [ 
        {
            "set1" : [1,2,3,4,5],
            "set2" : [1,2,5,10,22],
            "commonToBoth" : [1,2,5]
        }
    ],
    "ok" : 1
}

MISE À JOUR

Pour que trois documents ou plus soient croisés, vous aurez besoin de $reduce opérateur pour aplatir les tableaux. Cela vous permettra de croiser n'importe quel nombre de tableaux, donc au lieu de simplement faire une intersection des deux tableaux des docs 1 et 3, cela s'appliquera également à plusieurs tableaux.

Envisagez d'exécuter l'opération d'agrégation suivante :

db.test.aggregate([
    { "$match": { "_id": { "$in": [1, 3] } } },
    {
        "$group": {
            "_id": 0,
            "sets": { "$push": "$set" },
            "initialSet": { "$first": "$set" }
        }
    },
    {
        "$project": {
            "commonSets": {
                "$reduce": {
                    "input": "$sets",
                    "initialValue": "$initialSet",
                    "in": { "$setIntersection": ["$$value", "$$this"] }
                }
            }
        }
    }
])