本文说明在 FastAPI 应用程序中使用 lifespan
事件来管理资源的加载和卸载。lifespan
事件允许你在应用启动时执行一些初始化代码,并在应用关闭时执行一些清理代码。这是通过使用异步上下文管理器实现的,具体来说,是通过 asynccontextmanager
装饰器创建一个异步生成器函数。
# 导入asynccontextmanager用于创建异步上下文管理器
from contextlib import asynccontextmanager
# 导入FastAPI,用于创建和管理Web应用
from fastapi import FastAPI
# 定义一个模拟的机器学习模型函数,它接受一个浮点数x并返回x乘以42的结果
def fake_answer_to_everything_ml_model(x: float):
return x * 42
# 创建一个空字典,用于存储机器学习模型
ml_models = {
}
# 使用asynccontextmanager装饰器定义一个异步上下文管理器函数lifespan
@asynccontextmanager
async def lifespan(app: FastAPI):
# 在异步上下文管理器中,"进入上下文"时加载机器学习模型
ml_models["answer_to_everything"] = fake_answer_to_everything_ml_model
# yield
yield
# 在异步上下文管理器中,"退出上下文"时清理机器学习模型,释放资源
ml_models.clear()
# 创建FastAPI应用实例,并将lifespan函数作为其生命周期管理器
app = FastAPI(lifespan=lifespan)
# 定义一个GET请求路由,用于处理预测请求
@app.get(