MongoDB聚合运算符：$switch

MongoDB聚合运算符：$switch

$switch聚合运算符评估一系列case表达式，当找到一个计算结果为 true的表达式时， $switch会执行指定的表达式并脱离控制流。

语法

$switch: {
   branches: [
      { case: <expression>, then: <expression> },
      { case: <expression>, then: <expression> },
      ...
   ],
   default: <expression>
}

branches数组中的对象必须仅包含case字段和then字段。

• branches：控制分支文档的数组，数组至少要有一个元素，分支为包含以下字段的文档：
  • case：可解析为布尔值的表达式，如果不是布尔类型，则强制转换为布尔类型
  • then：可以是任何合法的表达式
• default：可选字段，没有分支表达式为true时的路径。尽管是可选的，但如果未指定默认值并且没有分支为true时$switch将返回错误。

使用

各种case语句不需要相互排斥，$switch会执行第一个结果为true的分支，如果没有分支为true，$switch执行默认选项。

以下情况会导致$switch出错：

• 分支字段缺失或不是至少有一个条目的数组。
• 分支数组中的对象不包含case字段。
• 分支数组中的对象不包含then字段。
• 分支数组中的对象包含case或then以外的字段。
• 未指定默认值，且case值均为true

第一个符合条件的分支

{
   $switch: {
      branches: [
         { case: { $eq: [ 0, 5 ] }, then: "equals" },
         { case: { $gt: [ 0, 5 ] }, then: "greater than" },
         { case: { $lt: [ 0, 5 ] }, then: "less than" }
      ]
   }
}

结果：

"less than"

缺省分支

{
   $switch: {
      branches: [
         { case: { $eq: [ 0, 5 ] }, then: "equals" },
         { case: { $gt: [ 0, 5 ] }, then: "greater than" }
      ],
      default: "Did not match"
   }
}

结果：

"Did not match"

强制转换case为布尔值

{
   $switch: {
      branches: [
         { case: "this is true", then: "first case" },
         { case: false, then: "second case" }
      ],
      default: "Did not match"
   }
}

结果：

"First case"

举例

grades集合有下列文档：

{ "_id" : 1, "name" : "Susan Wilkes", "scores" : [ 87, 86, 78 ] }
{ "_id" : 2, "name" : "Bob Hanna", "scores" : [ 71, 64, 81 ] }
{ "_id" : 3, "name" : "James Torrelio", "scores" : [ 91, 84, 97 ] }

下面的聚合操作使用$switch运算符根据每个学生的平均分数显示特定消息：

db.grades.aggregate( [
  {
    $project:
      {
        "name" : 1,
        "summary" :
        {
          $switch:
            {
              branches: [
                {
                  case: { $gte : [ { $avg : "$scores" }, 90 ] },
                  then: "Doing great!"
                },
                {
                  case: { $and : [ { $gte : [ { $avg : "$scores" }, 80 ] },
                                   { $lt : [ { $avg : "$scores" }, 90 ] } ] },
                  then: "Doing pretty well."
                },
                {
                  case: { $lt : [ { $avg : "$scores" }, 80 ] },
                  then: "Needs improvement."
                }
              ],
              default: "No scores found."
            }
         }
      }
   }
] )

操作返回下面的结果：

{ "_id" : 1, "name" : "Susan Wilkes", "summary" : "Doing pretty well." }
{ "_id" : 2, "name" : "Bob Hanna", "summary" : "Needs improvement." }
{ "_id" : 3, "name" : "James Torrelio", "summary" : "Doing great!" }