MongoDB 文档中多个数组大小的总和

发布于03月16日

在"医院"Collection 中，有医院每个部门的文件.每个文件都有详细信息，如科室名称、分配给该科室的doctor 和护士.文档 struct 如下所示:

{
    "department": "emergency_care",
    "doctors": ["d1", "d2", "d3"],
    "nurses": ["n1", "n2", "n3", "n4", "n5"]
},
{
    "department": "child_health",
    "doctors": ["d4", "d5"],
    "nurses": ["n6"]
},
{
    "department": "first_aid_room",
    "nurses": ["n7", "n8"]
}

如何计算住院总人数，这是所有科室所有doctor 和护士的总和.

预期输出:13

推荐答案

您可以try 以下查询:

第一组全部(使用_id: null)获得doctor 和护士的总数.
如果值不是数组，则此处使用$cond到$sum 0，否则使用数组大小.
然后使用$project级输出这两个值的总和.

db.collection.aggregate([
  {
    "$group": {
      "_id": null,
      "totalDoctors": {
        "$sum": {
          "$cond": {
            "if": {"$isArray": ["$doctors"]},
            "then": {"$size": "$doctors"},
            "else": 0
          }
        }
      },
      "totalNurses": {
        "$sum": {
          "$cond": {
            "if": {"$isArray": ["$nurses"]},
            "then": {"$size": "$nurses"},
            "else": 0
          }
        }
      }
    }
  },
  {
    "$project": {
      "_id": 0,
      "nPersons": {
        "$add": ["$totalDoctors","$totalNurses"
        ]
      }
    }
  }
])

例here