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

Est-il possible de regrouper et d'additionner plusieurs colonnes avec le framework d'agrégation de MongoDB ?

Depuis MongoDB 3.4, cela peut être réalisé un peu plus simplement en utilisant plusieurs pipelines d'agrégation avec $facet .

tiré de la docs :

Ainsi, pour votre cas d'utilisation, cela serait réalisé par ce qui suit :

const aggregatorOpts = [
    { $match: { character: 'broquaint' } }, // Match the character
    {
        // Seperate into 2 or more pipes that will count class and
        // race seperatly
        $facet: {
            race: [
                // Group by race and get the count:
                // [
                //   {
                //     _id: 'Halfling',
                //     count: 3
                //   }
                //   {
                //     _id: 'Naga',
                //     count: 2
                //   }
                // ]

                // $sortByCount is the same as
                // { $group: { _id: <expression>, count: { $sum: 1 } } },
                // { $sort: { count: -1 } }

                { $sortByCount: '$race' },

                // Now we want to transform the array in to 1 document,
                // where the '_id' field is the key, and the 'count' is the value.
                // To achieve this we will use $arrayToObject. According the the
                // docs, we have to first rename the fields to 'k' for the key,
                // and 'v' for the value. We use $project for this:
                {
                    $project: {
                        _id: 0,
                        k: '$_id',
                        v: '$count',
                    },
                },
            ],
            // Same as above but for class instead
            class: [
                { $sortByCount: '$class' },
                {
                    $project: {
                        _id: 0,
                        k: '$_id',
                        v: '$count',
                    },
                },
            ],
        },
    },
    {
        // Now apply the $arrayToObject for both class and race.
        $addFields: {
            // Will override the existing class and race arrays
            // with their respective object representation instead.
            class: { $arrayToObject: '$class' },
            race: { $arrayToObject: '$race' },
        },
    },
];

db.races.aggregate(aggregatorOpts)

Ce qui produit ce qui suit :

[
  {
    "race": {
      "Halfling": 3,
      "Naga": 2
    },
    "class": {
      "Hunter": 3,
      "Rogue": 1,
      "Fighter": 1,
    }
  }
]

Si vous êtes satisfait du formatage de sortie fourni par @Asya, vous pouvez supprimer le $project et $addFields étapes, et laissez simplement le $sortByCount partie dans chaque sous-pipeline.

Avec ces nouvelles fonctionnalités, l'agrégation est beaucoup plus facile à étendre avec des comptes supplémentaires, il suffit d'ajouter un autre pipeline d'agrégation dans $facet .Il est même un peu plus facile de compter les sous-groupes, mais ce serait une question distincte.