fastapi 结合 celery 将异步过程转变为同步过程

使用 asyncio 的协程方式查询任务结果,查询超时时间为 120 s,每 200 ms 查询一次:


# 使用 asyncio 的协程方式查询任务结果,查询超时时间为 120 s,每 200 ms 查询一次
@router.post('/backgroud_type', response_model=BackgroundTypeResponse)
async def backgroud_type(request: Request, backgroundTypeRequest: BackgroundTypeRequest):
    logging.info(f"{backgroundTypeRequest}")
    task = request.app.state.celery_app.send_task(
        name="tasks.backgroud_type",
        priority=backgroundTypeRequest.priority,
        args=[backgroundTypeRequest.dict()]
    )
    try:
        return await async_check_task_result(task.id, request.app.state.celery_app)
    except asyncio.TimeoutError:
        raise HTTPException(status_code=408, detail="Request timed out")

async def async_check_task_result(task_id, celery_app, timeout=120):
    start_time = asyncio.get_running_loop().time()
    result_async = AsyncResult(task_id, app=celery_app)
    while True:
        if result_async.ready():
            return {
                'task_id': task_id,
                'result': result_async.get()
            }
        if (asyncio.get_running_loop().time() - start_time) > timeout:
            raise asyncio.TimeoutError
        await asyncio.sleep(0.2)

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值