GIL
在Python中,由于歷史原因(GIL),使得Python中多線程的效果非常不理想.GIL使得任何時刻Python只能利用一個CPU核,并且它的調(diào)度算法簡單粗暴:多線程中,讓每個線程運(yùn)行一段時間t,然后強(qiáng)行掛起該線程,繼而去運(yùn)行其他線程,如此周而復(fù)始,直到所有線程結(jié)束.
這使得無法有效利用計(jì)算機(jī)系統(tǒng)中的"局部性",頻繁的線程切換也對緩存不是很友好,造成資源的浪費(fèi).
據(jù)說Python官方曾經(jīng)實(shí)現(xiàn)了一個去除GIL的Python解釋器,但是其效果還不如有GIL的解釋器,遂放棄.后來Python官方推出了"利用多進(jìn)程替代多線程"的方案,在Python3中也有concurrent.futures這樣的包,讓我們的程序編寫可以做到"簡單和性能兼得".
多進(jìn)程/多線程+Queue
一般來說,在Python中編寫并發(fā)程序的經(jīng)驗(yàn)是:計(jì)算密集型任務(wù)使用多進(jìn)程,IO密集型任務(wù)使用多進(jìn)程或者多線程.另外,因?yàn)樯婕暗劫Y源共享,所以需要同步鎖等一系列麻煩的步驟,代碼編寫不直觀.另外一種好的思路是利用多進(jìn)程/多線程+Queue的方法,可以避免加鎖這樣麻煩低效的方式.
現(xiàn)在在Python2中利用Queue+多進(jìn)程的方法來處理一個IO密集型任務(wù).
假設(shè)現(xiàn)在需要下載多個網(wǎng)頁內(nèi)容并進(jìn)行解析,單進(jìn)程的方式效率很低,所以使用多進(jìn)程/多線程勢在必行.
我們可以先初始化一個tasks隊(duì)列,里面將要存儲的是一系列dest_url,同時開啟4個進(jìn)程向tasks中取任務(wù)然后執(zhí)行,處理結(jié)果存儲在一個results隊(duì)列中,最后對results中的結(jié)果進(jìn)行解析.最后關(guān)閉兩個隊(duì)列.
下面是一些主要的邏輯代碼.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
# -*- coding:utf-8 -*-#IO密集型任務(wù)#多個進(jìn)程同時下載多個網(wǎng)頁#利用Queue+多進(jìn)程#由于是IO密集型,所以同樣可以利用threading模塊import multiprocessingdef main(): tasks = multiprocessing.JoinableQueue() results = multiprocessing.Queue() cpu_count = multiprocessing.cpu_count() #進(jìn)程數(shù)目==CPU核數(shù)目 create_process(tasks, results, cpu_count) #主進(jìn)程馬上創(chuàng)建一系列進(jìn)程,但是由于阻塞隊(duì)列tasks開始為空,副進(jìn)程全部被阻塞 add_tasks(tasks) #開始往tasks中添加任務(wù) parse(tasks, results) #最后主進(jìn)程等待其他線程處理完成結(jié)果def create_process(tasks, results, cpu_count): for _ in range(cpu_count): p = multiprocessing.Process(target=_worker, args=(tasks, results)) #根據(jù)_worker創(chuàng)建對應(yīng)的進(jìn)程 p.daemon = True #讓所有進(jìn)程可以隨主進(jìn)程結(jié)束而結(jié)束 p.start() #啟動def _worker(tasks, results): while True: #因?yàn)榍懊嫠芯€程都設(shè)置了daemon=True,故不會無限循環(huán) try: task = tasks.get() #如果tasks中沒有任務(wù),則阻塞 result = _download(task) results.put(result) #some exceptions do not handled finally: tasks.task_done()def add_tasks(tasks): for url in get_urls(): #get_urls() return a urls_list tasks.put(url)def parse(tasks, results): try: tasks.join() except KeyboardInterrupt as err: print "Tasks has been stopped!" print err while not results.empty(): _parse(results)if __name__ == '__main__': main() |
利用Python3中的concurrent.futures包
在Python3中可以利用concurrent.futures包,編寫更加簡單易用的多線程/多進(jìn)程代碼.其使用感覺和Java的concurrent框架很相似(借鑒?)
比如下面的簡單代碼示例
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
def handler(): futures = set() with concurrent.futures.ProcessPoolExecutor(max_workers=cpu_count) as executor: for task in get_task(tasks): future = executor.submit(task) futures.add(future)def wait_for(futures): try: for future in concurrent.futures.as_completed(futures): err = futures.exception() if not err: result = future.result() else: raise err except KeyboardInterrupt as e: for future in futures: future.cancel() print "Task has been canceled!" print e return result |
總結(jié)
要是一些大型Python項(xiàng)目也這般編寫,那么效率也太低了.在Python中有許多已有的框架使用,使用它們起來更加高效.
但是自己的一些"小打小鬧"的程序這樣來編寫還是不錯的.:)
