Podczas korzystania asyncio, jak można pozwolić wszystkie zadania uruchomione, aby zakończyć przed zamknięciem pętli zdarzeń

Question

Mam następujący kod:

@asyncio.coroutine 
def do_something_periodically(): 
    while True: 
     asyncio.async(my_expensive_operation()) 
     yield from asyncio.sleep(my_interval) 
     if shutdown_flag_is_set: 
      print("Shutting down") 
      break 

uruchomić tę funkcję aż do zakończenia. Problem występuje, gdy jest ustawione wyłączanie - funkcja zostaje zakończona, a wszelkie oczekujące zadania nigdy nie są uruchamiane. (Widać to jako błąd:

task: <Task pending coro=<report() running at script.py:33> wait_for=<Future pending cb=[Task._wakeup()]>> 

). Jak poprawnie zaplanować zamknięcie?

Aby podać pewien kontekst, piszę monitor systemu, który odczytuje z/proc/stat co 5 sekund, oblicza użycie procesora w tym okresie, a następnie wysyła wynik do serwera. Chcę kontynuować planowanie tych zadań monitorowania, dopóki nie otrzymam sigterm, gdy zatrzymam planowanie, poczekam na zakończenie wszystkich bieżących zadań i zakończę wdzięcznie.

Answer 1

Możesz odzyskać niedokończone zadania i ponownie uruchomić pętlę, aż do zakończenia, a następnie zamknąć pętlę lub zamknąć program.

pending = asyncio.Task.all_tasks() 
loop.run_until_complete(asyncio.gather(*pending)) 

• oczekiwaniu znajduje się lista oczekujących zadań.
• asyncio.gather() pozwala czekać na kilka zadań jednocześnie.

Jeśli chcesz, aby wszystkie zadania są wykonywane wewnątrz współprogram (może masz „main” współprogram), można zrobić to w ten sposób, na przykład:

@asyncio.coroutine 
def do_something_periodically(): 
    while True: 
     asyncio.async(my_expensive_operation()) 
     yield from asyncio.sleep(my_interval) 
     if shutdown_flag_is_set: 
      print("Shutting down") 
      break 

    yield from asyncio.gather(*asyncio.Task.all_tasks()) 

Również w tym przypadku, ponieważ wszystkie zadania są tworzone w tym samym współprogram, masz już dostęp do zadań:

@asyncio.coroutine 
def do_something_periodically(): 
    tasks = [] 
    while True: 
     tasks.append(asyncio.async(my_expensive_operation())) 
     yield from asyncio.sleep(my_interval) 
     if shutdown_flag_is_set: 
      print("Shutting down") 
      break 

    yield from asyncio.gather(*tasks)