Python使用threading和queue实现多线程时如何在queue队列所有任务已经完成时退出程序?

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官网上的这段例子

timespace

回复 1# SeriousCool
文档的例子主要是解释Queue的用法，而线程部分，为了简单则是无限循环。
如果要控制线程退出，还要修改代码，起码应该有个标志位控制，在此基础上更接近现实的用法：

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.             # 超时控制，便于及时响应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.     # 线程退出标志位
    
17.     exit_flag = threading.Event()
    
18.     exit_flag.clear()
    
19. 
    
20.     # 创建共享队列和线程池
    
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.     # 任务处理完成，通知线程退出，并join等待
    
32.     exit_flag.set()
    
33.     for t in threads:
    
34.         t.join()
    
35.         
    
36. if __name__ == '__main__':
    
37.     main()

复制代码

SeriousCool

timespace 发表于 2014-03-18 13:29
回复 1# SeriousCool
文档的例子主要是解释Queue的用法，而线程部分，为了简单则是无限循环。
如果要控制 ...

非常感谢~晚点测试下