MongoDB Beanie如何查询仅返回数组中的单个用户条目？

你的原查询会找到包含符合条件用户的Document文档，但返回的是完整的文档对象（包含所有用户数据）。要仅返回符合条件的单个用户，可通过以下两种方式实现：

方式1：使用投影过滤数组元素 #

利用MongoDB的$elemMatch操作符，在查询时指定投影规则，只返回数组中匹配条件的第一个元素：

q = await Document.find_one(
    (Document.course == 1) &
    (Document.user.email == "user1@mail.com") &
    (Document.user.last_name == "Don"),
    projection={"user": {"$elemMatch": {"email": "user1@mail.com", "last_name": "Don"}}}
)

# 提取单个用户对象
if q and q.user:
    matched_user = q.user[0]

如果使用Beanie的原生语法，也可以这样写：

q = await Document.find_one(
    (Document.course == 1) &
    (Document.user.email == "user1@mail.com") &
    (Document.user.last_name == "Don"),
    projection=Document.user.element_match({"email": "user1@mail.com", "last_name": "Don"})
)

方式2：使用聚合管道直接返回用户对象 #

通过聚合管道可以更灵活地处理嵌套数组，最终直接返回符合条件的用户对象：

pipeline = [
    # 第一步：匹配包含目标用户的文档
    {"$match": {
        "course": 1,
        "user.email": "user1@mail.com",
        "user.last_name": "Don"
    }},
    # 第二步：展开用户数组
    {"$unwind": "$user"},
    # 第三步：筛选出符合条件的单个用户
    {"$match": {
        "user.email": "user1@mail.com",
        "user.last_name": "Don"
    }},
    # 第四步：将用户对象作为根文档返回
    {"$replaceRoot": {"newRoot": "$user"}}
]

# 获取结果，最多返回1个用户
matched_user_list = await Document.aggregate(pipeline).to_list(length=1)
matched_user = matched_user_list[0] if matched_user_list else None

关键说明 #

• 原查询的问题：仅筛选出包含目标用户的Document，但不会对user数组进行过滤，因此返回的文档包含所有用户数据。
• $elemMatch会返回数组中第一个匹配条件的元素，适合确定数组中仅有一个匹配项的场景。
• 聚合管道方式更适合需要复杂过滤或确保只返回精准匹配用户的场景。

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