SQLAlchemy：只查模型部分字段提升性能

本文旨在解决在使用 SQLAlchemy 进行数据库查询时，如何高效地仅获取模型的部分列数据，避免不必要的数据传输，提升性能。通过对比不同查询方式，解释了 load_only 的使用场景，并强调了缓存可能带来的影响，帮助开发者在 FastAPI 等框架中编写更优化的数据库查询代码。

在使用 SQLAlchemy 进行数据库操作时，有时我们只需要获取模型的部分列数据，而不是全部列。这可以减少数据库查询的开销，提高程序的性能。本文将介绍几种方法来实现这个目标，并分析它们的优缺点。

直接使用 select 语句指定列

最直接的方法是在 select 语句中明确指定需要查询的列。例如，假设我们有一个 PostSQL 模型，包含 id、title 等多个列，我们只想获取 id 和 title 列的数据，可以这样写：

from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import Column, Integer, String
from sqlalchemy.orm import declarative_base

Base = declarative_base()

class PostSQL(Base):
    __tablename__ = "posts"
    id = Column(Integer, primary_key=True)
    title = Column(String)
    content = Column(String)

async def fetch_posts(db: AsyncSession):
    query = select(PostSQL.id, PostSQL.title)
    result = await db.execute(query)
    posts = result.all()
    return posts

这段代码会返回一个包含元组的列表，每个元组包含 id 和 title 两个值。这种方法简单直接，但返回的是元组，而不是 PostSQL 模型的实例。

使用 load_only 选项

load_only 选项可以用于加载模型的指定列，而忽略其他列。这在某些情况下可以提高性能。

from sqlalchemy.orm import load_only

async def fetch_posts_with_load_only(db: AsyncSession):
    query = select(PostSQL).options(load_only(PostSQL.id, PostSQL.title))
    result = await db.execute(query)
    posts = result.scalars().all()
    return posts

这段代码返回的是 PostSQL 模型的实例，但只有 id 和 title 列的数据被加载。

注意事项：

• load_only 只是告诉 SQLAlchemy 不要立即加载其他列的数据。如果后续代码尝试访问未加载的列，SQLAlchemy 仍然会发起额外的数据库查询来加载这些数据（延迟加载）。
• 某些 IDE 或 Python 环境可能存在缓存机制，导致即使使用了 load_only，仍然会显示所有列的数据。这通常是由于之前加载过完整的模型数据，并被缓存起来了。可以尝试清除缓存或重启环境来解决这个问题。

总结

• 如果只需要部分列的数据，并且不需要 PostSQL 模型的实例，直接在 select 语句中指定列是最简单的方法。
• 如果需要 PostSQL 模型的实例，并且希望延迟加载其他列的数据，可以使用 load_only 选项。
• 注意缓存可能带来的影响，确保代码的行为符合预期。

通过合理选择查询方式，可以有效地减少数据库查询的开销，提高程序的性能。在实际开发中，需要根据具体的需求和场景来选择最合适的方案。