Quando si utilizza asyncio, come si fa a consentire tutte le attività in esecuzione per finire prima di arrestare il ciclo degli eventi

Question

Ho il seguente codice:

@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 

ho eseguito questa funzione fino al completo. Il problema si verifica quando viene impostato lo spegnimento: la funzione viene completata e le attività in sospeso non vengono mai eseguite. (Si vede questo come un errore

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

). Come posso programmare un arresto in modo corretto?

Per fornire un contesto, sto scrivendo un monitor di sistema che legge/proc/stat ogni 5 secondi, calcola l'utilizzo della CPU in quel periodo e quindi invia il risultato a un server. Voglio continuare a pianificare questi lavori di monitoraggio fino a quando non ricevo il sigterm, quando smetto di pianificare, attendo che tutti i lavori in corso finiscano, e esco con garbo.

Answer 1

È possibile recuperare attività non completate ed eseguire nuovamente il ciclo fino al termine, quindi chiudere il ciclo o uscire dal programma.

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

• attesa è un elenco di attività in sospeso.
• asyncio.gather() consente di attendere più attività contemporaneamente.

Se si vuole garantire tutte le attività sono completate all'interno di un coroutine (forse hai un coroutine "principale"), si può fare in questo modo, per esempio:

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

Inoltre questo caso, dal momento che tutte le attività sono creati nella stessa coroutine, che già hanno accesso alle funzioni:

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