快捷導(dǎo)航

Python3中多線程編程的隊(duì)列運(yùn)作示例

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

這篇文章主要介紹了Python3中多線程編程的隊(duì)列運(yùn)作示例,文中用一個(gè)簡(jiǎn)單的例子展示了Python下嘗試多線程時(shí)隊(duì)列的進(jìn)站出站是如何運(yùn)作的,需要的朋友可以參考下

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

#! /usr/bin/env python3

import time
import threading
import queue

# 一個(gè)線程，間隔一定的時(shí)間，把一個(gè)遞增的數(shù)字寫(xiě)入隊(duì)列
# 生產(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秒

# 一個(gè)線程，從隊(duì)列取出數(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() # 當(dāng)隊(duì)列為空時(shí)，會(huì)阻塞，直到有數(shù)據(jù)
      print(num)

def main():
  work_queue = queue.Queue()

  producer = Producer(work_queue)
  producer.daemon = True # 當(dāng)主線程退出時(shí)子線程也退出
  producer.start()

  printer = Printer(work_queue)
  printer.daemon = True # 當(dāng)主線程退出時(shí)子線程也退出
  printer.start()

  work_queue.join() # 主線程會(huì)停在這里，直到所有數(shù)字被get()，并且task_done()，因?yàn)闆](méi)有調(diào)用task_done()，所在這里會(huì)一直阻塞，直到用戶按^C

if __name__ == '__main__':
  main()

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

您可能感興趣的文章: