一篇文章浅析Python自带的线程池和进程池

前言

成都创新互联公司专注为客户提供全方位的互联网综合服务,包含不限于成都做网站、网站建设、外贸营销网站建设、湖州网络推广、微信平台小程序开发、湖州网络营销、湖州企业策划、湖州品牌公关、搜索引擎seo、人物专访、企业宣传片、企业代运营等,从售前售中售后,我们都将竭诚为您服务,您的肯定,是我们最大的嘉奖;成都创新互联公司为所有大学生创业者提供湖州建站搭建服务,24小时服务热线:18982081108,官方网址:www.cdcxhl.com

大家好,我是星期八。

我们都知道,不管是Java,还是C++,还是Go,还是Python,都是有线程这个概念的。

但是我们知道,线程是不能随便创建的,就像每招一个员工一样,是有代价的,无限制招人肯定最后各种崩溃。

所以通常情况下,我们会引出线程池这个概念。

本质就是我就招了几个固定的员工,给他们派活,某一个人的活干完了再去任务中心领取新的活。

防止任务太多,一次性招太多工人,最后系统崩溃。

开心一刻

理想的多线程

实际的多线程

from concurrent.futures import ...

可能也是因为线程池这个东西用的越来越多了吧,从Python3.2+之后,就成了内置模块。

对的,直接就能使用,不需要pip进行安装什么的。

concurrent.futures下面主要有俩接口。

  • ThreadPoolExecutor 线程池。
  • ProcessPoolExecutor进程池。

这里可没有什么所谓的异步池。

个人看法:虽然异步的性能很高,但是目前除了Go以外,其他实现的都不是太好,用法上面有些怪异,当然,你们可以说我菜,我承认。

线程池

示例代码

 
 
 
 
  1. import time 
  2. from concurrent.futures import ThreadPoolExecutor 
  3. import random 
  4.  
  5. # max_workers表示工人数量,也就是线程池里面的线程数量 
  6. pool = ThreadPoolExecutor(max_workers=10) 
  7. # 任务列表 
  8. task_list = ["任务1", "任务2", "任务3", "任务4", ] 
  9.  
  10.  
  11. def handler(task_name): 
  12.     # 随机睡眠,模仿任务执行时间不确定性 
  13.     n = random.randrange(5) 
  14.     time.sleep(n) 
  15.     print(f"任务内容:{task_name}") 
  16.  
  17.  
  18. if __name__ == '__main__': 
  19.     # 遍历任务, 
  20.     for task in task_list: 
  21.         """ 
  22.             交给函数处理,submit会将所有任务都提交到一个地方,不会阻塞 
  23.             然后线程池里面的每个线程会来取任务, 
  24.             比如:线程池有3个线程,但是有5个任务 
  25.             会先取走三个任务,每个线程去处理 
  26.             其中一个线程处理完自己的任务之后,会再来提交过的任务区再拿走一个任务 
  27.         """ 
  28.         pool.submit(handler, task) 
  29.     print("main执行完毕") 

执行结果

 

发现的问题

其实这个就是并发的,不要怀疑,但是你有没有发现个问题,main先执行,这说明啥?

这说明,我main跑完之后,是不管子线程的死活的。

那能不能设置一下,所有的子线程都执行完之后,main函数在执行完?

当然可以,需要一个参数即可。

 
 
 
 
  1. pool.shutdown() 

要完成上述的问题,我们需要一个参数,加上这个参数之后。

就可以让主线程等待所有子线程执行完之后,主线程再执行完。

示例代码

 
 
 
 
  1. ... 
  2. if __name__ == '__main__': 
  3.     # 遍历任务, 
  4.     for task in task_list: 
  5.         """ 
  6.             交给函数处理,submit会将所有任务都提交到一个地方 
  7.             然后线程池里面的每个线程会来取任务, 
  8.             比如:线程池有3个线程,但是有5个任务 
  9.             会先取走三个任务,每个线程去处理 
  10.             其中一个线程处理完自己的任务之后,会再来提交过的任务区再拿走一个任务 
  11.         """ 
  12.         pool.submit(handler, task) 
  13.     pool.shutdown() 
  14.     print("main执行完毕") 

主要就是13行的pool.shutdown()。

执行结果

 

这次结果就是我们想要的了,hhh!!!

 
 
 
 
  1. add_done_callback 

add_done_callback可以理解为是回调函数,线程执行完之后,会自动调用指定的回调函数。

并且能拿到线程执行函数的返回值。

有什么用,我也没用过,怪我才疏学浅叭。

示例代码

 
 
 
 
  1. import time 
  2. from concurrent.futures import ThreadPoolExecutor 
  3. import random 
  4. from concurrent.futures._base import Future 
  5.  
  6. # max_workers表示工人数量,也就是线程池里面的线程数量 
  7. pool = ThreadPoolExecutor(max_workers=10) 
  8. # 任务列表 
  9. task_list = ["任务1", "任务2", "任务3", "任务4", ] 
  10.  
  11.  
  12. def handler(task_name): 
  13.     # 随机睡眠,模仿任务执行时间不确定性 
  14.     n = random.randrange(5) 
  15.     time.sleep(n) 
  16.     print(f"任务内容:{task_name}") 
  17.     return f"任务内容:{task_name}" 
  18.  
  19.  
  20. def done(res: Future): 
  21.     print("done拿到的返回值:", res.result()) 
  22.  
  23.  
  24. if __name__ == '__main__': 
  25.     # 遍历任务, 
  26.     for task in task_list: 
  27.         futrue = pool.submit(handler, task)  # type:Future 
  28.         futrue.add_done_callback(done) 
  29.     pool.shutdown() 
  30.     print("main执行完毕") 

注意:第17,27,28行代码!

执行效果

 

我想,可能通常用在一些善后工作叭。

多进程方式

其实通过上述几个例子,我们基本是知道怎么使用上面这个线程池了。

但是都知道Python的线程,因为GIL(全局解释器锁)的原因,是不能并发到多个物理核心上的。

所以是IO密集型的,像爬虫,读写文件,使用线程池是ok的。

但是如果说我就是野,就是头铁,非要用Python做计算型应用,像图片压缩、视频流推送,那没办法,需要使用多进程池方式。

其实通过concurrent这个接口,可以很方便的创建进程池,只需要修改两个地方。

 
 
 
 
  1. ... 
  2. # 改成导入进程池方式 
  3. from concurrent.futures import ProcessPoolExecutor 
  4. ... 
  5. if __name__ == '__main__': 
  6.     ... 
  7.     # 进程池方式 
  8.     pool = ProcessPoolExecutor(max_workers=10) 
  9.     ... 

只需要修改这俩地方即可,其他和上述用法一摸一样。

总结

本篇主要讲的是Python自带的线程池和进程池。

比较有特色的是,ThreadPoolExecutor,ProcessPoolExecutor的接口是一样的。

只需要修改导入的包就行。

concurrent的接口主要有pool.submit(),pool.shutdown(),futrue.add_done_callback()。

基本这几个都够自己用了。

本文转载自微信公众号「 Python爬虫与数据挖掘」,可以通过以下二维码关注。转载本文请联系 Python爬虫与数据挖掘公众号。

网站名称:一篇文章浅析Python自带的线程池和进程池
文章网址:http://www.gawzjz.com/qtweb/news47/190547.html

网站建设、网络推广公司-创新互联,是专注品牌与效果的网站制作,网络营销seo公司;服务项目有等

广告

声明:本网站发布的内容(图片、视频和文字)以用户投稿、用户转载内容为主,如果涉及侵权请尽快告知,我们将会在第一时间删除。文章观点不代表本网站立场,如需处理请联系客服。电话:028-86922220;邮箱:631063699@qq.com。内容未经允许不得转载,或转载时需注明来源: 创新互联