Python多任务代码示例

from multiprocessing import Pool  # 导入进程池模块
import os  # 导入系统模块
import time  # 导入时间模块
import random


def to_work(num):
    # 打印进程号
    print("----开始执行任务%d,进程的pid为%d,父进程的pid为%d----" % (num, os.getpid(), os.getppid()))
    # 模拟延时 random.random() 生成0-1之间的数
    time.sleep(random.random())
    print("第%d号进程完成任务%d第一阶段" % (os.getpid(), num))
    time.sleep(random.random())
    print("第%d号进程完成任务%d第二阶段" % (os.getpid(), num))
    time.sleep(random.random())
    print("第%d号进程完成任务%d第三阶段" % (os.getpid(), num))
    time.sleep(random.random())
    print("----任务%d执行完毕----" % (num))


def main():
    # 创建一个最多开启3进程的进程池
    po = Pool(3)
    # 将5个任务添加到进程池
    for i in range(1, 6):
        # async非堵塞添加,to_work为参数名,i为传递的参数,单个参数一定要加逗号!一定要加逗号!一定要加逗号!
        po.apply_async(to_work, (i,))

    print("----开始----")

    # 关闭进程池,不再接收新的任务,开始执行任务
    po.close()

    # 主进程等待所有子进程结束
    po.join()
    print("----结束----")


if __name__ == "__main__":
    main()