Python web框架fastapi中间件的使用

from fastapi import FastAPI
from starlette.requests import Request
from starlette.responses import JSONResponse
 
app = FastAPI()
 
# 自定义中间件函数
def custom_header_middleware(request: Request, call_next):
    response = call_next(request)
    response.headers["Custom-Header"] = "MyCustomValue"
    return response
 
# 将中间件函数应用到FastAPI应用实例
app.add_middleware(
    BaseHTTPMiddleware,
    dispatch=custom_header_middleware,
)
 
# 一个简单的路由，用于测试中间件效果
@app.get("/")
async def main():
    return JSONResponse({"message": "Hello World"})

这段代码定义了一个自定义的中间件函数custom_header_middleware，它会在响应中添加一个自定义的HTTP头部。然后，我们将这个中间件函数添加到FastAPI应用实例中。最后，我们定义了一个简单的路由/，用于测试中间件是否正常工作。