Python-同时运行多个异步功能

Question

我本质上是在制作一个 pinger，它具有一个包含密钥/Webhook 对的二维列表，在 ping 一个密钥后，将响应发送到一个 webhook，

该二维列表如下所示：

some_list = [["key1", "webhook1"], ["key2", "webhook2"]]

我的程序本质上是一个循环，并且我'我不太确定如何在函数中旋转 some_list 数据。

这是我的脚本的一个小演示：

async def do_ping(some_pair):
    async with aiohttp.ClientSession() as s:
        tasks = await gen_tasks(s, some_pair)
        results = await asyncio.gather(*tasks*)
        sleep(10)
        await do_ping(some_pair)

我已经尝试过：

async def main(): 
    for entry in some_list: 
        asyncio.run(do_ping(entry))

但是由于 do_ping 函数是一个自调用循环，它只是一遍又一遍地调用第一个函数，并且永远不会得到给它之后的人。希望找到解决方案，无论是线程还是类似的，如果您有更好的方法来构造 some_list 值（我认为这将是一个字典），请随时放弃该反馈

Answer 1

您使方法递归 await do_ping(some_pair)，它永远不会结束 main 中的循环继续。我会像这样重构应用程序：

async def do_ping(some_pair):
    async with aiohttp.ClientSession() as s:
        while True:
            tasks = await gen_tasks(s, some_pair)
            results = await asyncio.gather(*tasks)
            await asyncio.sleep(10)


async def main(): 
    tasks = [do_ping(entry) for entry in some_list]
    await asyncio.gather(*tasks)


if __name__ == "__main__":
    asyncio.run(main())

或者，您可以将重复和睡眠逻辑移至 main 中：

async def do_ping(some_pair):
    async with aiohttp.ClientSession() as s:
        tasks = await gen_tasks(s, some_pair)
        results = await asyncio.gather(*tasks)


async def main(): 
    while True:
        tasks = [do_ping(entry) for entry in some_list]
        await asyncio.gather(*tasks)
        await asyncio.sleep(10)


if __name__ == "__main__":
    asyncio.run(main())

您还可以在调用睡眠之前启动任务，然后收集它们。这将使 ping 更一致地以 10 秒间隔开始，而不是 10 秒 + 收集结果所需的时间：

async def main(): 
    while True:
        tasks = [
            asyncio.create_task(do_ping(entry))
            for entry in some_list
        ]
        await asyncio.sleep(10)
        await asyncio.wait(tasks)

---

编辑正如 creolo 您应该只创建一个 ClientSession 对象。请参阅https://docs.aiohttp.org/en/stable/client_reference.html

Session封装了一个连接池（连接器实例），默认支持keepalive。除非您在应用程序的生命周期内连接到大量未知数量的不同服务器，否则建议您在应用程序的生命周期内使用单个会话，以便从连接池中受益。

async def do_ping(session, some_pair):
    tasks = await gen_tasks(session, some_pair)
    results = await asyncio.gather(*tasks)

async def main(): 
    async with aiohttp.ClientSession() as session:
        while True:
            tasks = [
                asyncio.create_task(do_ping(session, entry))
                for entry in some_list
            ]
            await asyncio.sleep(10)
            await asyncio.wait(tasks)