MongoDB 聚合查询详解，与 mysql 聚合类比有什么不同

MongoDB 聚合查询

什么是聚合查询

聚合操作主要用于处理数据并返回计算结果。聚合操作将来自多个文档的值组合在一起，按条件分组后，再进行一系列操作（如求和、平均值、最大值、最小值）以返回单个结果。

MongoDB 的聚合查询

​ 聚合是 MongoDB 的高级查询语言，它允许我们通过转化合并由多个文档的数据来生成新的在单个文档里不存在的文档信息。MongoDB 中聚合 (aggregate) 主要用于处理数据（例如分组统计平均值、求和、最大值等），并返回计算后的数据结果，有点类似 sql 语句中的 count (*)、group by。

​ 在 MongoDB 中，有两种方式计算聚合：Pipeline 和 MapReduce。Pipeline 查询速度快于 MapReduce，但是 MapReduce 的强大之处在于能够在多台 Server 上并行执行复杂的聚合逻辑。MongoDB 不允许 Pipeline 的单个聚合操作占用过多的系统内存。

聚合管道方法

MongoDB 的聚合框架就是将文档输入处理管道，在管道内完成对文档的操作，最终将文档转换为聚合结果，MongoDB 的聚合管道将 MongoDB 文档在一个管道处理完毕后将结果传递给下一个管道处理，管道操作是可以重复的。

​ 最基本的管道阶段提供过滤器，其操作类似查询和文档转换，可以修改输出文档的形式。其他管道操作提供了按特定字段对文档进行分组和排序的工具，以及用于聚合数组内容（包括文档数组）的工具。

​ 此外，在管道阶段还可以使用运算符来执行诸如计算平均值或连接字符串之类的任务。聚合管道可以在分片集合上运行。

聚合流程

db.collection.aggregate () 是基于数据处理的聚合管道，每个文档通过一个由多个阶段（stage）组成的管道，可以对每个阶段的管道进行分组、过滤等功能，然后经过一系列的处理，输出相应的结果。

聚合管道方法的流程参见下图

上图的聚合操作相当于 MySQL 中的以下语句：

select cust_id as _id, sum(amount) as total from orders where status like "%A%" group by cust_id;

详细流程

聚合语法

db.collection.aggregate(pipeline, options)

参数说明

注意事项

​ 使用 db.collection.aggregate () 直接查询会提示错误，但是传一个空数组如 db.collection.aggregate ([]) 则不会报错，且会和 find 一样返回所有文档。

常用聚合管道

与 mysql 聚合类比

为了便于理解，先将常见的 mongo 的聚合操作和 mysql 的查询做下类比

$count

返回包含输入到 stage 的文档的计数，理解为返回与表或视图的 find () 查询匹配的文档的计数。db.collection.count () 方法不执行 find () 操作，而是计数并返回与查询匹配的结果数。

语法

{ $count:  }

$count 阶段相当于下面 $group+$project 的序列：

db.collection.aggregate([
    {
        "$group": {
            "_id": null,
            "count": {// 这里count自定义，相当于mysql的select count(*) as tables
                "$sum": 1
            }
        }
    },
    {
        "$project": {// 返回不显示_id字段
            "_id": 0
        }
    }
])

示例

查询人数是 100000 以上的城市的数量

• $match：阶段排除 pop 小于等于 100000 的文档，将大于 100000 的文档传到下个阶段
• $count：阶段返回聚合管道中剩余文档的计数，并将该值分配给名为 count 的字段。

db.zips.aggregate([
    {
        "$match": {
            "pop": {
                "$gt": 100000
            }
        }
    },
    {
        "$count": "count"
    }
])

$group

按指定的表达式对文档进行分组，并将每个不同分组的文档输出到下一个阶段。输出文档包含一个_id 字段，该字段按键包含不同的组。

​ 输出文档还可以包含计算字段，该字段保存由 $group 的_id 字段分组的一些 accumulator 表达式的值。 $group 不会输出具体的文档而只是统计信息。

语法

{ $group: { _id: , : {  :  }, ... } }

• _id 字段是必填的；但是，可以指定_id 值为 null 来为整个输入文档计算累计值。
• 剩余的计算字段是可选的，并使用  运算符进行计算。
• _id 和  表达式可以接受任何有效的表达式。

accumulator 操作符

$group 阶段的内存限制为 100M，默认情况下，如果 stage 超过此限制，$group 将产生错误，但是，要允许处理大型数据集，请将 allowDiskUse 选项设置为 true 以启用 $group 操作以写入临时文件。

注意：

• "$addToSet":expr，如果当前数组中不包含 expr，那就将它添加到数组中。
• "$push":expr，不管 expr 是什么值，都将它添加到数组中，返回包含所有值的数组。

示例

按照 state 分组，并计算每一个 state 分组的总人数，平均人数以及每个分组的数量

db.zips.aggregate([
    {
        "$group": {
            "_id": "$state",
            "totalPop": {
                "$sum": "$pop"
            },
            "avglPop": {
                "$avg": "$pop"
            },
            "count": {
                "$sum": 1
            }
        }
    }
])

查找不重复的所有的 state 的值

db.zips.aggregate([
    {
        "$group": {
            "_id": "$state"
        }
    }
])

按照 city 分组，并且分组内的 state 字段列表加入到 stateItem 并显示

db.zips.aggregate([
    {
        "$group": {
            "_id": "$city",
            "stateItem": {
                "$push": "$state"
            }
        }
    }
])

下面聚合操作使用系统变量 $$ROOT 按 item 对文档进行分组，生成的文档不得超过 BSON 文档大小限制

db.zips.aggregate([
    {
        "$group": {
            "_id": "$city",
            "item": {
                "$push": "$$ROOT"
            }
        }
    }
]).pretty();

$match

过滤文档，仅将符合指定条件的文档传递到下一个管道阶段。

$match 接受一个指定查询条件的文档，查询语法与读操作查询语法相同。

语法

{ $match: {  } }

管道优化

​ $match 用于对文档进行筛选，之后可以在得到的文档子集上做聚合，$match 可以使用除了地理空间之外的所有常规查询操作符，在实际应用中尽可能将 $match 放在管道的前面位置。这样有两个好处：

• 一是可以快速将不需要的文档过滤掉，以减少管道的工作量；
• 二是如果再投射和分组之前执行 $match，查询可以使用索引。

使用限制

• 不能在 $match 查询中使用 $ 作为聚合管道的一部分。
• 要在 $match 阶段使用 $text，$match 阶段必须是管道的第一阶段。
• 视图不支持文本搜索。

示例

使用 $match 做简单的匹配查询，查询缩写是 NY 的城市数据

db.zips.aggregate([
    {
        "$match": {
            "state": "NY"
        }
    }
]).pretty();

使用 $match 管道选择要处理的文档，然后将结果输出到 $group 管道以计算文档的计数

db.zips.aggregate([
    {
        "$match": {
            "state": "NY"
        }
    },
    {
        "$group": {
            "_id": null,
            "sum": {
                "$sum": "$pop"
            },
            "avg": {
                "$avg": "$pop"
            },
            "count": {
                "$sum": 1
            }
        }
    }
]).pretty();

$unwind

从输入文档解构数组字段以输出每个元素的文档，简单说就是 可以将数组拆分为单独的文档。

语法

要指定字段路径，在字段名称前加上 $ 符并用引号括起来。

{ $unwind:  }

v3.2 + 支持如下语法

{
  $unwind:
    {
      path: ,
      #可选,一个新字段的名称用于存放元素的数组索引。该名称不能以$开头。
      includeArrayIndex: , 
      #可选，default :false，若为true,如果路径为空，缺少或为空数组，则$unwind输出文档
      preserveNullAndEmptyArrays:  
 	} 
 }

​ 如果为输入文档中不存在的字段指定路径，或者该字段为空数组，则 $unwind 默认会忽略输入文档，并且不会输出该输入文档的文档。

版本 3.2 中的新功能：要输出数组字段丢失的文档，null 或空数组，请使用选项 preserveNullAndEmptyArrays。

示例

以下聚合使用 $unwind 为 loc 数组中的每个元素输出一个文档：

db.zips.aggregate([
    {
        "$match": {
            "_id": "01002"
        }
    },
    {
        "$unwind": "$loc"
    }
]).pretty();

$project

$project 可以从文档中选择想要的字段，和不想要的字段（指定的字段可以是来自输入文档或新计算字段的现有字段），也可以通过管道表达式进行一些复杂的操作，例如数学操作，日期操作，字符串操作，逻辑操作。

语法

$project 管道符的作用是选择字段（指定字段，添加字段，不显示字段，_id：0，排除字段等），重命名字段，派生字段。

{ $project: {  } }

specifications 有以下形式：

:  是否包含该字段，field:1/0，表示选择 / 不选择 field

_id:  是否指定_id 字段

:  添加新字段或重置现有字段的值。 在版本 3.6 中更改：MongoDB 3.6 添加变量 REMOVE。如果表达式的计算结果为 $$REMOVE，则该字段将排除在输出中。

: v3.4 新增功能，指定排除字段

• 默认情况下，_id 字段包含在输出文档中。要在输出文档中包含输入文档中的任何其他字段，必须明确指定 $project 中的包含。 如果指定包含文档中不存在的字段，$project 将忽略该字段包含，并且不会将该字段添加到文档中。
• 默认情况下，_id 字段包含在输出文档中。要从输出文档中排除_id 字段，必须明确指定 $project 中的_id 字段为 0。
• v3.4 版新增功能 - 如果指定排除一个或多个字段，则所有其他字段将在输出文档中返回。 如果指定排除_id 以外的字段，则不能使用任何其他 $project 规范表单：即，如果排除字段，则不能指定包含字段，重置现有字段的值或添加新字段。此限制不适用于使用 REMOVE 变量条件排除字段。
• v3.6 版本中的新功能 - 从 MongoDB 3.6 开始，可以在聚合表达式中使用变量 REMOVE 来有条件地禁止一个字段。
• 要添加新字段或重置现有字段的值，请指定字段名称并将其值设置为某个表达式。
• 要将字段值直接设置为数字或布尔文本，而不是将字段设置为解析为文字的表达式，请使用 $literal 操作符。否则，$project 会将数字或布尔文字视为包含或排除该字段的标志。
• 通过指定新字段并将其值设置为现有字段的字段路径，可以有效地重命名字段。
• 从 MongoDB 3.2 开始，$project 阶段支持使用方括号 [] 直接创建新的数组字段。如果数组规范包含文档中不存在的字段，则该操作会将空值替换为该字段的值。
• 在版本 3.4 中更改 - 如果 $project 是一个空文档，MongoDB 3.4 和更高版本会产生一个错误。
• 投影或添加 / 重置嵌入文档中的字段时，可以使用点符号

示例

以下 $project 阶段的输出文档中只包含_id，city 和 state 字段

db.zips.aggregate([
    {
        "$project": {
            "_id": 1,
            "city": 1,
            "state": 1
        }
    }
]).pretty();

_id 字段默认包含在内。要从 $ project 阶段的输出文档中排除_id 字段，请在 project 文档中将_id 字段设置为 0 来指定排除_id 字段。

db.zips.aggregate([
    {
        "$project": {
            "_id": 0,
            "city": 1,
            "state": 1
        }
    }
]).pretty();

以下 $ project 阶段从输出中排除 loc 字段

db.zips.aggregate([
    {
        "$project": {
            "loc": 0
        }
    }
]).pretty();

可以在聚合表达式中使用变量 REMOVE 来有条件地禁止一个字段，

db.zips.aggregate([
    {
        "$project": {
            "_id": 1,
            "city": 1,
            "state": 1,
            "pop": 1,
            "loc": {
                "$cond": {
                    "if": {
                        "$gt": [
                            "$pop",
                            1000
                        ]
                    },
                    "then": "$$REMOVE",
                    "else": "$loc"
                }
            }
        }
    }
]).pretty();

我们还可以改变数据，将人数大于 1000 的城市坐标重置为 0

db.zips.aggregate([
    {
        "$project": {
            "_id": 1,
            "city": 1,
            "state": 1,
            "pop": 1,
            "loc": {
                "$cond": {
                    "if": {
                        "$gt": [
                            "$pop",
                            1000
                        ]
                    },
                    "then": [
                        0,
                        0
                    ],
                    "else": "$loc"
                }
            }
        }
    }
]).pretty();

新增字段列

db.zips.aggregate([
    {
        "$project": {
            "_id": 1,
            "city": 1,
            "state": 1,
            "pop": 1,
            "desc": {
                "$cond": {
                    "if": {
                        "$gt": [
                            "$pop",
                            1000
                        ]
                    },
                    "then": "人数过多",
                    "else": "人数过少"
                }
            },
            "loc": {
                "$cond": {
                    "if": {
                        "$gt": [
                            "$pop",
                            1000
                        ]
                    },
                    "then": [
                        0,
                        0
                    ],
                    "else": "$loc"
                }
            }
        }
    }
]).pretty();

$limit

限制传递到管道中下一阶段的文档数

语法

{ $limit:  }

示例，此操作仅返回管道传递给它的前 5 个文档。 $limit 对其传递的文档内容没有影响。

db.zips.aggregate({
    "$limit": 5
});

注意

​ 当 $sort 在管道中的 $limit 之前立即出现时，$sort 操作只会在过程中维持前 n 个结果，其中 n 是指定的限制，而 MongoDB 只需要将 n 个项存储在内存中。当 allowDiskUse 为 true 并且 n 个项目超过聚合内存限制时，此优化仍然适用。

$skip

跳过进入 stage 的指定数量的文档，并将其余文档传递到管道中的下一个阶段

语法

{ $skip:  }

示例，此操作将跳过管道传递给它的前 5 个文档， $skip 对沿着管道传递的文档的内容没有影响。

db.zips.aggregate({
    "$skip": 5
});

$sort

对所有输入文档进行排序，并按排序顺序将它们返回到管道。

语法

{ $sort: { : , :  ... } }

$sort 指定要排序的字段和相应的排序顺序的文档。  可以具有以下值之一：

• 1 指定升序。
• -1 指定降序。
• {$meta：“textScore”} 按照降序排列计算出的 textScore 元数据。

示例

要对字段进行排序，请将排序顺序设置为 1 或 - 1，以分别指定升序或降序排序，如下例所示：

db.zips.aggregate([
    {
        "$sort": {
            "pop": -1,
            "city": 1
        }
    }
])

$sortByCount

根据指定表达式的值对传入文档分组，然后计算每个不同组中文档的数量。每个输出文档都包含两个字段：包含不同分组值的_id 字段和包含属于该分组或类别的文档数的计数字段，文件按降序排列。

语法

{ $sortByCount:   }

使用示例

下面举了一些常用的 mongo 聚合例子和 mysql 对比，假设有一条如下的数据库记录（表名：zips）作为例子：

统计所有数据

SQL 的语法格式如下

select count(1) from zips;

mongoDB 的语法格式

db.zips.aggregate([
    {
        "$group": {
            "_id": null,
            "count": {
                "$sum": 1
            }
        }
    }
])

对所有城市人数求合

SQL 的语法格式如下

select sum(pop) AS tota from zips;

mongoDB 的语法格式

db.zips.aggregate([
    {
        "$group": {
            "_id": null,
            "total": {
                "$sum": "$pop"
            }
        }
    }
])

对城市缩写相同的城市人数求合

SQL 的语法格式如下

select state,sum(pop) AS tota from zips group by state;

mongoDB 的语法格式

db.zips.aggregate([
    {
        "$group": {
            "_id": "$state",
            "total": {
                "$sum": "$pop"
            }
        }
    }
])

state 重复的城市个数

SQL 的语法格式如下

select state,count(1) AS total from zips group by state;

mongoDB 的语法格式

db.zips.aggregate([
    {
        "$group": {
            "_id": "$state",
            "total": {
                "$sum": 1
            }
        }
    }
])

state 重复个数大于 100 的城市

SQL 的语法格式如下

select state,count(1) AS total from zips group by state having count(1)>100;

mongoDB 的语法格式

db.zips.aggregate([
    {
        "$group": {
            "_id": "$state",
            "total": {
                "$sum": 1
            }
        }
    },
    {
        "$match": {
            "total": {
                "$gt": 100
            }
        }
    }
])

作者：博学谷狂野架构师