# _*_ 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