Comment enregistrer les requêtes et réponses HTTP brutes dans Python FastAPI ?

Comment enregistrer les requêtes/réponses HTTP brutes dans Python FastAPI ?

FastAPI fournit plusieurs méthodes pour capturer et enregistrer les requêtes et réponses HTTP brutes, répondant à des exigences spécifiques et à des considérations de performances.

Option 1 : Middleware

Le middleware permet d'intercepter et de traiter les requêtes et les réponses. Vous pouvez créer un middleware qui consomme le corps de la requête dans un flux et le stocke. Pour le corps de la réponse, lisez-le dans un objet bytes et renvoyez une réponse personnalisée. Utilisez une BackgroundTask pour enregistrer les données de manière asynchrone afin d'éviter d'avoir un impact sur le temps de réponse.

Exemple :

async def some_middleware(request: Request, call_next):
    req_body = await request.body()
    response = await call_next(request)
    res_body = b''
    async for chunk in response.body_iterator:
        res_body += chunk
    task = BackgroundTask(log_info, req_body, res_body)
    return Response(content=res_body, background=task)

Option 2 : classe APIRoute personnalisée

Créez une classe APIRoute personnalisée pour manipuler les corps de requête et de réponse. Cette approche vous permet de limiter la journalisation à des routes spécifiques définies dans un APIRouter.

Exemple :

class LoggingRoute(APIRoute):
    async def custom_route_handler(request: Request) -> Response:
        req_body = await request.body()
        response = await original_route_handler(request)
        res_body = b''
        async for item in response.body_iterator:
            res_body += item
        task = BackgroundTask(log_info, req_body, res_body)
        response = Response(content=res_body, background=task)
        return response

Considérez les limites du stockage de corps de requête/réponse volumineux dans mémoire et utilisez une BackgroundTask pour éviter de bloquer le traitement des requêtes. Si nécessaire, limitez la journalisation à des itinéraires spécifiques ou excluez les points de terminaison qui renvoient des réponses en streaming.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!