FastAPI 如何在自定义APIRoute类中访问路由依赖项

在本文中，我们将介绍如何在使用FastAPI时，在自定义APIRoute类中访问路由依赖项的方法和技巧。

FastAPI是一个高性能的Python框架，它结合了Pydantic和Starlette两个优秀的库，可以轻松地构建高效的Web API。

阅读更多：FastAPI 教程

什么是APIRoute类

在FastAPI中，APIRoute类是处理HTTP请求的基本组件之一。它负责路由的匹配和处理请求，是构建Web API的核心。

APIRoute类是FastAPI内部定义的一个类，但它也可以通过继承它来自定义自己的APIRoute类。通过自定义APIRoute类，我们可以实现更高级的路由处理逻辑，以满足特定的需求。

路由依赖项

在FastAPI中，我们可以使用Depends装饰器定义一个路由依赖项。路由依赖项是一种可以在路由处理函数中使用的可重用的组件，它可以在处理函数执行之前执行一些预处理逻辑，例如身份验证、数据库连接等。

路由依赖项可以是同步的（返回普通值）或异步的（返回协程）。我们可以通过在路由处理函数的参数列表中声明路由依赖项来使用它们。

但是，在自定义APIRoute类中，我们无法直接访问这些路由依赖项。因为APIRoute类和路由处理函数是解耦的，它们之间没有直接的联系。

那么，如何在自定义APIRoute类中访问路由依赖项呢？继续往下看。

访问路由依赖项的方法

要在自定义APIRoute类中访问路由依赖项，我们可以通过重写__call__方法来实现。

__call__方法是在请求到达路由时被调用的方法。我们可以通过重写该方法，在执行路由处理函数之前访问路由依赖项。

下面是一个示例，演示如何在自定义APIRoute类中访问路由依赖项。

from fastapi import APIRouter, Depends
from fastapi.routing import APIRoute

class CustomAPIRoute(APIRoute):
    async def __call__(self, request):
        # 在执行路由处理函数之前执行一些逻辑

        # 获取路由依赖项
        db = await self.app.state.db.get_connection()

        # 设置路由依赖项，使其可以在路由处理函数中使用
        request.state.db = db

        # 调用父类的__call__方法，执行路由处理函数
        response = await super().__call__(request)

        # 在路由处理函数执行完毕后释放资源
        await self.app.state.db.release_connection(db)

        return response

router = APIRouter(route_class=CustomAPIRoute)

@router.get("/items/")
async def get_items(db=Depends(get_db)):
    # 在路由处理函数中可以访问到路由依赖项
    items = await db.fetchall("SELECT * FROM items")
    return items

在上面的示例中，我们创建了一个自定义的APIRoute类CustomAPIRoute，并重写了其__call__方法。

在__call__方法中，我们可以获取路由依赖项，将其设置为请求对象的属性，使得路由处理函数可以在其参数列表中直接使用。

在示例中，我们通过Depends(get_db)定义了一个路由依赖项db，路由处理函数get_items中可以直接使用该依赖项来访问数据库。