Mongo Aggregation - Using variables created in $project

user_2518241 0
Viewed 2501

Is it possible to use variables defined in the $project phase in that same phase?

For example, I have this aggregation pipeline:

pipeline = [
  { '$match': {} },
  {
    '$group': {
      '_id': '$_id',
      'n': { '$first': 'n' }
    }
  },
  {
    '$project': {
      'name': 1,
      'n': 1,
      'revenue': { '$multiply': ['$n', 2] },
      'cost': { '$multiply': ['$revenue', 0.25] }
    }
  }
]

I'd like to use the $revenue variable (that I defined in $project) in the same $project stage to compute the value of cost, but this does not work.

Is there any way I can do this easily and efficiently? I thought about doing several projections but I would need to project many variables (~25 variables) every time I need to compute one additional variable, and I have a bunch of variables to compute that are depending on each other (~5 variables), and my code would probably look bad since there would be a lot of projecting of the same variables. How should I go about this?

aggregation-framework pymongo mongodb python
3 Answers

Just landed here while searching for the same issue. My current solution is adding more $project stages and always take all other fields. For your example it would look like this:

pipeline = [{   
    '$match': {}
},
{
    '$group': {
        '_id': '$_id',
        'n': {'$first': 'n'}
},
{
    '$project': {
        'name': 1,
        'n': 1,
        'revenue': {'$multiply': ['$n', 2]}
    }
},
{
    '$project': {
        'name': 1,
        'n': 1,
        'revenue': 1,
        'cost': {'$multiply': ['$revenue', 0.25]}
    }
}]

I did not find any better solution yet if you want to keep all previous computed fields, without reusing expressions several times.

user_49458310

Starting in Mongo 3.4, you can use a series of $addFields stages rather than verbose $project stages.

This way, you can keep on projecting additional fields, some of which can re-use fields previously computed:

// { name: "a", n: 3 }
// { name: "b", n: 5 }
db.collection.aggregate([
  { $addFields: { revenue: { $multiply: ["$n", 2] } } },
  { $addFields: { cost: { $multiply: ["$revenue", 0.25] } } } // re-use "revenue"
])
// { name: "a", n: 3, revenue: 6,  cost: 1.5 }
// { name: "b", n: 5, revenue: 10, cost: 2.5 }
user_92971440
Related