Question

我正在评估MongoDB聚合框架如何满足我们的需求，因为我们当前正在SQL Server上运行。我在执行特定查询时遇到了困难：

假设我有以下伪记录（在sql表中建模为列，在mongodb集合中建模为完整文档）

{
   name: 'A',
   timespent: 100,
},
{
   name: 'B',
   timespent: 200,
},
{
   name: 'C',
   timespent: 300,
},
{
   name: 'D',
   timespent: 400,
},
{
   name: 'E',
   timespent: 500,
}

我想将timepent字段分组到范围并计算出现次数，以便我得到例如以下伪记录：

results{
   0-250: 2,
   250-450: 2,
   450-650: 1
}

请注意，这些范围（250,450和650）是动态的，并且可能会被用户随时间改变。在SQL中，我们使用以下内容提取结果：

select range, COUNT(*) as total from (
select case when Timespent <= 250 then '0-250'
when Timespent <= 450 then '200-450'
else '450-600' end as range
from TestTable) as r
group by r.range

再次注意，这个sql是由我们的应用程序动态构建的，以适应任何时候可用的特定范围。

我正在努力在mongodb聚合框架中找到适当的构造来执行此类查询。我可以通过在管道中插入$ match来查询单个范围的结果（即获取单个范围的结果），但我不知道如何在单个管道查询中提取所有范围及其计数。

Answer 1

与aggregation framework中的“case”SQL语句相对应的是$ cond运算符（请参阅manual）。 $ cond语句可以嵌套来模拟“when-then”和“else”，但我选择了另一种方法，因为它更容易阅读（并生成，见下文）：我将使用$ concat运算符来编写范围字符串，然后用作分组键。

所以对于给定的集合：

db.xx.find()
{ "_id" : ObjectId("514919fb23700b41723f94dc"), "name" : "A", "timespent" : 100 }
{ "_id" : ObjectId("514919fb23700b41723f94dd"), "name" : "B", "timespent" : 200 }
{ "_id" : ObjectId("514919fb23700b41723f94de"), "name" : "C", "timespent" : 300 }
{ "_id" : ObjectId("514919fb23700b41723f94df"), "name" : "D", "timespent" : 400 }
{ "_id" : ObjectId("514919fb23700b41723f94e0"), "name" : "E", "timespent" : 500 }

聚合（硬编码）看起来像这样：

db.xx.aggregate([
  { $project: {
    "_id": 0,
    "range": {
      $concat: [{
        $cond: [ { $lte: ["$timespent", 250] }, "range 0-250", "" ]
      }, {
        $cond: [ { $and: [
          { $gte: ["$timespent", 251] }, 
          { $lt:  ["$timespent", 450] } 
        ] }, "range 251-450", "" ]
      }, {
        $cond: [ { $and: [
          { $gte: ["$timespent", 451] }, 
          { $lt:  ["$timespent", 650] } 
        ] }, "range 450-650", "" ]
      }]
    }
  }},
  { $group: { _id: "$range", count: { $sum: 1 } } },
  { $sort: { "_id": 1 } },
]);

结果是：

{
    "result" : [
        {
            "_id" : "range 0-250",
            "count" : 2
        },
        {
            "_id" : "range 251-450",
            "count" : 2
        },
        {
            "_id" : "range 450-650",
            "count" : 1
        }
    ],
    "ok" : 1
}

为了生成聚合命令，您必须将“范围”投影构建为JSON对象（或者您可以生成一个字符串，然后使用JSON.parse（string））

生成器看起来像这样：

var ranges = [ 0, 250, 450, 650 ];
var rangeProj = {
  "$concat": []
};

for (i = 1; i < ranges.length; i++) {
  rangeProj.$concat.push({
    $cond: {
      if: {
        $and: [{
          $gte: [ "$timespent", ranges[i-1] ]
        }, {
          $lt: [ "$timespent", ranges[i] ]
        }]
      },
      then: "range " + ranges[i-1] + "-" + ranges[i],
      else: ""
    }
  })
}

db.xx.aggregate([{
  $project: { "_id": 0, "range": rangeProj }
}, {
  $group: { _id: "$range", count: { $sum: 1 } }
}, {
  $sort: { "_id": 1 }
}]);

将返回与上面相同的结果。

Answer 2

从MongoDB 3.4开始，我们可以使用$switch运算符在$project阶段执行多切换语句。

$group管道运营商将文档按＆＃34;范围＆＃34;并返回＆＃34; count＆＃34;对于使用$sum累加器运算符的每个组。

db.collection.aggregate(
    [  
        { "$project": { 
            "range": { 
                "$switch": { 
                    "branches": [ 
                        { 
                            "case": { "$lte": [ "$timespent", 250 ] }, 
                            "then": "0-250" 
                        }, 
                        { 
                            "case": { 
                                "$and": [ 
                                    { "$gt": [ "$timespent", 250 ] }, 
                                    { "$lte": [ "$timespent", 450 ] } 
                                ] 
                            }, 
                            "then": "251-450" 
                        }, 
                        { 
                            "case": { 
                                "$and": [ 
                                    { "$gt": [ "$timespent", 450 ] }, 
                                    { "$lte": [ "$timespent", 650 ] } 
                                ] 
                            }, 
                            "then": "451-650" 
                        } 
                    ], 
                    "default": "650+" 
                } 
            } 
        }}, 
        { "$group": { 
            "_id": "$range", 
            "count": { "$sum": 1 } 
        }}
    ]
)

我们的藏品中包含以下文件，

{ "_id" : ObjectId("514919fb23700b41723f94dc"), "name" : "A", "timespent" : 100 },
{ "_id" : ObjectId("514919fb23700b41723f94dd"), "name" : "B", "timespent" : 200 },
{ "_id" : ObjectId("514919fb23700b41723f94de"), "name" : "C", "timespent" : 300 },
{ "_id" : ObjectId("514919fb23700b41723f94df"), "name" : "D", "timespent" : 400 },
{ "_id" : ObjectId("514919fb23700b41723f94e0"), "name" : "E", "timespent" : 500 }

我们的查询得出：

{ "_id" : "451-650", "count" : 1 }
{ "_id" : "251-450", "count" : 2 }
{ "_id" : "0-250", "count" : 2 }

我们可能希望在管道中添加$sort阶段按范围对文档进行排序，但这只会对lexicographic order中的文档进行排序，因为＆＃34;范围＆＃34;的类型。

在mongodb聚合框架中执行case语句

2 个答案: