FastAPI 启动事件 tart_event(),停止事件shutdown_event()

作用

需要在应用程序启动之前或者关闭时进行，例如数据库迁移，定时任务·····

实际代码

main.py

import uvicorn
 
from app import create_app
 
app = create_app()
 
if __name__ == '__main__':
    uvicorn.run(
        app='main:app',
        host='0.0.0.0',
        port=8000,
        reload=True
    )

#!/usr/bin/python
# -*- coding: utf-8 -*-
# @time    : 2023/2/3 11:28
# @author  : pugongying
# @description : 初始化项目
 
from fastapi import FastAPI
 
from app.xxx import task_init
from app.config import log_init
from app.middleware import middleware_init
from app.plan import plan
from app.routers import router_init
 
 
async def start_event():
    task_init()
    plan.run_plan()
    print('系统启动')
 
 
async def shutdown_event():
    print('系统关闭')
 
 
def create_app():
    app = FastAPI(title="xxx服务",
                  description="xxx服务接口文档",
                  version="1.0.0",
                  on_startup=[start_event],
                  on_shutdown=[shutdown_event]
                  )
 
    # 初始化日志
    log_init()
 
    # 初始化路由
    router_init(app)
 
    # 初始化中间件
    middleware_init(app)
 
    # 建表
    # db_init(app)
 
    return app