一、前言
在thread和process中,应当优选process,因为process更稳定,而且,process可以分布到多台机器上,而thread最多只能分布到同一台机器的多个cpu上。
python的multiprocessing模块不但支持多进程,其中managers子模块还支持把多进程分布到多台机器上。可以写一个服务进程作为调度者,将任务分布到其他多个进程中,依靠网络通信进行管理。
二、案例分析
在做爬虫程序时,抓取某个网站的所有图片,如果使用多进程的话,一般是一个进程负责抓取图片的链接地址,将链接地址放到queue中,另外的进程负责 从queue中取链接地址进行下载和存储到本地。
怎么用分布式进程实现?
一台机器上的进程负责抓取链接地址,其他机器上的进程负责系在存储。那么遇到的主要问题是将queue 暴露到网络中,让其他机器进程都可以访问,分布式进程就是将这个过程进行了封装,可以将这个过程称为本地队列的网络化。
例:
1.py
from multiprocessing.managers import basemanagerfrom multiprocessing import freeze_support, queue# 任务个数task_number = 10# 收发队列task_quue = queue(task_number)result_queue = queue(task_number)def get_task(): return task_quuedef get_result(): return result_queue# 创建类似的queuemanagerclass queuemanager(basemanager): passdef win_run(): # 注册在网络上,callable 关联了queue 对象 # 将queue对象在网络中暴露 # window下绑定调用接口不能直接使用lambda,所以只能先定义函数再绑定 queuemanager.register('get_task_queue', callable=get_task) queuemanager.register('get_result_queue', callable=get_result) # 绑定端口和设置验证口令 manager = queuemanager(address=('127.0.0.1', 8001), authkey='qiye'.encode()) # 启动管理,监听信息通道 manager.start() try: # 通过网络获取任务队列和结果队列 task = manager.get_task_queue() result = manager.get_result_queue() # 添加任务 for url in ["imageurl_" + str(i) for i in range(10)]: print('url is %s' % url) task.put(url) print('try get result') for i in range(10): print('result is %s' % result.get(timeout=10)) except: print('manager error') finally: manager.shutdown()if __name__ == '__main__': freeze_support() win_run()
连接服务器,端口和验证口令注意保持与服务器进程中完全一致从网络获取queue,进行本地化,从task队列获取任务,并且把结果写入result队列
2.py
#coding:utf-8import timefrom multiprocessing.managers import basemanager# 创建类似的manager:class manager(basemanager): pass#使用queuemanager注册获取queue的方法名称manager.register('get_task_queue')manager.register('get_result_queue')#连接到服务器:server_addr = '127.0.0.1'print('connect to server %s...' % server_addr)# 端口和验证口令注意保持与服务进程设置的完全一致:m = manager(address=(server_addr, 8001), authkey='qiye')# 从网络连接:m.connect()#获取queue的对象:task = m.get_task_queue()result = m.get_result_queue()#从task队列取任务,并把结果写入result队列:while(not task.empty()): image_url = task.get(true,timeout=5) print('run task download %s...' % image_url) time.sleep(1) result.put('%s--->success'%image_url)#结束:print('worker exit.')
任务进程要通过网络连接到服务进程,所以要指定服务进程的ip。
运行结果如下:
获取图片地址,将地址传到2.py。
接收1.py传递的地址,进行图片的下载,控制台显示爬取结果。
三、总结
本文基于python基础,python的分布式进程接口简单,封装良好,适合需要把繁重任务分布到多台机器的环境下。通过讲解queue的作用是用来传递任务和接收结果。
以上就是一篇文章带你了解python的分布式进程接口的详细内容。