fastapi自定义中间件

在FastAPI中，自定义中间件可以通过装饰器来实现。中间件是一种在请求处理之前和之后运行的函数。

以下是一个自定义中间件的示例：

from fastapi import FastAPI
from starlette.requests import Request
from starlette.responses import JSONResponse
 
app = FastAPI()
 
# 自定义中间件
@app.middleware("http")
async def custom_middleware(request: Request, call_next):
    # 在请求处理之前可以做的操作，比如验证、日志记录等
    print(f"Request: {request.method} {request.url}")
 
    # 调用下一个中间件或路由函数
    response = await call_next(request)
 
    # 在请求处理之后可以做的操作
    # 注意：response可能已经被修改，比如头部、内容等
    return response
 
@app.get("/")
async def main():
    return JSONResponse({"message": "Hello World"})

在这个例子中，custom_middleware 是自定义的中间件函数。它在请求处理之前和之后被调用。call_next 是一个特殊的参数，它是一个高阶函数，用于调用后续的中间件或路由函数。自定义中间件可以用来进行权限验证、请求记录、响应内容修改等操作。