2016-12-07 4 views
1

Wie kann ich herausfinden, wie viele aktive Tasks in der Ereignisschleife sind? In Dokumentation finde ich nur asyncio.Task.all_tasks() aber es ist einfach Inkrementzählers:Zähle aktive Tasks in Ereignisschleife

import asyncio 


async def coro(): 
    await asyncio.sleep(1) 


async def main(): 
    tasks = [] 
    print('Tasks count: ', len(asyncio.Task.all_tasks())) 
    for idx in range(3): 
     task = asyncio.ensure_future(coro()) 
     tasks.append(task) 
     print('Tasks count: ', len(asyncio.Task.all_tasks())) 
    await asyncio.gather(*tasks) 
    print('Tasks count: ', len(asyncio.Task.all_tasks())) 


loop = asyncio.get_event_loop() 
loop.run_until_complete(main()) 

Ausgang:

Tasks count: 1 
Tasks count: 2 
Tasks count: 3 
Tasks count: 4 
Tasks count: 4 

Erwartete Ausgabe:

Tasks count: 1 
Tasks count: 2 
Tasks count: 3 
Tasks count: 4 
Tasks count: 1 

Antwort

4

all_tasks kehrt alle eingetragenen Aufgaben, einschließlich zum fertigen . Sie können einfach die Aufgaben filtern, indem done() die Aktiven zu bekommen:

import asyncio 


async def coro(): 
    await asyncio.sleep(1) 


async def main(): 
    tasks = [] 
    print('Tasks count: ', len(asyncio.Task.all_tasks())) 
    for idx in range(3): 
     task = asyncio.ensure_future(coro()) 
     tasks.append(task) 
     print('Tasks count: ', len(asyncio.Task.all_tasks())) 
    await asyncio.gather(*tasks) 
    print('Tasks count: ', len(asyncio.Task.all_tasks())) 
    print('Active tasks count: ', len(
     [task for task in asyncio.Task.all_tasks() if not task.done()]) 
    ) 

loop = asyncio.get_event_loop() 
loop.run_until_complete(main()) 

Welche Ausgänge:

Tasks count: 1 
Tasks count: 2 
Tasks count: 3 
Tasks count: 4 
Tasks count: 4 
Active tasks count: 1