Python使用threading和queue實(shí)現(xiàn)多線程時(shí)如何在queue隊(duì)列所有任務(wù)已經(jīng)完成時(shí)退出程序?

SeriousCool

def worker():
    while True:
        item = q.get()
        do_work(item)
        q.task_done()

q = Queue()
for i in range(num_worker_threads):
     t = Thread(target=worker)
     t.daemon = True
     t.start()

for item in source():
    q.put(item)

q.join()

比如python官網(wǎng)上的這段例子

timespace

回復(fù) 1# SeriousCool
文檔的例子主要是解釋Queue的用法，而線程部分，為了簡(jiǎn)單則是無(wú)限循環(huán)。
如果要控制線程退出，還要修改代碼，起碼應(yīng)該有個(gè)標(biāo)志位控制，在此基礎(chǔ)上更接近現(xiàn)實(shí)的用法：

1. import queue
   
2. import sys
   
3. import threading
   
4. 
   
5. def worker(name, q, exit_flag):
   
6.     while not exit_flag.is_set():
   
7.         try:
   
8.             # 超時(shí)控制，便于及時(shí)響應(yīng)exit_flag
   
9.             item = q.get(timeout=3)
   
10.         except queue.Empty:
    
11.             continue
    
12.         print('thread {}, item {}'.format(name, item), file=sys.stderr)
    
13.         q.task_done()
    
14. 
    
15. def main():
    
16.     # 線程退出標(biāo)志位
    
17.     exit_flag = threading.Event()
    
18.     exit_flag.clear()
    
19. 
    
20.     # 創(chuàng)建共享隊(duì)列和線程池
    
21.     q = queue.Queue()
    
22.     num_of_threads = 5
    
23.     threads = [threading.Thread(target=worker, args=(str(i+1), q, exit_flag))
    
24.                for i in range(num_of_threads)]
    
25.     for t in threads:
    
26.         t.start()
    
27.     for i in range(50):
    
28.         q.put(i)
    
29.     q.join()
    
30. 
    
31.     # 任務(wù)處理完成，通知線程退出，并join等待
    
32.     exit_flag.set()
    
33.     for t in threads:
    
34.         t.join()
    
35.         
    
36. if __name__ == '__main__':
    
37.     main()

復(fù)制代碼

SeriousCool

timespace 發(fā)表于 2014-03-18 13:29
回復(fù) 1# SeriousCool
文檔的例子主要是解釋Queue的用法，而線程部分，為了簡(jiǎn)單則是無(wú)限循環(huán)。
如果要控制 ...

非常感謝~晚點(diǎn)測(cè)試下