MongoDB
 sql >> Datenbank >  >> NoSQL >> MongoDB

MongoDB-Gruppe und -Summe mit ID als Schlüssel

Mit MongoDb 3.6 und höher können Sie die Verwendung von $arrayToObject nutzen -Operator und ein $replaceRoot Pipeline, um das gewünschte Ergebnis zu erhalten. Sie müssten die folgende aggregierte Pipeline ausführen:

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

Bei älteren Versionen ist die Datei $cond -Operator in $group Der Pipeline-Schritt kann effektiv verwendet werden, um die Anzahl basierend auf dem Wert des Rollenfelds auszuwerten. Ihre gesamte Aggregationspipeline kann wie folgt aufgebaut sein, um das Ergebnis im gewünschten Format zu erzeugen:

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"
        }
    }
])

Wenn Sie die Rollen im Voraus nicht kennen und das Pipeline-Array dynamisch erstellen möchten, führen Sie im Kommentarpfad distinct aus Befehl auf dem Rollenfeld. Dadurch erhalten Sie ein Objekt, das eine Liste der unterschiedlichen Rollen enthält:

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

Anhand der obigen Liste können Sie nun Ihre Pipeline zusammenstellen, indem Sie ein Objekt erstellen, dessen Eigenschaften mit reduce() von JavaScript festgelegt werden Methode. Folgendes demonstriert dies:

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 );

Verwenden Sie diese beiden Dokumente in Ihrer endgültigen Aggregationspipeline als:

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

Sehen Sie sich die Demo unten an.

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>