上海网站建设定制公谷歌推广培训

需求背景:

需要再中间件中获取body 中的信息 但是 又想要在之后 还可以重复取 这个body 因为有的接口写法是直接从body中获取参数,然而这个body是数据流的形式,一旦取一次就导致后面取不到里面的值了 。

解决方式：

1.保存请求体：

在中间件中读取请求体后，将其保存到一个变量中，并将其附加到请求中，以便后续处理函数可以访问。

from fastapi import FastAPI, Request
from starlette.middleware.base import BaseHTTPMiddleware
from starlette.middleware.cors import CORSMiddlewareapp = FastAPI()class CustomMiddleware(BaseHTTPMiddleware):async def dispatch(self, request: Request, call_next):# 读取请求体body = await request.body()  # 获取原始请求体request._body = body  # 将其附加到请求对象# 继续处理请求response = await call_next(request)return response# 添加中间件
app.add_middleware(CustomMiddleware)@app.post("/items/")
async def create_item(request: Request):# 读取请求体body = await request.json()  # 现在可以安全读取return {"body": body}

但是这里有点问题  如果create_item(request: Request) 是这样的写法 create_item(xxx: xxx = Body()) 这里会默认的从body中读取数据而不是我们重新放进去的 _body ，而这用用法比较多的时候我们改动范围比较大所以需要第二种方式实现

2.将body原路塞回去

class LoggingMiddleware(BaseHTTPMiddleware):async def dispatch(self, request: Request, call_next):request_body = await request.body()# 定义一个新的 receive 方法，返回缓存的请求体async def receive():return {"type": "http.request", "body": request_body}# 替换请求的 _receive 方法request._receive = receive

这样做的话再其他的地方再次调用 request.body()了  不需要大范围的修改 create_item(xxx: xxx = Body()) 这种写法了