在 Python FastAPI 中记录特定路由的原始 HTTP 请求/响应
问题:
我们正在开发使用 FastAPI 的 Web 服务将部署在 Kubernetes 中。出于审计目的,我们需要记录特定路由请求和响应的原始 JSON 正文。请求和响应 JSON 正文大小约为 1MB,日志记录过程不会显着影响响应时间至关重要。
解决方案:
选项 1:使用中间件
-
创建一个中间件:
定义一个函数并使用 @app.middleware("http") 装饰器来处理传入请求和传出响应。
-
捕获请求正文:
使用 request.body() 或 request.stream() 捕获请求正文。
-
处理响应正文:
将响应正文作为字节对象读取,并向客户端返回自定义响应。
-
日志数据:
使用后台任务来记录请求以及对文件或数据库的响应正文。
选项 2:使用自定义 APIRoute类
-
定义自定义 APIRoute:
创建扩展 APIRoute 基类的自定义 APIRoute 类,允许修改请求和响应正文。 -
处理请求正文:
在自定义路由处理程序中,捕获在请求正文到达端点处理程序之前。
-
处理响应正文:
修改响应正文并创建一个新的 Response 对象。如果原始响应是 StreamingResponse,则向流迭代器添加日志记录功能。
-
后台日志记录:
将日志记录函数作为后台任务附加到响应对象。
注意:
考虑请求和响应的大小体,因为大的有效负载可能会导致服务器和客户端的内存问题或延迟。可能有必要将日志记录限制为特定路由或从日志记录中排除流响应。
以上是如何在 FastAPI 中有效记录原始 HTTP 请求/响应 JSON 以便审核?的详细内容。更多信息请关注PHP中文网其他相关文章!