CODEWITHSUNDEEP
node.jsmongodbmongooseaggregation-framework
Rana Deep
Rana Deep

Reputation: 617

Array intersection in mongodb with Aggregation framework

I have a collection with schema (mongoose) say :

{
  name : String,
  age  : Number,
  params : [Number]  // e.g. : params = [1,21,45,32,0] , usually they are very small arrays
}

The collection has 1000s of documents of this type

Say, I have a baseParams = [1,20,30,4,7];

I want to use aggregation and find the id of document whose params contain most number of numbers in baseParams array, like max(for each doc intersection(baseParams,params))

I finally need the _id of top 5 document sorted by age : 1

Any ideas anyone ?

Upvotes: 1

Views: 1787

Answers (1)

innoSPG
innoSPG

Reputation: 4656

what about this (in the mongo shell)? Simply translate to mongoose

db.ss.aggregate([
   {$unwind: '$params'},
   {$match: {params: {$in: [1,20,30,4,7]} } },
   {$group: {_id: {_id:"$_id", age: "$age"}, nb: {"$sum":1} } },
   {$sort: {nb:-1}},
   {$limit:5},
   {$project: {_id:"$_id._id", age:"$_id.age", nb: "$nb"} },
   {$sort:{age:1}}
 ])

The first stage $unwind break up the array field so that you have for each _id, a number of documents equal to the number of elt in params, each with a single value of the array params. $match select the document corresponding to what we want. $group group them back using the _id and the age as key and count the number of doc in each group; this corresponds exactly to the number of element in the intersection. $limit take the top five. $project and $sort do the rest of the job in sorting by age

Upvotes: 5

Related Questions