這篇文章主要介紹了Python多線程和隊列操作實例,本文直接給給實例代碼,需要的朋友可以參考下
Python3,開一個線程,間隔1秒把一個遞增的數字寫入隊列,再開一個線程,從隊列中取出數字並打印到終端
代碼如下:
#! /usr/bin/env python3
import time
import threading
import queue
# 一個線程,間隔一定的時間,把一個遞增的數字寫入隊列
# 生產者
class Producer(threading.Thread):
def __init__(self, work_queue):
super().__init__() # 必須調用
self.work_queue = work_queue
def run(self):
num = 1
while True:
self.work_queue.put(num)
num = num+1
time.sleep(1) # 暫停1秒
# 一個線程,從隊列取出數字,並顯示到終端
class Printer(threading.Thread):
def __init__(self, work_queue):
super().__init__() # 必須調用
self.work_queue = work_queue
def run(self):
while True:
num = self.work_queue.get() # 當隊列為空時,會阻塞,直到有數據
print(num)
def main():
work_queue = queue.Queue()
producer = Producer(work_queue)
producer.daemon = True # 當主線程退出時子線程也退出
producer.start()
printer = Printer(work_queue)
printer.daemon = True # 當主線程退出時子線程也退出
printer.start()
work_queue.join() # 主線程會停在這裡,直到所有數字被get(),並且task_done(),因為沒有調用task_done(),所在這裡會一直阻塞,直到用戶按^C
if __name__ == '__main__':
main()
queue是線程安全的,從多個線程訪問時無需加鎖。
如果在work_queue.get()之後調用work_queue.task_done(),那麼在隊列空時work_queue.join()會返回。
這裡work_queue.put()是間隔一定時間才往隊列放東西,如果調用work_queue.task_done(),在數字1被get()後,隊列空時,join()就返回,程序就結束了。
也就是程序只打印了1然後就退出了。
所以在這種使用情景下,不能調用task_done(),程序會一直循環下去。