node.js

mongodb

mongodb-query

aggregation-framework

Collection:

[
    { _id: "Foo", flag1: false, flag2: true, flag3: false },
    { _id: "Bar", flag1: true, flag2: false, flag3: true }
]

My question is, is it possible to call a method inside aggregate query?

aggregate({
    $project: {
        '_id': 1,
        'status' MyService.getStatus($flag1, $flag2, $flag3)
    }
});

If it is possible, what is the syntax of it? Result:

[
    { _id: "Foo", status: 'ok' },
    { _id: "Bar", status: 'broken' }
]

In my real world application I have 10 boolean flags per document. If the user gets this documents I would like to convert the flags and give them a meaning (for the user). E.g. consider a document represents a tire.

flag1 = true means tire have good pressure, false means low pressure
flag2 = true means depth of tire profile is good, false means little profile
and so on

So in summary I would like to say a tire is OK if

 flag1, flag2 are true and flag3 is false

and a tire needs to be replaced (BROKEN or REPLACE) when

flag1, flag2 are false and flag3 is true

When a document is returned to the user the flags should be removed. Instead we have the status field that says the tire is either OK or BROKEN.

Solution 1

External functions don't work with the aggregation framework. Everything is parsed to BSON on input, so no JavaScript or anything else is allowed. This is all basically processed from BSON "operator" definition to native C++ code implementation so it is really fast.

What this comes down to is "converting" your expected logic to what the aggregation framework can process. There are in fact "logical" operators such as $or and $and that work in this context:

db.collection.aggregate([
    { "$project": {
       "_id": 1,
       "status": {
           "$cond": [
               { "$or": [
                   // Your first set of rules requires "false" for "flag1" or 
                   // "flag2" and "true" for "flag3"
                   { "$and": [
                       { "$not": [
                           { "$or": [ "$flag1", "$flag2" ] },
                       ]},
                       "$flag3"
                   ]},
                   // Your second set of rules requires "true" for "flag1" or 
                   // "flag2" and "false" for "flag3"
                   { "$and": [
                       { "$or": [ "$flag1", "$flag2" ] },
                       { "$not": [ "$flag3" ] }
                   ]},
               ]},
               "ok",
               "broken"
           ]
       }
    }}
])

So no external functions, just implement the logic with the operators that the aggregation framework supplies. In addition to the basic logical implementations there is $not to "reverse" the ligic and $cond which acts as a "ternary" in order to provide a different result from true/false evaluation.

Solution 2

The aggregate call can be passed a callback function, which is called after the aggregation has completed.

function getValuesAndMessges( params, callback ) {

  db.collection.aggregate([
    { "$project": {
       "_id": 1,
       "flag1": { "$first": "$flag1" },
       "flag2": { "$first": "$flag2" },
       "flag3": { "$first": "$flag3" },
    }}
  ], function( err, results ) {
    if ( !err ) {
      result.forEach( result => {
        // process items in results here, setting a value
        // using the actual logic for writing message ...
        if( flag1 )
          result.message = "broken";
        else
          result.messsge = 'OK';
      });
    }
    callback(err, results);
  });
}

this way each of your aggregated items (based on your conditions / parameters) will have a message property (or whatever properties you choose to write) set and you can use them in your calling function.

Solution 3

Yes we can call function in aggregation project with simple way.

let getStatus = (flag) => {
    return flag=='ok' ? 'ok' :'broken';
}
aggregate({
    $project: {
        '_id': 1,
        'status': getStatus($flag3)
    }
});

Solution 4

Inserting a function to an aggregation pipeline is possible since mongoDB version 4.4. for example:

db.collection.aggregate([
  {
    $project: {
      status: {
        $function: {
          "body": "function (flagA, flagB, flagC) {if (flagA && flagB && !flagC) { return 'ok'} return 'broken'}",
          "args": [
            "$flag1",
            "$flag2",
            "$flag3"
          ],
          "lang": "js"
        }
      }
    }
  }
])

Playground example