Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How can I get top n buckets for an aggregation and all other buckets combined into an "other" bucket?

Assume a collection with schema like as shown below:

{
    "customer" : <unique-id-for-customer>,
    "purchase" : <number>,
}

Now, I want to get the the top 5 customers (by purchase-queantity) and the 6th bucket is "others" which combines all the purchase quantity from other customers.

Basically, the output of aggregation should be like:

{ "_id" : "customer100", "purchasequantity" : 4000000 }
{ "_id" : "customer5", "purchasequantity" : 81800 }
{ "_id" : "customer4", "purchasequantity" : 40900 }
{ "_id" : "customer3", "purchasequantity" : 440 }
{ "_id" : "customer1", "purchasequantity" : 300 }
{"_id" : "others", "purchasequantity" : 29999}
like image 658
Nipun Talukdar Avatar asked Jan 11 '16 05:01

Nipun Talukdar


1 Answers

What you want is called weighting. To do that you need to add a weight to your documents by $projecting them and using the $cond operator, then sort them by "weight" in ascending other and by "purchasequantity" in descending order.

db.collection.aggregate([
    { "$project": { 
        "purchasequantity": 1, 
        "w": { 
            "$cond": [ { "$eq": [ "$_id", "others" ] }, 1, 0 ] 
        } 
    }}, 
    { "$sort": { "w": 1, "purchasequantity": -1 } } 
])

Which returns:

{ "_id" : "customer100", "purchasequantity" : 4000000, "w" : 0 }
{ "_id" : "customer5", "purchasequantity" : 81800, "w" : 0 }
{ "_id" : "customer4", "purchasequantity" : 40900, "w" : 0 }
{ "_id" : "customer3", "purchasequantity" : 440, "w" : 0 }
{ "_id" : "customer1", "purchasequantity" : 300, "w" : 0 }
{ "_id" : "others", "purchasequantity" : 29999, "w" : 1 }
like image 145
styvane Avatar answered Oct 18 '22 07:10

styvane