fastapi怎么样开开启一个一直运行的子任务而不占用阻塞主程序运行
import threading
from fastapi import FastAPI
app = FastAPI()
def background_task():
while True:
# 你的持续运行任务逻辑
print(“Background task is running…”)
# 可以添加一些休眠以避免过度占用CPU
# time.sleep(1)
@app.on_event(“startup”)
async def startup_event():
# 在应用启动时启动后台任务
thread = threading.Thread(target=background_task)
thread.daemon = True # 设置为守护线程,确保在主程序退出时也能终止
thread.start()
@app.get(“/”)
async def read_root():
return {“Hello”: “World”}
import asyncio
from fastapi import FastAPI
app = FastAPI()
async def background_task():
while True:
# 你的异步持续运行任务逻辑
print(“Async background task is running…”)
await asyncio.sleep(1) # 异步休眠,不会阻塞事件循环
@app.on_event(“startup”)
async def startup_event():
# 在应用启动时启动后台任务
app.state.background_task = asyncio.create_task(background_task())
@app.get(“/”)
async def read_root():
return {“Hello”: “World”}
确保在应用关闭时取消所有后台任务
@app.on_event(“shutdown”)
async def shutdown_event():
if hasattr(app.state, “background_task”):
app.state.background_task.cancel()
try:
await app.state.background_task
except asyncio.CancelledError:
pass