Python FastAPI系列：常用FastAPI middleware中间件的详细使用

from fastapi import FastAPI
from starlette.requests import Request
from starlette.responses import JSONResponse
 
app = FastAPI()
 
@app.middleware("http")
async def add_process_time_header(request: Request, call_next):
    start = time.time()
    response = await call_next(request)
    process_time = time.time() - start
    response.headers["X-Process-Time"] = str(process_time)
    return response
 
@app.middleware("http")
async def custom_middleware(request: Request, call_next):
    # 在请求处理之前可以进行一些操作
    # 比如请求的验证、权限控制等
    # 如果调用 call_next，请求会继续到下一个中间件或路由
    response = await call_next(request)
    
    # 在请求处理之后可以进行一些操作
    # 比如修改响应、添加额外的头部信息等
    return response
 
@app.get("/")
async def main():
    return JSONResponse({"message": "Hello World"})

这个示例展示了如何在FastAPI应用中定义和使用middleware。首先，我们定义了一个add_process_time_header的中间件，它计算请求处理的时间并将其添加到响应头中。接着，我们定义了一个custom_middleware的中间件，它可以用于在请求处理前后进行自定义操作。最后，我们定义了一个简单的路由/，用于演示如何在没有其他中间件或路由修改的情况下，原样返回一个JSON响应。