FastAPI 学习之路(三十)中间件

  什么是中间件呢,"中间件"是一个函数,它在每个请求被特定的路径操作处理之前,以及在每个响应返回之前工作.

  • 它接收你的应用程序的每一个请求.

  • 然后它可以对这个请求做一些事情或者执行任何需要的代码.

  • 然后它将请求传递给应用程序的其他部分 (通过某种路径操作).

  • 然后它获取应用程序生产的响应 (通过某种路径操作).

  • 它可以对该响应做些什么或者执行任何需要的代码.

  • 然后它返回这个 响应.

      那么看下如何创建中间件?

      

要创建中间件你可以在函数的顶部使用装饰器 @app.middleware("http").

中间件参数接收如下参数:

  • request.

  • 一个函数 call_next 它将接收 request 作为参数.

    • 这个函数将 request 传递给相应的 路径操作.

    • 然后它将返回由相应的路径操作生成的 response.

  • 然后你可以在返回 response 前进一步修改它.

     

    我们看下如何去实现中间件。

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)
    response.headers['leizishuocceshikaifa']="leizishuoceshikaifa"
    return response
@app.post("/create")
def post():
    return "post"

我们实现一个中间件,

            在 response 的前和后 在任何路径操作收到request前,可以添加要和请求一起运行的代码.也可以在响应生成但是返回之前添加代码。

        我们去看下接口输出的结果

 

 

 我们看下接口返回的Headers

 

 我们可以看到,我标记红色的地方,就是我们在请求的中间件中,处理的,默认就给我们返回我们处理后的。

posted @ 2021-10-19 06:55  北漂的雷子  阅读(1016)  评论(0编辑  收藏  举报