Peewee prefetch 简单用法与优化技巧

本文详解如何通过 Peewee 的 `prefetch()` 函数一次性预加载关联数据，彻底避免模板渲染中因循环访问外键引发的 N+1 查询问题，将查询复杂度从 O(n+1) 降至 O(1)。

在使用 Peewee 构建 Web 应用（如配合 FastAPI 或 Starlette）时，一个常见且隐蔽的性能陷阱是 N+1 查询问题：当主模型（如 Sales）被获取后，在模板或视图中遍历其反向关联（如 sales.items），每次访问 it.item.item_name 都会触发一次额外的 Item 查询——即使只渲染一个销售单，若有 50 个商品，就会产生 1 次主查询 + 50 次子查询，严重拖慢响应速度。

你尝试的 JOIN 查询（如 Sales.select().join(SalesItem).join(Item)）看似合理，但实际仍会触发 N+1，原因在于：

• Peewee 的 JOIN 默认不自动填充反向关系对象；
• sales.items 仍是懒加载的 QueryResultWrapper，模板中首次访问 it.item 时仍会按需查库；
• 手动 JOIN 还可能因多对一关系导致重复主记录，需额外去重处理。

✅ 正确解法是使用 Peewee 内置的 prefetch() ——它采用“分步批量查询”策略：先查主表，再用主表 ID 批量查所有关联表，最后在内存中完成关系绑定，全程仅执行 2 条 SQL（或更少），且保持对象层级结构不变，完美适配模板语法。

以下是优化后的完整实现：

# view.py
from peewee import prefetch

def html_get(request, sales_id):
    # 步骤1：构建主查询（Sales）
    sales_query = Sales.select().where(Sales.sales_id == sales_id)

    # 步骤2：构建预加载查询链（SalesItem → Item）
    # 注意：SalesItem.select().join(Item) 确保每个 SalesItem 都附带其关联的 Item 实例
    sales_with_items = prefetch(
        sales_query,
        SalesItem.select().join(Item)  # 关联查询：SalesItem + 对应的 Item
    )

    return templates.TemplateResponse(
        'view_sales.html', 
        {'sales': sales_with_items}
    )

? 关键要点说明：

• prefetch() 第二个参数必须是 Model.select() 查询对象（非字符串或字段名），且需显式 .join(Item) 以包含关联字段；
• SalesItem.select().join(Item) 会生成一条 SELECT ... FROM salesitem JOIN item ON ... 查询，一次性拉取所有关联商品及名称；
• Peewee 自动将结果按 sales_id 分组，并为每个 Sales 实例的 items 属性注入已预加载的 SalesItem 对象（每个 SalesItem.item 已是完整 Item 实例，无需再查库）；
• 模板代码完全无需修改：{{ it.item.item_name }} 将直接访问内存对象，零额外查询。

⚠️ 注意事项：

• 确保模型定义中 SalesItem 的 backref='items' 与 Sales 类中的反向引用名一致（你代码中已正确配置）；
• 若需进一步减少查询数（如同时预加载多个层级），可传入多个查询参数：prefetch(query, SalesItem.select().join(Item), AnotherModel.select())；
• prefetch 不支持 LIMIT/OFFSET 在主查询中与复杂 JOIN 混用，如有分页需求，建议改用 paginate() 或手动分页逻辑。

通过这一优化，原本 O(n+1) 的查询降为 O(2)（1 次 Sales 查询 + 1 次批量关联查询），无论单据含多少商品，数据库交互次数恒定，性能提升显著且代码简洁可维护。