天天看點

python筆記11-多線程之Condition(條件變量)

前言

當小夥伴a在往火鍋裡面添加魚丸,這個就是生産者行為;另外一個小夥伴b在吃掉魚丸就是消費者行為。當火鍋裡面魚丸達到一定數量加滿後b才能吃,這就是一種條件判斷了。

這就是本篇要講的Condition(條件變量)

Condition

Condition(條件變量)通常與一個鎖關聯。需要在多個Contidion中共享一個鎖時,可以傳遞一個Lock/RLock執行個體給構造方法,否則它将自己生成一個RLock執行個體。

可以認為,除了Lock帶有的鎖定池外,Condition還包含一個等待池,池中的線程處于狀态圖中的等待阻塞狀态,直到另一個線程調用notify()/notifyAll()通知;得到通知後線程進入鎖定池等待鎖定。

Condition():

  • acquire(): 線程鎖
  • release(): 釋放鎖
  • wait(timeout): 線程挂起,直到收到一個notify通知或者逾時(可選的,浮點數,機關是秒s)才會被喚醒繼續運作。wait()必須在已獲得Lock前提下才能調用,否則會觸發RuntimeError。
  • notify(n=1): 通知其他線程,那些挂起的線程接到這個通知之後會開始運作,預設是通知一個正等待該condition的線程,最多則喚醒n個等待的線程。notify()必須在已獲得Lock前提下才能調用,否則會觸發RuntimeError。notify()不會主動釋放Lock。
  • notifyAll(): 如果wait狀态線程比較多,notifyAll的作用就是通知所有線程

生産者與消費者

# coding=utf-8
import threading
import time

con = threading.Condition()

num = 0

# 生産者
class Producer(threading.Thread):

    def __init__(self):
        threading.Thread.__init__(self)

    def run(self):
        # 鎖定線程
        global num
        con.acquire()
        while True:
            print "開始添加!!!"
            num += 1
            print "火鍋裡面魚丸個數:%s" % str(num)
            time.sleep(1)
            if num >= 5:
                print "火鍋裡面裡面魚丸數量已經到達5個,無法添加了!"
                # 喚醒等待的線程
                con.notify()  # 喚醒小夥伴開吃啦
                # 等待通知
                con.wait()
        # 釋放鎖
        con.release()

# 消費者
class Consumers(threading.Thread):
    def __init__(self):
        threading.Thread.__init__(self)

    def run(self):
        con.acquire()
        global num
        while True:
            print "開始吃啦!!!"
            num -= 1
            print "火鍋裡面剩餘魚丸數量:%s" %str(num)
            time.sleep(2)
            if num <= 0:
                print "鍋底沒貨了,趕緊加魚丸吧!"
                con.notify()  # 喚醒其它線程
                # 等待通知
                con.wait()
        con.release()

p = Producer()
c = Consumers()
p.start()
c.start()