FastAPI 中间件(一) 自定义中间件

作者：麦克煎蛋   出处：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：传递request给相应的路径操作，然后从路径操作中返回response。我们可以在response返回之前对其进行更改。

我们可以在调用函数 call_next之前，即任何路径操作接收到request之前，添加一些自定义逻辑。

也可以在response产生后但返回之前，添加一些自定义逻辑，如上所示的添加自定义头信息。