为了能到远方,脚下的每一步都不能少.|

bitterteaer

园龄:3年8个月粉丝:1关注:0

使用fastapi background task踩的第一个坑

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


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方法即可

本文作者:bitterteaer

本文链接:https://www.cnblogs.com/bitterteaer/p/17555421.html

版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。

posted @   bitterteaer  阅读(604)  评论(0编辑  收藏  举报
点击右上角即可分享
微信分享提示
评论
收藏
关注
推荐
深色
回顶
收起