关于fastapi异步接口卡死的坑及解决

简介: 开发任务是使用fastapi去写一个对工业设备(PLC)的通信接口,方便其他后端服务与设备对接,将设备的功能抽象出来供MES调用。通信协议是使用modbus TCP,由于fastapi是异步框架,很多以前在同步函数里开发的代码移植过来发现出现了异常,这也是不断踩坑的过程,问题解决之后也能体会到异步框架的优美与高效。

背景

开发任务是使用fastapi去写一个对工业设备(PLC)的通信接口,方便其他后端服务与设备对接,将设备的功能抽象出来供MES调用。


通信协议是使用modbus TCP,由于fastapi是异步框架,很多以前在同步函数里开发的代码移植过来发现出现了异常,这也是不断踩坑的过程,问题解决之后也能体会到异步框架的优美与高效。

问题细节

在视图函数中我们需要创建一个socket连接,使用它跟plc进行通信,完毕之后还要记得关闭该连接,否则就会造成异步框架的卡死,其他请求就进不来了。


刚开始我的代码是这样的:

@router.put("/start/{id}")
async def start(
    id:int,
    batsize:int=Body(default=15,description='批次大小'),
    noksize:int=Body(default=3,description='最大连续失败次数')
    ):
    dip = await PLC.get(pk=id)   # 配置数据库中查询设备网络信息
    zmqconn = get_conn(dip.host,dip.port)   # 获取网络连接
    # 发送命令
    try:
        ret = zmqconn.request(action='start_batch',rqargs={'batch_size':batsize,'nok_num':noksize})
    except Exception as e:
        raise HTTPException(status_code=503,detail=str(e))
    else:
        return ret

写完之后我自己测试了一下,发现是可以正常发起命令的,而且接口也返回了期望的结果,但是我快速点了几下,发现问题就来了,第六下的时候,接口卡住无法返回,异步函数中也不执行任何动作,这在之前的同步代码中是没有出现的。


后来我经过调试发现是socket连接并没有成功创建,怀疑是之前的连接没有关闭,导致了网络阻塞,于是加入了关闭连接的代码。

@router.put("/start/{id}")
async def start(
    id:int,
    batsize:int=Body(default=15,description='批次大小'),
    noksize:int=Body(default=3,description='最大连续失败次数')
    ):
    dip = await PLC.get(pk=id)
    zmqconn = get_conn(dip.host,dip.port)
    try:
        ret = zmqconn.request(action='start_batch',rqargs={'batch_size':batsize,'nok_num':noksize})
    except Exception as e:
        raise HTTPException(status_code=503,detail=str(e))
    finally:
        zmqconn.close()
    return ret

这个时候就完全没有卡住的问题了,接口可以一直点。


所以对于异步函数,我们一定要注意手动释放资源,否则可能会造成意想不到的问题。

fastapi依赖注入

但是这又出现一个问题,我们的服务是经常用到前面的socket框架的,其他接口也要用到,总不能每个接口都把这段代码复制粘贴一边吧。


fastapi作为最优秀的异步框架,有一个很大的特点就是非常苛刻的代码复用性,它在设计的时候考虑到了重复代码的问题。


至于苛刻到什么程度呢,举个例子,如果不同的接口用了相同的一组参数,那么你可以把这组参数进行封装,写到依赖里面,用的时候引入一下,那么你新的接口就有了一组一模一样的参数。


这里我们利用fastapi的依赖注入,把接口参数以及socket连接都封装起来,供其他接口调用。

依赖代码

async def get_conn_dep(
    id:int = Path(description='路径参数,设备id'),
):
    dip = await PLC.get(pk=id)
    try:
        zmqconn = get_conn(dip.host,dip.port)
        yield zmqconn                # 向接口函数注入链接
    finally:
        zmqconn.close()     # 接口函数执行完毕后会异步执行关闭

而我们的接口函数就可以非常精简

@router.put("/start/{id}")
async def start(
    zmqconn = Depends(get_conn_dep),     # 这一行直接复用了一堆代码以及相应的路径字段,简直太优雅了
    batsize:int=Body(default=15,description='批次大小'),
    noksize:int=Body(default=3,description='最大连续失败次数')
    ):
    try:
        ret = zmqconn.request(action='start_batch',rqargs={'batch_size':batsize,'nok_num':noksize})
    except Exception as e:
        raise HTTPException(status_code=503,detail=str(e))
    else:
        return ret

我们使用swagger疯狂点击该接口,并传入不同的参数进行测试,发现接口响应非常流畅,异常处理的时候也返回了我们想要的错误信息。


我们上面的代码使用了fastapi异步接口(据说性能逼近go),并使用了tortoise异步ORM,tortoise的使用与django-orm几乎一样,但它是一个纯异步orm,这两个配合起来开发简直是非常的舒服。

  • fastapi.
  • tortoise-orm.
相关文章
|
NoSQL Redis
使用slowapi对FastApi的接口进行限速
使用slowapi对FastApi的接口进行限速
2372 0
|
API 数据库
FastAPI(63)- Concurrency and async / await 并发、异步/等待
FastAPI(63)- Concurrency and async / await 并发、异步/等待
645 0
|
3月前
|
API Python
在线问诊 Python、FastAPI、Neo4j — 提供咨询接口服务
在线问诊 Python、FastAPI、Neo4j — 提供咨询接口服务
36 0
|
JSON API 数据格式
使用(Python)FastAPI快速构建你的后端接口服务
使用(Python)FastAPI快速构建你的后端接口服务
1024 0
|
JSON 数据安全/隐私保护 数据格式
python fastapi 入门教程,每个案例都使用postman进行测试写的接口
python fastapi 入门教程,每个案例都使用postman进行测试写的接口
|
网络安全 Windows
基于fastapi实现6个接口(token拦截, 2个业务流程,接口参数依赖校验)已经通过postman测试,记录部署服务器和windows,用于pytest接口自动化框架的接口测试对象
基于fastapi实现6个接口(token拦截, 2个业务流程,接口参数依赖校验)已经通过postman测试,记录部署服务器和windows,用于pytest接口自动化框架的接口测试对象
|
JSON 前端开发 JavaScript
fastapi高性能异步框架,极速上手
fastapi高性能异步框架,极速上手
|
NoSQL 测试技术 Redis
FastAPI(八十四)实战开发《在线课程学习系统》--接口测试(下)
FastAPI(八十四)实战开发《在线课程学习系统》--接口测试(下)
FastAPI(八十四)实战开发《在线课程学习系统》--接口测试(下)
|
存储 测试技术 数据安全/隐私保护
FastAPI(八十三)实战开发《在线课程学习系统》--注册接口单元测试
FastAPI(八十三)实战开发《在线课程学习系统》--注册接口单元测试
FastAPI(八十三)实战开发《在线课程学习系统》--注册接口单元测试
|
测试技术 API 数据库
FastAPI(61)- 异步测试
FastAPI(61)- 异步测试
618 0