FastAPI 中间件
中间件是一个函数，它在请求特定路径操作之前，以及在每个响应之后执行。
使用装饰器创建中间件
创建一个函数，在函数的顶部使用装饰器 @app.middleware("http").
函数参数如下：
-
request: Request
-
call_next 函数：这个函数将接收 request 作为参数，将 request 传递给相应的 路径操作，然后它将返回由相应的路径操作生成的 response.
import uvicorn
from fastapi import FastAPI, Requestapp = FastAPI()@app.middleware("http")
async def m2(request: Request, call_next):"""第二个中间件"""print("第二个中间件 m2 ...request")response = await call_next(request)print("第二个中间件 m2 ...response")return response@app.middleware("http")
async def m1(request: Request, call_next):"""第一个中间件"""print("第一个中间件 m1 ...request")response = await call_next(request)print("第一个中间件 m1 ...response")return response@app.get("/item/{item_id}")
async def item(item_id: int):print("请求get item id", item_id)return {"item_id": item_id}if __name__ == '__main__':uvicorn.run(app)运行结果：
第一个中间件 m1 ...request
第二个中间件 m2 ...request
请求get item id 11111
第二个中间件 m2 ...response
第一个中间件 m1 ...response
INFO: 127.0.0.1:58664 - "GET /item/11111 HTTP/1.1" 200 OK继承BaseHTTPMiddleware
1、定义中间件类：首先，你需要定义一个类，并继承自 starlette.middleware.base.BaseHTTPMiddleware。
2、实现 dispatch 方法：在你的类中，你需要实现 dispatch 方法。这个方法会在请求被传递给下一个中间件或最终的应用程序之前被调用。
3、注册你的中间件：在你的 FastAPI 应用中注册你的中间件。
创建my_middlewares包
示例：计算API接口调用耗时 中间件
创建CalculateTimeMiddleware.py 文件
import timefrom fastapi import Request, FastAPI
from starlette.middleware.base import BaseHTTPMiddlewareclass CalculateTimeMiddleware(BaseHTTPMiddleware):"""计算API接口耗时 单位: 秒"""async def dispatch(self, request: Request, call_next):print("中间件: 计算API接口耗时...")start_time = time.perf_counter()response = await call_next(request)process_time = time.perf_counter() - start_timeprint(f"请求路径: {request.url.path}, 耗时: {process_time} 秒")return response编写Main.py 文件
import uvicorn
from fastapi import FastAPI
from starlette.requests import Requestfrom my_middlewares.AuthMiddleware import AuthMiddleware
from my_middlewares.CalculateTimeMiddleware import CalculateTimeMiddlewareapp = FastAPI()# 注册中间件
app.add_middleware(AuthMiddleware) # type: ignore
app.add_middleware(CalculateTimeMiddleware) # type: ignore@app.get("/item/{item_id}")
async def item(item_id: int, request: Request):# 获取上下文变量user_id = request.state.user_idprint("user_id:", user_id)return {"item_id": item_id}if __name__ == '__main__':uvicorn.run(app)示例：中间件传递参数
使用上下文变量（Context Variables），FastAPI的Request对象允许你设置和获取上下文变量，这在多个中间件或路由处理器之间共享数据非常有用。
编写AuthMiddleware.py文件
from starlette.middleware.base import BaseHTTPMiddlewareclass AuthMiddleware(BaseHTTPMiddleware):"""权限校验，并传递 user_id"""async def dispatch(self, request, call_next):print("中间件: 传递 user_id ...")user_id="user001"# 设置上下文变量 将user_id存储在request.state中request.state.user_id = user_idresponse = await call_next(request)return response运行结果：
中间件: 计算API接口耗时...
中间件: 传递 user_id ...
user_id: user001
请求路径: /item/111, 耗时: 0.0006680000005871989 秒