关闭异步循环

Question

import asyncio
import aiohttp
aut_token = ("token")
tasks = []
iter_flag = True
interval = 0
seq = 0
class WAPI:

    async def receiver(WAPI_S):
        async for msg in WAPI_S:
            global interval
            global seq
            data = msg.json()
            seq = data.get("s")
            if data.get("op") == 10:
                interval = data.get("d").get("heartbeat_interval") / 1000 
            if data.get("op") == 11:
                pass
            raise aiohttp.ClientError

    async def heartbeating(WAPI_S):
        while iter_flag:
            await WAPI_S.send_json({
                            "op": 1,
                            "d": seq
                        })
            
            await asyncio.sleep(interval)

    async def event_manager():
        loop = asyncio.get_running_loop()
        try:
            async with aiohttp.ClientSession() as session:
                async with session.ws_connect("url") as WAPI_S: 
                    task_receive = loop.create_task(WAPI.receiver(WAPI_S)); task_heartbeating = loop.create_task(WAPI.heartbeating(WAPI_S))
                    tasks.append(task_receive); tasks.append(task_heartbeating)
                    await asyncio.gather(*tasks)
        except aiohttp.ClientError:
            global iter_flag
            iter_flag = False
            await asyncio.sleep(interval)
            for task in tasks:
                task.cancel()
            try:
                loop.close()
            except:
                loop.stop()
                
asyncio.run(WAPI.event_manager())

我正在尝试通过循环关闭来实现捕获 ClientError 异常，但是，loop.close 抛出“RuntimeError：事件循环在 Future 完成之前停止。”

如何正确实现拦截？

Answer 1

您无需手动跟踪任务，只需使用

asyncio.all_tasks():

Return a set of not yet finished Task objects run by the loop.

然后是：

pending = asyncio.all_tasks()
c in pending:
    wait_for(c, timeout=5)

此外，您正试图在循环中停止循环。这是我最常使用的模式：

async def main():
   <do some stuff>

loop = asyncio.get_event_loop()
try:
    loop.run_until_complete(main())
except ExceptionYouWantToHandle:
    <cleaning up>
finally:
    loop.stop()

而在 event_manager 中，您只需 return 一旦获得执行或通过异常

关闭异步循环

Close asyncio loop

python

api

asynchronous

python-asyncio

aiohttp