MongoDB如何将变量用作聚合操作符？

问题描述 #

现有一段MongoDB聚合代码，其中$project阶段的$cond判断使用了固定的$lt操作符，但需要替换为item.conditions中定义的operator变量（比如示例中的$gt、$lt）。尝试直接用'$$val.operator'作为键的写法无法生效，需要找到可行的实现方式。

原可运行代码：

let validations = [
    {
        "label": "kpis.avgLatency",
        "program":"ping-test",
        "multiple":false,
        "conditions" :[
            {
                "operator" : "$gt",
                "value": NumberInt(-107)
            },
            {
                "operator" : "$lt",
                "value": NumberInt(0)
            }
        ]
    }
]
validations.map((item) => {
    let test = db.task.aggregate([
      {$addFields: {validation: item}},
      {$match: { program: item.program}},
      {$project:{
        validation2: {
          $map: {
            input: item.conditions,
            as: "val",
            in: { $cond:{ if: { $lt: ['$validation.label', '$$val.value'] } , then: true, else: false} }
          }
        }
      }}
      ]).toArray();

    print(test);
})

尝试无效的写法：

in: { $cond:{ if: { '$$val.operator': ['$validation.label', '$$val.value'] } , then: true, else: false} }

解决方案 #

MongoDB聚合框架不支持直接将变量作为操作符使用，需要通过条件分支匹配操作符类型来实现动态判断。以下是两种可行方案：

方法1：使用$switch（推荐，可读性更强） #

validations.map((item) => {
    let test = db.task.aggregate([
      {$addFields: {validation: item}},
      {$match: { program: item.program}},
      {$project:{
        validation2: {
          $map: {
            input: item.conditions,
            as: "val",
            in: {
              $cond: {
                if: {
                  $switch: {
                    branches: [
                      { case: { $eq: ['$$val.operator', '$gt'] }, then: { $gt: ['$validation.label', '$$val.value'] } },
                      { case: { $eq: ['$$val.operator', '$lt'] }, then: { $lt: ['$validation.label', '$$val.value'] } },
                      { case: { $eq: ['$$val.operator', '$eq'] }, then: { $eq: ['$validation.label', '$$val.value'] } },
                      // 可根据需求添加更多操作符分支，比如$gte、$lte等
                    ],
                    default: false // 未匹配到操作符时的默认返回值
                  }
                },
                then: true,
                else: false
              }
            }
          }
        }
      }}
    ]).toArray();

    print(test);
})

方法2：使用嵌套$cond #

如果涉及的操作符种类较少，也可以用嵌套$cond实现：

validations.map((item) => {
    let test = db.task.aggregate([
      {$addFields: {validation: item}},
      {$match: { program: item.program}},
      {$project:{
        validation2: {
          $map: {
            input: item.conditions,
            as: "val",
            in: {
              $cond: {
                if: {
                  $cond: [
                    { $eq: ['$$val.operator', '$gt'] },
                    { $gt: ['$validation.label', '$$val.value'] },
                    { $cond: [
                      { $eq: ['$$val.operator', '$lt'] },
                      { $lt: ['$validation.label', '$$val.value'] },
                      { $cond: [
                        { $eq: ['$$val.operator', '$eq'] },
                        { $eq: ['$validation.label', '$$val.value'] },
                        false
                      ]}
                    ]}
                  ]
                },
                then: true,
                else: false
              }
            }
          }
        }
      }}
    ]).toArray();

    print(test);
})

说明 #

两种方案都是通过先判断operator的具体值，再执行对应操作符的逻辑，绕开MongoDB不能直接用变量作为操作符的限制。可根据实际使用的操作符种类，灵活扩展分支或嵌套层级。

内容的提问来源于stack exchange，提问作者daniegarcia254