当前位置 : 主页 > 编程语言 > python >

如何在FastAPI中实现请求日志记录和监控

来源:互联网 收集:自由互联 发布时间:2023-07-30
如何在FastAPI中实现请求日志记录和监控 引言: FastAPI是一个基于Python 3.7+的高性能Web框架,它提供了许多强大的功能和特性,包括自动化的请求和响应模型验证、安全性、性能优化等。

如何在FastAPI中实现请求日志记录和监控

引言:
FastAPI是一个基于Python 3.7+的高性能Web框架,它提供了许多强大的功能和特性,包括自动化的请求和响应模型验证、安全性、性能优化等。在实际开发中,我们经常需要在应用程序中记录请求日志以便进行排错和监控分析。本文将介绍如何在FastAPI中实现请求日志记录和监控,并提供相应的代码示例。

一、安装依赖包
在开始之前,我们需要安装一些必要的依赖包。打开终端,执行以下命令:

pip install fastapi uvicorn loguru

其中,loguru是一个易用的日志记录库,我们将使用它来记录请求日志。

二、创建一个FastAPI应用程序
首先,我们需要创建一个FastAPI应用程序。在项目目录下,创建一个名为main.py的文件,并写入以下代码:

from fastapi import FastAPI

app = FastAPI()

@app.get("/")
async def root():
    return {"message": "Hello World"}

三、记录请求日志
接下来,我们将使用loguru库来记录请求日志。在main.py文件中添加以下代码:

from loguru import logger
import uvicorn
from fastapi import FastAPI

app = FastAPI()

@app.on_event("startup")
async def startup_event():
    logger.add("logs/request.log", rotation="10 MB")

@app.get("/")
async def root():
    logger.info("Hello World")
    return {"message": "Hello World"}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)

上述代码中,我们首先导入loguru库的logger对象,并添加一个文件日志记录器。我们指定了记录日志文件的路径为logs/request.log,并设置了日志文件的最大大小为10MB。然后,在root()函数中,我们使用logger.info()方法记录请求日志。

四、启动应用程序
保存main.py文件并返回终端,执行以下命令启动FastAPI应用程序:

uvicorn main:app --reload

终端将输出应用程序的访问URL,如http://127.0.0.1:8000。在浏览器中访问该URL,我们将看到"Hello World"消息。打开logs/request.log文件,我们将看到请求日志的记录。

五、监控请求
除了记录请求日志,我们还可以监控请求的处理时间和状态码。为了实现这个功能,我们需要使用FastAPI提供的Middleware。在main.py文件中添加以下代码:

from loguru import logger
import time
import uvicorn
from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware

app = FastAPI()

@app.on_event("startup")
async def startup_event():
    logger.add("logs/request.log", rotation="10 MB")

@app.on_event("shutdown")
async def shutdown_event():
    logger.remove(handler_id="request_logger")

@app.middleware("http")
async def log_request(request: Request, call_next):
    start_time = time.time()
    response = await call_next(request)
    process_time = time.time() - start_time

    logger.info(
        "Request: {method} {url} {status_code} {process_time:.2f}ms",
        method=request.method,
        url=request.url,
        status_code=response.status_code,
        process_time=process_time * 1000,
    )

    return response

@app.get("/")
async def root():
    logger.info("Hello World")
    return {"message": "Hello World"}

if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)

上述代码中,我们首先导入time模块,并在root()函数中添加了一个睡眠时间,以模拟请求处理时间。然后,我们在log_request()中间件函数中添加了计算请求处理时间和记录请求状态码的逻辑。在shutdown_event()函数中,我们删除了之前添加的日志记录器。

现在,重新启动FastAPI应用程序,并在浏览器中访问应用程序的URL。在浏览器中刷新页面,打开logs/request.log文件,我们将看到包含请求方法、URL、状态码和处理时间的请求日志记录。

结论:
本文介绍了如何在FastAPI中实现请求日志记录和监控。我们使用loguru库来记录请求日志,并借助FastAPI的Middleware来监控请求的处理时间和状态码。通过这些功能,我们可以更好地追踪和监控应用程序的请求和响应。

上一篇:Python中的ORM框架Pony ORM详解
下一篇:没有了
网友评论