python进程池的使用

2022-10-31 09:31:10 浏览数 (2)

进程池(Pool)可以提供指定数量的进程供用户使用。主要是起限制进程数量的作用。当新的请求提交到pool时,如果进程池没有满,那么就可以创建一个新的进程来响应请求。如果进程池满了,那么就要等到有进程结束,才可以创建新的进程。

上代码

代码语言:javascript复制
import multiprocessing
import time


def process_func(process_id):
    print('process id %d start' % process_id)
    time.sleep(3)
    print('process id %d end' % process_id)


def main():
    pool =multiprocessing.Pool(processes=3)

    for i in range(10):
        #向进程池中添加要执行的任务
        pool.apply_async(process_func, args=(i, ))
    print('applied')
    #先调用close关闭进程池,不能再有新任务被加入到进程池中
    pool.close()

    #用join函数等待所有子进程结束
    pool.join()
    print('joined')

if __name__ == '__main__':
    main()

如果每次调用的都是同一个函数,可以使用Pool的map函数。

把上面的例子改一行代码就好了

代码语言:javascript复制
import multiprocessing
import time


def process_func(process_id):
    print('process id %d start' % process_id)
    time.sleep(3)
    print('process id %d end' % process_id)


def main():
    pool =multiprocessing.Pool(processes=3)

    #向进程池中添加要执行的任务
    pool.map(process_func, range(10))
    print('applied')
    #先调用close关闭进程池,不能再有新任务被加入到进程池中
    pool.close()

    #用join函数等待所有子进程结束
    pool.join()
    print('joined')

if __name__ == '__main__':
    main()

map的第一个参数是要执行的函数,第二个参数必须是一个可迭代对象。map方法会自动地帮助我们迭代第二个参数,并把迭代出的元素作为参数分批传递给第一个要执行的函数并执行。

0 人点赞