在 FastAPI 中使用中间件非常简单,只需在应用程序创建时将中间件添加到应用程序中即可。
首先,创建一个中间件函数,这个函数会在每个请求处理之前运行,可以对请求进行一些处理或者验证。例如:
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
app = FastAPI()
# 创建一个中间件函数
async def custom_middleware(request, call_next):
# 在请求处理之前执行一些操作
print("Before handling the request")
response = await call_next(request)
# 在请求处理之后执行一些操作
print("After handling the request")
return response
# 将中间件添加到应用程序
app.add_middleware(custom_middleware)
在上面的示例中,我们创建了一个名为custom_middleware
的中间件函数,并将其添加到了应用程序中。在这个中间件函数中,我们可以在处理请求前后执行一些操作。
另外,FastAPI 还提供了一些内置的中间件,例如CORSMiddleware
用于处理跨域资源共享。你可以像下面这样添加内置中间件:
# 添加 CORS 中间件
app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_credentials=True, allow_methods=["*"], allow_headers=["*"])
通过添加中间件,可以轻松地对请求进行处理、验证或者修改,使应用程序更加灵活和强大。