如何通过MongoDB聚合或查询计算指定文档的前后文档数？

可以通过MongoDB的聚合查询或计数查询直接在数据库端完成计算，无需将全量数据拉取到客户端循环统计，以下是具体实现方案：

场景回顾 #

你当前的实现是通过Python循环遍历聚合后的全量文档，统计指定emblem文档的前后数量。以你的数据为例：

• 原始文档集合：

[
    {"emblem": 1, "name": "Z"},
    {"emblem": 2, "name": "H"},
    {"emblem": 3, "name": "T"},
    {"emblem": 4, "name": "Q"},
    {"emblem": 6, "name": "L"}
]

• 聚合排序规则：先按name小写后升序，再按emblem升序，排序后顺序为2,6,4,3,1
• 需要统计emblem=6的文档：前面有3个文档，后面有1个文档

解决方案：数据库端直接计算 #

步骤1：获取目标文档的排序键值 #

首先需要拿到目标文档对应的排序字段值（这里是转换后的lowerName和emblem），用于后续比较：

from pymongo import MongoClient

mongo = MongoClient()
DB_1 = mongo["DB_1"]
collection_1 = DB_1["collection_1"]

# 获取目标文档的排序键值
target_emblem = 6
target_doc = next(collection_1.aggregate([
    {"$match": {"emblem": target_emblem}},
    {"$addFields": {"lowerName": {"$toLower": "$name"}}},
    {"$project": {"lowerName": 1, "emblem": 1, "_id": 0}}
]))
# target_doc 结果：{"lowerName": "l", "emblem": 6}

步骤2：统计前面的文档数量（before） #

统计所有排序位置在目标文档之前的文档总数，条件为：

• lowerName小于目标值，或者
• lowerName等于目标值且emblem小于目标值

用聚合查询实现：

before_result = collection_1.aggregate([
    # 先统一转换name为小写，匹配排序规则
    {"$addFields": {"lowerName": {"$toLower": "$name"}}},
    # 筛选排序在目标文档之前的文档
    {"$match": {
        "$or": [
            {"lowerName": {"$lt": target_doc["lowerName"]}},
            {"$and": [
                {"lowerName": target_doc["lowerName"]},
                {"emblem": {"$lt": target_doc["emblem"]}}
            ]}
        ]
    }},
    # 统计数量
    {"$count": "before"}
])

before_count = next(before_result)["before"] if before_result.alive else 0
# 这里结果为3

步骤3：统计后面的文档数量（after） #

统计所有排序位置在目标文档之后的文档总数，条件与before相反：

after_result = collection_1.aggregate([
    {"$addFields": {"lowerName": {"$toLower": "$name"}}},
    {"$match": {
        "$or": [
            {"lowerName": {"$gt": target_doc["lowerName"]}},
            {"$and": [
                {"lowerName": target_doc["lowerName"]},
                {"emblem": {"$gt": target_doc["emblem"]}}
            ]}
        ]
    }},
    {"$count": "after"}
])

after_count = next(after_result)["after"] if after_result.alive else 0
# 这里结果为1

适配分页场景 #

如果你的分页需要结合搜索/过滤条件，只需在上述聚合的$match阶段添加对应的过滤规则即可。比如有搜索关键词时，先添加{"$match": {"name": {"$regex": "xxx"}}}到聚合管道最前面，再执行后续的排序键转换和前后数量统计。

这种方式避免了全量数据传输，在数据量大的场景下性能更优，能直接为分页按钮提供[prev ({before_count} documents)] [next ({after_count} documents)]的显示数据。

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