Byłeś bardzo blisko, ale oczywiście $eq
po prostu zwraca true/false
wartość, więc aby uzyskać tę wartość liczbową, potrzebujesz $cond
:
db.collection(collectionName).aggregate([
{ "$group" : {
"_id": "$item",
"good_count": {
"$sum": {
"$cond": [ { "$eq": [ "$rating", "good" ] }, 1, 0]
}
},
"neutral_count":{
"$sum": {
"$cond": [ { "$eq": [ "$rating", "neutral" ] }, 1, 0 ]
}
},
"bad_count": {
"$sum": {
"$cond": [ { "$eq": [ "$rating", "bad" ] }, 1, 0 ]
}
}
}}
])
Jako operator „trójargumentowy” $cond
przyjmuje warunek logiczny jako pierwszy argument (if), a następnie zwraca drugi argument, w którym ocena jest true
(wtedy) lub trzeci argument, gdzie false
(w przeciwnym razie). To sprawia, że true/false
wraca do 1
i 0
przesłać do $sum
odpowiednio.
Zauważ też, że wielkość liter jest wrażliwa dla $eq
. Jeśli masz różne przypadki, prawdopodobnie chcesz $toLower
w wyrażeniach:
"$cond": [ { "$eq": [ { "$toLower": "$rating" }, "bad" ] }, 1, 0 ]
Z nieco innej strony, poniższa agregacja jest zwykle bardziej elastyczna dla różnych możliwych wartości i krąży wokół sum warunkowych pod względem wydajności:
db.collection(collectionName).aggregate([
{ "$group": {
"_id": {
"item": "$item",
"rating": { "$toLower": "$rating" }
},
"count": { "$sum": 1 }
}},
{ "$group": {
"_id": "$_id.item",
"results": {
"$push": {
"rating": "$_id.rating",
"count": "$count"
}
}
}}
])
Zamiast tego dałoby to wynik w następujący sposób:
{
"_id": "item_1"
"results":[
{ "rating": "good", "count": 12 },
{ "rating": "neutral", "count": 10 }
{ "rating": "bad", "count": 67 }
]
}
To wszystkie te same informacje, ale nie trzeba było jawnie dopasowywać wartości i w ten sposób działa znacznie szybciej.