FastAPI asyncpg – 无法执行操作：另一个操作正在进行中

在本文中，我们将介绍使用FastAPI和asyncpg时可能遇到的一个常见问题：无法执行操作：另一个操作正在进行中。

阅读更多：FastAPI 教程

问题描述

在使用FastAPI和asyncpg进行数据库操作时，有时会遇到”cannot perform operation: another operation is in progress”的错误。这个错误通常发生在一个操作正在进行中时，我们尝试执行另一个操作。比如，在一个请求处理函数中进行了一个查询操作，而在同一个请求中又尝试执行了一个更新操作。这样会导致asyncpg引发错误，提示我们无法同时执行多个数据库操作。

解决方案

解决这个问题的方法是使用asyncpg的连接池功能。连接池可以管理数据库连接的分配和释放，确保每个请求都能够获得和释放一个可用的数据库连接。

创建连接池

我们可以在应用启动时创建一个连接池，并在每个请求中使用该连接池获取数据库连接。

import asyncpg
from fastapi import FastAPI

app = FastAPI()

@app.on_event("startup")
async def startup():
    app.state.pool = await asyncpg.create_pool(dsn="postgres://user:password@localhost/mydatabase")

@app.on_event("shutdown")
async def shutdown():
    await app.state.pool.close()

@app.get("/")
async def root():
    async with app.state.pool.acquire() as connection:
        # 使用连接执行数据库操作
        # ...

使用连接池

在每次需要执行数据库操作的请求处理函数中，我们都可以使用连接池来获取数据库连接。在使用完连接之后，我们需要手动释放连接，使其返回到连接池中，供其他请求使用。

@app.get("/users/{user_id}")
async def get_user(user_id: int):
    # 从连接池中获取数据库连接
    async with app.state.pool.acquire() as connection:
        # 使用连接执行数据库操作
        # ...

@app.post("/users")
async def create_user(user: User):
    # 从连接池中获取数据库连接
    async with app.state.pool.acquire() as connection:
        # 使用连接执行数据库操作
        # ...

通过使用连接池，我们可以避免多个数据库操作同时进行的情况，解决”cannot perform operation: another operation is in progress”的错误。

总结

当使用FastAPI和asyncpg进行数据库操作时，如果遇到”cannot perform operation: another operation is in progress”错误，我们可以通过使用连接池来解决这个问题。连接池可以管理数据库连接的分配和释放，确保每个请求都能够获得和释放一个可用的数据库连接。这样就可以避免多个数据库操作同时进行，确保每次操作都能顺利执行。