python 非阻塞模式 進程池

"""
非阻塞模式：當進程池有空閑，就將任務放入，它會繼續執行主進程
"""
import os
import time
from multiprocessing import Pool
from random import random

container = []


def task(name):
    print("進行任務{}  進程ID:{}".format(name, os.getpid()))
    start = time.time()
    # random() 的值是0到1之間，這里就表示休眠 0到2秒
    time.sleep(random() * 2)
    end = time.time()
    return "完成任務:{}  用時:{}  進程ID:{}".format(name, (end - start), os.getpid())


def callback_func(n):
    """將進程返回的內容n，放進這個函數內運行"""
    container.append(n)  # 添加返回內容到container，這里也可以對返回的函數進行其它的處理


if __name__ == '__main__':
    pool = Pool(5)  # 創建一個進程池，里面放5個進程
    tasks = ["任務1", "任務2", "任務3", "任務4", "任務5", "任務6", "任務7"]
    for t in tasks:
        # apply_async 非阻塞模式，只要存在沒有使用的進程，就將任務放進去
        pool.apply_async(task, args=(t,), callback=callback_func)

    pool.close()
    pool.join()  # 將進程池插入到這里，只要當它里面的任務全部都執行完，才繼續

    for c in container:
        print(c)

    print("over")