Najpierw dodaj pole voteType
w każdym głosowaniu. To pole wskazuje jego typ. Mając to pole, nie musisz trzymać głosów w dwóch oddzielnych tablicach mlVoters
i egVoters
; możesz zamiast tego połączyć te tablice w jedną tablicę na dokument, a następnie rozwinąć.
W tym momencie masz jeden dokument na głos, z polem wskazującym, jakiego to rodzaju. Teraz wystarczy pogrupować się przez e-mail, a na etapie grupowym wykonać dwie sumy warunkowe, aby policzyć, ile głosów każdego typu przypada na każdy e-mail.
Na koniec dodajesz pole totalCount
jako suma pozostałych dwóch zliczeń.
db.documents.aggregate([
{
$addFields: {
mlVoters: {
$ifNull: [ "$mlVoters", []]
},
egVoters: {
$ifNull: [ "$egVoters", []]
}
}
},
{
$addFields: {
"mlVoters.voteType": "ml",
"egVoters.voteType": "eg"
}
},
{
$project: {
voters: { $concatArrays: ["$mlVoters", "$egVoters"] }
}
},
{
$unwind: "$voters"
},
{
$project: {
email: "$voters.email",
voteType: "$voters.voteType"
}
},
{
$group: {
_id: "$email",
mlCount: {
$sum: {
$cond: {
"if": { $eq: ["$voteType", "ml"] },
"then": 1,
"else": 0
}
}
},
egCount: {
$sum: {
$cond: {
"if": { $eq: ["$voteType", "eg"] },
"then": 1,
"else": 0
}
}
}
}
},
{
$addFields: {
totalCount: {
$sum: ["$mlCount", "$egCount"]
}
}
}
])