作者:麦克煎蛋 出处:https://www.cnblogs.com/mazhiyong/ 转载请保留这段声明,谢谢!
我们可以在FastAPI应用中使用中间件。
中间件实际上是一个函数,在每个request处理之前被调用,同时又在每个response返回之前被调用。
1、首先接收访问过来的request。
2、然后针对request或其他功能执行自定义逻辑。
3、传递request给应用程序继续处理。
4、接收应用所产生的response。
5、然后针对response或其他功能执行自定义逻辑。
6、返回response。
后台任务(如果有)会在所有中间件执行完以后才会执行。
我们通过在函数顶部添加装饰器 @app.middleware("http")
的方式来新建一个中间件。
import time from fastapi import FastAPI, Request app = FastAPI() @app.middleware("http") async def add_process_time_header(request: Request, call_next): start_time = time.time() response = await call_next(request) process_time = time.time() - start_time response.headers["X-Process-Time"] = str(process_time) return response
中间件函数接收两个参数:
1、 request
2、函数 call_next:传递
我们可以在response返回之前对其进行更改。request
给相应的路径操作,然后从路径操作中返回response。
我们可以在调用函数 call_next之前,即任何路径操作接收到request之前,添加一些自定义逻辑。
也可以在response产生后但返回之前,添加一些自定义逻辑,如上所示的添加自定义头信息。