FastAPI 保存上下文将在端点中可用

Question

我的目标是从请求中提取一些信息并将其放在某个全局上下文中。 我尝试的第一件事是执行此操作的中间件功能

user_ctx: ContextVar[typing.Optional[UUID]] = ContextVar('user_ctx', default=None)

def save_user_ctx(request: Request):
    uid = ... # logic that extracts user uid from request header
    user_ctx.set(uid)

我将它添加到另一个中间件函数中，例如my_middleware 。

所以它确实设置了上下文，但据我所知，中间件在另一个线程中工作，所以如果我尝试类似

@router.post('/some_url', response_model=SomeResponseModel)
def some_url(data: SomeRequestModel, auth_checker = Depends(my_middleware)):
    user_ctx.get() # >> None

auth_checker具有所有身份验证信息，因此中间件可以正常工作。 但是uid = user_ctx.get()返回None 。

这只是一个例子。 请

我想要实现的是在自动为端点提供服务的线程中存储来自请求的一些上下文。 我的意思是不写ctx.set(smth)作为每个端点的第一行。

我也尝试写装饰器

def save_request_ctx(func):
    @wraps(func)
    def wrap(*args, **kwargs):
        ctx.set(...)
        return func(*args, **kwargs)
    return wrap

但是，如果端点没有request: Request作为它获取的参数 - 它不会出现在args或kwargs 。 因此，此外，如果您可以告诉如何强制每个端点将request作为参数，而无需在每个端点函数中手动添加它 - 它也将被接受为解决方案。

如果我对我想要的东西的解释很复杂，我会尝试澄清它。

Answer 1

为此，您可能想使用我的starlette-context 。

为了实现您想要的效果，最好将ContextMiddleware子类化并覆盖set_context方法。 返回的内容将在您可以在任何地方导入的context对象中可用。

示例代码

from starlette_context.middleware import ContextMiddleware


class YourMiddleware(ContextMiddleware):
    async def set_context(self, request: Request) -> dict:
        return {"from_middleware": True}

注册此中间件后，您可以在视图/记录器中导入

from starlette_context import context

并使用context["from_middleware"] 。

请注意，我为大多数常见用例准备了“插件”。 如果您想访问应用程序中任何位置的请求/关联 ID，只需使用 init 中间件

from starlette_context import context, plugins
from starlette_context.middleware import ContextMiddleware

middleware = [
    Middleware(
        ContextMiddleware,
        plugins=(plugins.RequestIdPlugin(), plugins.CorrelationIdPlugin()),
    )
]

app = Starlette(debug=True, middleware=middleware)

FastAPI 使用 Starlette 接口，所以你应该很高兴。

关于

但是，如果端点没有 request: Request 作为它获取的参数 - 它不会出现在 args 或 kwargs 中。 因此，此外，如果您可以告诉如何强制每个端点将请求作为参数，而无需在每个端点函数中手动添加它 - 它也将被接受为解决方案。

我无法回答这个问题，因为我从未尝试过使用没有请求参数的视图。

编辑：我添加了不需要请求对象的RawContextMiddleware 。