Khi sử dụng asyncio, làm thế nào để bạn cho phép tất cả các nhiệm vụ chạy để kết thúc trước khi tắt vòng lặp sự kiện

Question

Tôi có đoạn mã sau:

@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 

tôi chạy chức năng này cho đến khi hoàn tất. Vấn đề xảy ra khi tắt máy được cài đặt - chức năng hoàn thành và mọi tác vụ đang chờ xử lý sẽ không bao giờ chạy. (Bạn thấy đây là lỗi

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

). Làm cách nào để lên lịch tắt máy một cách chính xác?

Để đưa ra một số ngữ cảnh, tôi viết một màn hình hệ thống đọc từ/proc/stat sau mỗi 5 giây, tính toán mức sử dụng CPU trong khoảng thời gian đó, và sau đó gửi kết quả đến máy chủ. Tôi muốn tiếp tục lập kế hoạch cho các công việc giám sát này cho đến khi tôi nhận được sigterm, khi tôi ngừng lập kế hoạch, hãy đợi tất cả các công việc hiện tại để hoàn thành và thoát ra một cách duyên dáng.

Answer 1

Bạn có thể lấy các nhiệm vụ chưa hoàn thành và chạy vòng lặp lại cho đến khi họ hoàn tất, sau đó đóng vòng lặp hoặc thoát khỏi chương trình của bạn.

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

• đang chờ xử lý là danh sách các tác vụ đang chờ xử lý.
• asyncio.gather() cho phép đợi nhiều tác vụ cùng một lúc.

Nếu bạn muốn đảm bảo tất cả các nhiệm vụ được hoàn thành bên trong một coroutine (có thể bạn có một "chính" coroutine), bạn có thể làm theo cách này, ví dụ:

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

Ngoài ra, trong trường hợp này, vì tất cả các nhiệm vụ được tạo ra trong coroutine cùng, bạn đã có quyền truy cập vào các nhiệm vụ:

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