2016-12-07 2 views
1

Как узнать, сколько активных задач в цикле событий? В документации я нахожу только asyncio.Task.all_tasks(), но это просто инкрементный счетчик:Подсчет активных задач в цикле событий

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()) 

Выход:

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

Ожидаемый выход:

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

ответ

4

all_tasks возвращает все зарегистрированные задачи, в том числе готовых из них , Вы можете просто отфильтровать задачи по done(), чтобы получить активные из них:

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()) 

который выводит:

Tasks count: 1 
Tasks count: 2 
Tasks count: 3 
Tasks count: 4 
Tasks count: 4 
Active tasks count: 1 
Смежные вопросы