FastAPI（55）- Events: startup - shutdown 启动/关闭事件

小菠萝测试笔记

发布于 2021-10-09 10:55:42

1.1K0

发布于 2021-10-09 10:55:42

文章被收录于专栏：自动化、性能测试自动化、性能测试

背景

可以定义需要在应用程序启动之前或应用程序关闭时执行的事件处理程序（函数）
这些函数可以用 async def 或普通 def
注意：只会执行主应用程序的事件处理程序，而不会执行子应用程序

实际代码

#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
# author: 小菠萝测试笔记
# blog:  https://www.cnblogs.com/poloyy/
# time: 2021/10/4 7:26 下午
# file: 45_event.py
"""
import uvicorn
from fastapi import FastAPI

app = FastAPI()

items = {}


# 添加在应用程序启动之前运行的函数
@app.on_event("startup")
async def startup_event():
    print("启动应用程序啦")
    items["foo"] = {"name": "Fighters"}
    items["bar"] = {"name": "Tenders"}


# 添加在应用程序关闭时运行的函数
@app.on_event("shutdown")
async def shutdown_event():
    print("关闭应用程序啦")
    with open("log.txt", mode="a") as log:
        log.write("Application shutdown")


@app.get("/items/{item_id}")
def read_items(item_id: str):
    return items[item_id]


if __name__ == '__main__':
    uvicorn.run(app="45_event:app", reload=True, host="127.0.0.1", port=8080)