19.FastAPI中间件

19.FastAPI中间件

在FastAPI中,可以通过中间件在每个请求被特定的路径操作处理之前,以及在每个响应返回之前进行处理。FastAPI中间件是一个函数,其作用为:

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

  • 可以对请求做一些事情或者执行任何需要的代码

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

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

  • 对该响应进行处理或者执行任何需要的代码

  • 返回响应

19.1定义中间件

可以使用装饰器 @app.middleware("http") 来定义中间件,中间件函数的参数:

  • request

    请求

  • call_next

    函数,call_next将接收 request 作为参数,这个函数将 request 传递给相应的路径操作,然后它将返回由相应的路径操作生成的 response, 接下来可以在返回 response 前进一步处理它。

示例代码:

import time
from fastapi import FastAPI
from fastapi import Request
​
app = FastAPI()
​
@app.middleware("http")
async def middle(request: Request, call_next):
    start_time = time.time()
    response = await call_next(request)
    m = time.time() - start_time
    response.headers["X-Run-Time"] = str(m)
    return response
​
@app.get(path='/test')
async def test():
    return "Hello world."

执行请求:

curl http://127.0.0.1:8000/test -i
HTTP/1.1 200 OK
date: Tue, 01 Feb 2022 11:55:52 GMT
server: uvicorn
content-length: 14
content-type: application/json
x-run-time: 0.001463174819946289
​
"Hello world."

上面的代码中,通过中间件来计算请求的实际执行时间。

19.2请求及响应顺序

在使用中间件的情况下,FastAPI请求和响应的处理顺序为:中间件处理请求 -> 路由处理请求 -> 路由处理响应 ->中间件处理响应。代码示例:

# coding: utf-8
import time
from fastapi import FastAPI
from fastapi import Request
​
app = FastAPI()
​
@app.middleware("http")
async def middle(request: Request, call_next):
    print("middleware start")
    start_time = time.time()
    response = await call_next(request)
    m = time.time() - start_time
    response.headers["X-Run-Time"] = str(m)
    print("middleware end")
    return response
​
@app.get(path='/test')
async def test():
    print("test start")
    hello = "Hello world."
    print("test end")
    return hello

执行请求:

curl http://127.0.0.1:8000/test -i
HTTP/1.1 200 OK
date: Tue, 01 Feb 2022 12:10:19 GMT
server: uvicorn
content-length: 14
content-type: application/json
x-run-time: 0.002346038818359375
​
"Hello world."

后台print输出:

middleware start
test start
test end
middleware end

19.3中间件的用途

FastAPI的中间件可以拦截所有请求,同时也可以拦截所有响应,所以,一般情况下,可以针对请求和响应做一些统一的公共处理,比如:拦截请求后,记录用户的行为,或者对于特殊的处理进行鉴权等。

你可能感兴趣的:(Python,FastAPI,python,web,后端)