monitoring/middleware/http_middleware.py

21 lines
519 B
Python

# _*_ coding: utf-8 _*_
# @Time :2022/5/29 09:13
# @Email :508737091@qq.com
# @Author :qiangyanwen
# @File :http_middleware.py
from fastapi.requests import Request
from starlette.types import Message
async def set_body(request: Request, body: bytes):
async def receive() -> Message:
return {"type": "http.request", "body": body}
request._receive = receive
async def get_body(request: Request) -> bytes:
body = await request.body()
await set_body(request, body)
return body