当使用 fastapi 时,中间件函数通常需要标记为异步(async)。然而,某些情况下,你可能需要将同步代码集成到中间件中。本文将探讨如何将中间件转换为同步模式。
首先,让我们了解一下中间件的本质。中间件是一种允许你在请求处理的各个阶段(如请求之前或响应之后)插入自定义逻辑的机制。在 fastapi 中,中间件函数被标记为异步,这意味着它们可以在处理请求时挂起。
然而,对于某些同步库,例如 peewee 和 redis-py,使用异步是不可能的。因此,如果你需要在中间件中使用这些库,就需要将其转换为同步模式。
要将中间件转换为同步模式,你可以使用 starlette.concurrency.run_in_threadpool() 辅助程序。此辅助程序允许你将在独立线程中运行的同步代码包装在协程中。
以下是如何实现同步中间件:
from fastapi import FastAPI, Request, Response from starlette.concurrency import run_in_threadpool app = FastAPI() def sync_middleware(request: Request, call_next): # 在单独的线程中运行同步代码 run_in_threadpool(sync_code) # 继续处理请求 response = await call_next(request) return response @app.middleware("http") async def my_middleware(request: Request, call_next): return await sync_middleware(request, call_next)
在上面的示例中,sync_middleware 函数包含需要以同步方式执行的代码(sync_code)。run_in_threadpool() 辅助程序将在独立线程中运行此代码,而中间件函数本身仍然是异步的。
通过这种方法,你可以将同步代码集成到你的中间件中,同时仍保持与 fastapi 的异步性质兼容。