解决SQLModel/FastAPI用户关注功能的MissingGreenlet异常

问题根源 #

在FastAPI异步路由中直接操作ORM对象的关系属性（比如current_user.following）时，SQLAlchemy会触发同步SQL查询来加载关联数据，但异步上下文未启动greenlet环境，导致抛出MissingGreenlet异常——哪怕修改lazy加载类型也无法解决，因为所有默认的关系加载逻辑都是同步实现的。

最优解决方案：直接操作中间关联表 #

绕开ORM关系属性的自动加载逻辑，直接创建并添加UserFollowLink对象到数据库，避免触发同步IO操作：

修改后的路由代码 #

from sqlalchemy import select
from sqlalchemy.orm import selectinload
from fastapi import HTTPException, status

@router.get("/{user_id}/follow")
async def follow_user(db: AsyncSession, current_user: CurrentUser, user_id: int):
    # 检查是否已关注，避免重复创建关联
    existing_follow = await db.exec(
        select(UserFollowLink).where(
            UserFollowLink.follower_id == current_user.id,
            UserFollowLink.followed_id == user_id
        )
    )
    if existing_follow.first():
        raise HTTPException(status.HTTP_400_BAD_REQUEST, "已关注该用户")
    
    # 直接创建关联记录
    follow_link = UserFollowLink(follower_id=current_user.id, followed_id=user_id)
    db.add(follow_link)
    await db.commit()
    
    # 预加载用户的following列表后返回（可选）
    updated_user = await db.get(
        User, 
        current_user.id, 
        options=[selectinload(User.following)]
    )
    return updated_user

备选方案：用greenlet_spawn包裹同步操作 #

如果一定要通过ORM关系属性操作，可以用SQLAlchemy的greenlet_spawn将同步逻辑包裹起来，强制在greenlet环境中执行：

from sqlalchemy.util import greenlet_spawn
from sqlalchemy.orm import selectinload
from fastapi import HTTPException, status

@router.get("/{user_id}/follow")
async def follow_user(db: AsyncSession, current_user: CurrentUser, user_id: int):
    user = await db.get(User, user_id)
    if not user:
        raise HTTPException(status.HTTP_404_NOT_FOUND, "用户不存在")
    
    # 用greenlet_spawn包裹同步的append操作
    await greenlet_spawn(lambda: current_user.following.append(user))
    db.add(current_user)
    await db.commit()
    
    # 预加载关系后返回用户数据
    updated_user = await db.get(
        User, 
        current_user.id, 
        options=[selectinload(User.following)]
    )
    return updated_user

关键注意事项 #

• 确保数据库依赖DB是AsyncSession类型，而非同步的Session；
• 返回用户数据时，必须用selectinload等异步预加载方法获取关系数据，避免再次触发同步IO。

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