使用 asyncio function 时从 while 循环返回一个值

Question

我正在尝试同时连接和接收来自多个 websocket 的消息。 为此，我使用 asyncio 制作它，它可以正确打印消息。 但问题是我只能打印它，不能退货。

我正在努力的伪代码的简化示例如下：

import websockets
import json

symbols_id = [1,2]

## LOOP RUNNING EXAMPLE OF ASYNCIO
async def get_connect(symbols_id):
    tasks = []
    for _id in symbols_id:
        print('conncetion to', _id)
        if _id == 1:
            a = 0
        elif _id == 2:
            a = 200
        tasks.append(asyncio.create_task(_loop(a)))

    return tasks

async def _loop(a):
    while True:
        print(a)
        a+=1
        await asyncio.sleep(2.5)

async def ping_func():
    while True:
        print('------ ping')
        await asyncio.sleep(5)


async def main():
    tasks = await get_connect(symbols_id)
    asyncio.create_task(ping_func())
    await asyncio.gather(*tasks)

asyncio.run(main())

从上面的代码可以看出，我使用print(a)在每个循环中打印a 。 我测试return a而不是print(a)但它没有帮助。

谢谢

Answer 1

yield a ？ return a将退出 function 和循环， yield通常是你在 asyncio 中想要的循环任务

Answer 2

最后我找到了使用yield和async for在每个循环中读取数据的方法。 通过将代码更改为以下代码，它将正常工作。

import websockets
import json

symbols_id = [1,2]

global a
a=0

## LOOP RUNNING EXAMPLE OF ASYNCIO
async def get_connect(symbols_id):
    tasks = []
    for _id in symbols_id:
        print('conncetion to', _id)
        if _id == 1:
            a = 0
        elif _id == 2:
            a = 200
        tasks.append(asyncio.create_task(_loop(a)))

    return tasks

async def _loop(param):
    global a
    a = param
    while True:
        print(a)
        a+=1
        await asyncio.sleep(2.5)

async def ping_func():
    while True:
        print('------ ping')
        await asyncio.sleep(5)

async def get_result():
    global a
    while True:
        yield a
        await asyncio.sleep(1)

async def main():
    tasks = await get_connect(symbols_id)
    asyncio.create_task(ping_func())
    async for x in get_result():
        print(x)
    await asyncio.gather(*tasks)

asyncio.run(main())

我对如何在另一个代码片段中使用此代码片段生成的数据感到困惑。 我发现的是：

1- 生成的数据可以通过全局变量访问。

2- 通过定义 class 和属性，可以从代码的每个部分访问它。