Python3中多線程編程的隊列運作示例

更新時間：2015年04月16日 10:56:02 投稿：goldensun

這篇文章主要介紹了Python3中多線程編程的隊列運作示例,文中用一個簡單的例子展示了Python下嘗試多線程時隊列的進站出站是如何運作的,需要的朋友可以參考下

Python3，開一個線程，間隔1秒把一個遞增的數(shù)字寫入隊列，再開一個線程，從隊列中取出數(shù)字并打印到終端

#! /usr/bin/env python3

import time
import threading
import queue

# 一個線程，間隔一定的時間，把一個遞增的數(shù)字寫入隊列
# 生產(chǎn)者
class Producer(threading.Thread):

  def __init__(self, work_queue):
    super().__init__() # 必須調(diào)用
    self.work_queue = work_queue
    
  def run(self):
    num = 1
    while True:
      self.work_queue.put(num)
      num = num+1
      time.sleep(1) # 暫停1秒

# 一個線程，從隊列取出數(shù)字，并顯示到終端
class Printer(threading.Thread):

  def __init__(self, work_queue):
    super().__init__() # 必須調(diào)用
    self.work_queue = work_queue

  def run(self):
    while True:
      num = self.work_queue.get() # 當隊列為空時，會阻塞，直到有數(shù)據(jù)
      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() # 主線程會停在這里，直到所有數(shù)字被get()，并且task_done()，因為沒有調(diào)用task_done()，所在這里會一直阻塞，直到用戶按^C

if __name__ == '__main__':
  main()

queue是線程安全的，從多個線程訪問時無需加鎖。
如果在work_queue.get()之后調(diào)用work_queue.task_done()，那么在隊列空時work_queue.join()會返回。
這里work_queue.put()是間隔一定時間才往隊列放東西，如果調(diào)用work_queue.task_done()，在數(shù)字1被get()后，隊列空時，join()就返回，程序就結(jié)束了。
也就是程序只打印了1然后就退出了。
所以在這種使用情景下，不能調(diào)用task_done()，程序會一直循環(huán)下去。

您可能感興趣的文章: