如何通过单次MongoDB查询同时返回更新前后的文档？

好问题！MongoDB的原生find_one_and_update（包括PyMongo里的对应方法）确实只能返回原始文档（ReturnDocument.BEFORE）或者更新后的文档（ReturnDocument.AFTER）其中一种，没办法单次调用就直接拿到两者。不过有两种靠谱的替代方案可以实现类似效果：

方案1：使用聚合管道+$merge（MongoDB 4.2+ 支持） #

利用聚合管道的原子性，我们可以先复制原始文档，再执行更新，最后同时输出原始和更新后的结果，并且将更新后的文档写回原集合。示例PyMongo代码如下：

import time
from pymongo import MongoClient

client = MongoClient()
db = client['your_database_name']

email = "user@example.com"
pipeline = [
    # 匹配目标文档
    {"$match": {"email": email}},
    # 将原始文档复制到新字段中
    {"$addFields": {"original_doc": "$$ROOT"}},
    # 执行更新操作
    {"$set": {"last_login": time.time()}},
    # 将更新后的文档写回原集合（原子替换）
    {"$merge": {"into": "users", "on": "_id", "whenMatched": "replace"}},
    # 输出我们需要的两个文档，隐藏默认_id
    {"$project": {"original_doc": 1, "updated_doc": "$$ROOT", "_id": 0}}
]

# 执行聚合并获取结果
result = list(db['users'].aggregate(pipeline))
if result:
    original_document = result[0]['original_doc']
    updated_document = result[0]['updated_doc']
    print("原始文档:", original_document)
    print("更新后的文档:", updated_document)

这个方案的核心优势是原子性，整个聚合过程是单原子操作，不会出现其他操作干扰中间状态的情况。

方案2：使用事务（需MongoDB副本集/分片集群） #

如果你的MongoDB部署在副本集或分片集群环境中，可以通过事务来实现“先查原始文档→更新→再查更新后文档”的原子操作，确保两次查询和更新的一致性。示例代码：

import time
from pymongo import MongoClient
from pymongo.read_concern import ReadConcern
from pymongo.write_concern import WriteConcern

client = MongoClient()
db = client['your_database_name']
email = "user@example.com"

# 开启会话和事务
with client.start_session() as session:
    session.start_transaction(
        read_concern=ReadConcern("local"),
        write_concern=WriteConcern(w="majority")
    )
    try:
        # 获取原始文档
        original_document = session.client['your_database_name']['users'].find_one({"email": email})
        if original_document:
            # 执行更新操作
            session.client['your_database_name']['users'].update_one(
                {"email": email},
                {"$set": {"last_login": time.time()}}
            )
            # 获取更新后的文档
            updated_document = session.client['your_database_name']['users'].find_one({"email": email})
        
        # 提交事务
        session.commit_transaction()
        if original_document:
            print("原始文档:", original_document)
            print("更新后的文档:", updated_document)
    except Exception as e:
        # 遇到异常回滚事务
        session.abort_transaction()
        raise e

这个方案逻辑更直观，但依赖MongoDB的事务支持（单节点部署无法使用）。

注意事项 #

• 如果你使用的MongoDB版本低于4.2，无法使用$merge，此时事务是更可靠的选择（若环境支持）；
• 若不需要严格的原子性，也可以先查询原始文档再执行更新，但这种方式存在并发风险（可能在查询和更新之间，文档被其他操作修改）。

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