MongoDB
 sql >> Database >  >> NoSQL >> MongoDB

Gruppo MongoDB e somma con id come chiave

Con MongoDb 3.6 e versioni successive, puoi sfruttare l'uso di $arrayToObject operatore e un $replaceRoot pipeline per ottenere il risultato desiderato. Dovresti eseguire la seguente pipeline aggregata:

db.users.aggregate([
    { 
        "$group": {
            "_id": { "$toLower": "$role" },
            "count": { "$sum": 1 }
        }
    },
    { 
        "$group": {
            "_id": null,
            "counts": {
                "$push": {
                    "k": "$_id",
                    "v": "$count"
                }
            }
        }
    },
    { 
        "$replaceRoot": {
            "newRoot": { "$arrayToObject": "$counts" }
        } 
    }    
])

Per le versioni precedenti, il $cond operatore nel $group il passaggio della pipeline può essere utilizzato in modo efficace per valutare i conteggi in base al valore del campo del ruolo. La tua pipeline di aggregazione complessiva può essere costruita come segue per produrre il risultato nel formato desiderato:

db.users.aggregate([    
    { 
        "$group": { 
            "_id": null,             
            "moderator_count": {
                "$sum": {
                    "$cond": [ { "$eq": [ "$role", "moderator" ] }, 1, 0 ]
                }
            },
            "superadmin_count": {
                "$sum": {
                    "$cond": [ { "$eq": [ "$role", "superadmin" ] }, 1, 0 ]
                }
            },
            "user_count": {
                "$sum": {
                    "$cond": [ { "$eq": [ "$role", "user" ] }, 1, 0 ]
                }
            },
            "admin_count": {
                "$sum": {
                    "$cond": [ { "$eq": [ "$role", "admin" ] }, 1, 0 ]
                }
            } 
        }  
    },
    {
        "$project": {
            "_id": 0, 
            "moderator": "$moderator_count",
            "superadmin": "$superadmin_count",
            "user": "$user_count",
            "admin": "$admin_count"
        }
    }
])

Dal percorso dei commenti, se non conosci i ruoli in anticipo e desideri creare l'array della pipeline in modo dinamico, esegui distinct comando nel campo del ruolo. Questo ti darà un oggetto che contiene un elenco dei ruoli distinti:

var result = db.runCommand ( { distinct: "users", key: "role" } )
var roles = result.values;
printjson(roles); // this will print ["moderator", "superadmin", "user",  "admin"]

Ora, dato l'elenco sopra, puoi assemblare la tua pipeline creando un oggetto che avrà le sue proprietà impostate usando reduce() di JavaScript metodo. Quanto segue lo dimostra:

var groupObj = { "_id": null },
    projectObj = { "_id": 0 }

var groupPipeline = roles.reduce(function(obj, role) { // set the group pipeline object 
    obj[role + "_count"] = {
        "$sum": {
            "$cond": [ { "$eq": [ "$role", role ] }, 1, 0 ]
        }
    };
    return obj;
}, groupObj );

var projectPipeline = roles.reduce(function(obj, role) { // set the project pipeline object 
    obj[role] = "$" + role + "_count";
    return obj;
}, projectObj );

Utilizza questi due documenti nella pipeline di aggregazione finale come:

db.users.aggregate([groupPipeline, projectPipeline]);

Controlla la demo qui sotto.

var roles = ["moderator", "superadmin", "user",  "admin"],
	groupObj = { "_id": null },
	projectObj = { "_id": 0 };

var groupPipeline = roles.reduce(function(obj, role) { // set the group pipeline object 
	obj[role + "_count"] = {
		"$sum": {
			"$cond": [ { "$eq": [ "$role", role ] }, 1, 0 ]
		}
	};
	return obj;
}, groupObj );

var projectPipeline = roles.reduce(function(obj, role) { // set the project pipeline object 
	obj[role] = "$" + role + "_count";
	return obj;
}, projectObj );

var pipeline = [groupPipeline, projectPipeline]

pre.innerHTML = JSON.stringify(pipeline, null, 4);
<pre id="pre"></pre>