SQLAlchemy+asyncpg无法捕获PostgreSQL延迟约束触发的异常

外键约束设置了DEFERRABLE INITIALLY DEFERRED属性后，PostgreSQL会将外键合规性校验推迟到事务最终COMMIT阶段才执行，而非SQL语句执行时即时校验。
你编写的try-except块仅包裹了conn.execute(stmt)插入语句执行逻辑，完全没有覆盖后续的事务提交流程：FastAPI集成SQLAlchemy时，默认的事务提交逻辑通常写在数据库连接的依赖项中，会在接口业务逻辑返回响应后才执行。此时外键校验触发的IntegrityError抛出时，你的try块早已运行退出，自然无法捕获异常，哪怕你在块里捕获通用Exception也没有效果。
而联合唯一键约束默认是即时校验，执行insert语句时就会触发冲突报错，刚好落在try块覆盖的代码范围内，所以能被正常捕获。

• 方案1：调整外键校验时机
  如果业务没有必须延迟校验外键的特殊需求（比如同一事务内需要调整多表插入顺序、临时允许外键关联数据暂不存在的场景），直接去掉外键的DEFERRABLE INITIALLY DEFERRED配置，使用默认的即时校验策略即可。调整后外键冲突会和唯一键冲突一样，在insert执行阶段抛出，现有try-except逻辑无需改动即可正常捕获。
• 方案2：将事务提交逻辑纳入异常捕获范围
  如果必须保留外键延迟校验的配置，就关闭框架默认的自动提交逻辑，手动在业务代码里执行事务提交，把提交动作也放到同一个try块中，参考代码如下：

  import asyncpg
  from sqlalchemy.exc import IntegrityError
  
  try:
      cr: sa.engine.CursorResult = await conn.execute(stmt)
      # 手动提交事务，将提交动作纳入异常捕获范围
      await conn.commit()
      return schemas.UserGroup(
          **user_group_dict,
          id=cr.inserted_primary_key[0],
      )
  except IntegrityError as e:
      # 异常后必须先回滚事务
      await conn.rollback()
      # 可通过原始异常类型区分错误场景，返回对应业务提示
      if isinstance(e.orig, asyncpg.exceptions.ForeignKeyViolationError):
          raise exceptions.conflict_exception("关联的用户或用户组不存在")
      raise exceptions.conflict_exception("用户已加入该用户组")
  

• 方案3：补充插入前前置校验（仅作补充）
  执行插入语句前，先查询校验传入的user_id、group_id是否在关联表存在，同时校验(user_id, group_id)组合是否已存在，从源头减少约束触发概率。注意该方案存在并发竞态问题，高并发场景下校验和插入的间隙仍可能出现数据写入导致约束报错，因此不能替代数据库层的异常捕获逻辑，仅作为优化提示体验的补充手段。

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