Docker-Compose部署FastAPI+PostgreSQL异步连接失败求助

核心原因 #

你混淆了SQLAlchemy异步引擎的驱动要求：异步模式必须用异步驱动（如asyncpg），而psycopg2是同步驱动，安装它反而会导致SQLAlchemy误判，触发错误。最初的ModuleNotFoundError是因为环境未装对应驱动，但方向错了——不该装psycopg2，该直接装asyncpg。

1. 修正依赖配置 #

在requirements.txt（或pyproject.toml）中，替换psycopg2为asyncpg：

fastapi
uvicorn
sqlalchemy[asyncio]
asyncpg
# 其他项目依赖...

2. 修正数据库引擎配置（EngineManager） #

确保SQLAlchemy异步引擎使用asyncpg连接字符串：

from sqlalchemy.ext.asyncio import create_async_engine

class EngineManager:
    def __init__(self, db_url: str):
        # 连接字符串格式：postgresql+asyncpg://user:password@host:port/dbname
        self.engine = create_async_engine(
            db_url,
            pool_pre_ping=True,
            echo=True  # 可选，用于调试SQL语句
        )

3. 修正Dockerfile配置 #

确保镜像安装asyncpg而非psycopg2：

FROM python:3.11-slim

WORKDIR /app

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

COPY . .

CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

4. 验证docker-compose.yml的数据库连接参数 #

确保web服务的环境变量使用asyncpg连接字符串：

version: '3.8'
services:
  web:
    build: .
    ports:
      - "8000:8000"
    environment:
      - DATABASE_URL=postgresql+asyncpg://postgres:your_password@db:5432/your_db_name
    depends_on:
      - db
  db:
    image: postgres:15-alpine
    environment:
      - POSTGRES_USER=postgres
      - POSTGRES_PASSWORD=your_password
      - POSTGRES_DB=your_db_name
    volumes:
      - postgres_data:/var/lib/postgresql/data/

volumes:
  postgres_data:

5. 修正DataSessionManager（确保用异步会话） #

from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.orm import sessionmaker

class DataSessionManager:
    def __init__(self, engine):
        self.async_session = sessionmaker(
            engine,
            class_=AsyncSession,
            expire_on_commit=False
        )

    async def get_session(self):
        async with self.async_session() as session:
            yield session

错误根源回顾 #

• 最初的ModuleNotFoundError：SQLAlchemy找不到异步驱动时，会尝试回退到同步驱动psycopg2，但环境未安装所以报错——但这是错误的解决方向，应直接安装异步驱动asyncpg。
• 安装psycopg2后的InvalidRequestError：异步引擎（create_async_engine）要求搭配异步驱动，psycopg2是同步驱动，二者不兼容导致SQLAlchemy抛出错误。

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