Заявка
Може да се направи с помощта на рамка за агрегиране . Помислете за следващия тръбопровод за агрегиране
db.collectionName.aggregate([
{
$group:
{
"_id": null,
"ds": { $push: "$$ROOT" },
"cs": { $push: "$c" }
}
}, /* (1) */
{ $unwind: "$ds" }, /* (2) */
{
$project:
{
"_id": "$ds._id",
"c": "$ds.c",
"cs": { $slice: [ "$cs", "$ds._id" ] }
}
}, /* (3): */
{ $unwind: "$cs" }, /* (4) */
{
$group:
{
"_id": "$_id",
"c": { $first: "$c" },
"csum": { $sum: "$cs" }
}
}, /* (5) */
{
$group:
{
"_id": null,
"ds": { $push: "$$ROOT" },
"gteC":
{
$push:
{
$cond:
{
if: { "$gte": [ "$csum", SET_DESIRED_VALUE_FOR_C_HERE ] },
then: "$$ROOT",
else: { }
}
}
}
}
}, /* (6) */
{
$project:
{
"_id": 0,
"docs":
{
$filter:
{
input: "$ds",
"as": "doc",
cond: { $lte: [ "$$doc.csum", { $min: "$gteC.csum" } ] }
}
}
}
}, /* (7) */
{ $unwind: "$docs" }, /* (8) */
{ $project: { "_id": "$docs._id", "c": "$docs.c" } } /* (9) */
]);
Резултати
Обяснение
Основната идея зад него е да се конструира помощен масив за всеки документ в колекцията (етапи 1-3 )
{ "_id" : 1, "c" : 2 } -> cs = [ 2 ]
{ "_id" : 2, "c" : 6 } -> cs = [ 2, 6 ]
{ "_id" : 3, "c" : 1 } -> cs = [ 2, 6, 1 ]
използвайки $slice
оператор за агрегиране на масив и след това го заменете със сбор от всички елементи, които съдържа (етапи 4-5 )
{ "_id" : 1, "c" : 2 } -> csum = 2
{ "_id" : 2, "c" : 6 } -> csum = 8
{ "_id" : 3, "c" : 1 } -> csum = 9
използвайки $unwind
етап и $sum
групов акумулатор .
След това конструирайте друг помощен масив от документи с csum >= C
(етап 6 )
/* Ex. (C = 8) */
gteC = [ { "_id" : 3, "c" : 1, "csum" : 9 }, { "_id" : 2, "c" : 6, "csum" : 8 } ]
Последната стъпка е да извлечете всички документи с csum <= Min { gteC.csum }
. Това се прави с помощта на $filter
оператор за агрегиране на масив (етап 7 ).
Аз обаче не със сигурност това е най-ефективното тръбопровод за агрегиране (ще съм благодарен за всякакви предложения за подобрение), за да постигнете това, което искате.
PS Преди да тествате заявката, не забравяйте да промените името на колекцията и да замените SET_DESIRED_VALUE_FOR_C_HERE.