526互联

使用fastapi background task踩的第一个坑

发布时间 2023-07-15 01:02:57作者: bitterteaer

场景复原（这份代码可以和我之前的业务产生同样的效果）


async def task():
    print("1......")
    time.sleep(5)
    print("ok")
    

@router.post("/1")
async def crate(background_tasks: BackgroundTasks):
    s = time.time()
    background_tasks.add_task(task)
    # task()
    t = time.time() - s
    return {"success": True, "time": t}


@router.post("/2")
async def crate(background_tasks: BackgroundTasks):
    s = time.time()
    # task()
    t = time.time() - s
    return {"success": True, "time": t}

产生的后果

本来应该是后台任务，但是这样写却会阻塞主线程，导致第二个请求到来时会需要等待第一个task完成

原因分析

task 定义为 async ，这意味着 fastapi（或者更确切地说是 starlette）将在 asyncio 事件循环中运行它。因此程序会被sleep阻塞，所以只要它正在运行，它就会阻塞事件循环。

解决方法

将task 不定义为async方法即可