Search code examples
mongodbgoaggregation-frameworkmgo

mongo aggregation query with mgo driver


I have the following query in mongodb:

db.devices.aggregate({
$match: {userId: "v73TuQqZykbxFXsWo", state: true}},
{
  $project: {
    userId: 1,
    categorySlug: 1,
    weight: { 
      $cond: [ 
        {"$or": [  
          {$eq: ["$categorySlug", "air_fryer"] }, 
          {$eq: ["$categorySlug", "iron"] } 
        ] }, 
      0, 1] } 
    } },  
    {$sort: {weight: 1}},
    { $limit : 10 }
);

I'm trying to write this in Go using the mgo driver but not able to wrap my head around this at all!

How do I translate this to a Go mgo query?


Solution

  • The examples on the docs would be sufficient to get started. However, if you are not familiar with golang, the $cond part could be a bit tricky. See below example code:

        collection := session.DB("dbName").C("devices")
    
        stage_match := bson.M{"$match":bson.M{"userId":"v73TuQqZykbxFXsWo", "state": true}}
    
        condition_weight := []interface{}{bson.M{"$or": []bson.M{
                           bson.M{"$eq": []string{"$categorySlug", "air_fryer"}},
                           bson.M{"$eq": []string{"$categorySlug", "iron"}},
        }}, 0, 1}
    
        stage_project:= bson.M{"$project": bson.M{"userId":1, "categorySlug":1, "weight": condition_weight}}
    
        stage_sort := bson.M{"$sort": bson.M{"weight":1}}
    
        stage_limit := bson.M{"$limit": 10}
    
        pipe := collection.Pipe([]bson.M{stage_match, stage_project, stage_sort, stage_limit})
    

    See also mgo: type Pipe