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产生后但返回之前,添加一些自定义逻辑,如上所示的添加自定义头信息。

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

上一篇:洛谷P5443 【[APIO2019]桥梁】


下一篇:windows kubectl 远程操作k8s