条件变量用于协调多线程执行,解决互斥锁无法处理的等待与通知问题。它结合锁和等待队列,支持线程在条件不满足时挂起并由其他线程唤醒,适用于生产者-消费者等场景。通过 threading.Condition 实现,推荐使用 with 语句管理锁,调用 wait() 前需持有锁,且应使用 while 循环检查条件以避免虚假唤醒,确保线程安全与正确同步。
在Python多线程编程中,当多个线程需要协调执行顺序或等待特定状态时,仅靠互斥锁(Lock)往往不够。这时就需要使用条件变量(Condition)。它允许线程在某个条件不满足时挂起自己,并在其他线程改变状态后被唤醒。
条件变量本质上是“锁 + 等待队列”的组合。它内部持有一个锁(默认为RLock),并维护一个等待该条件的线程队列。典型应用场景包括生产者-消费者模型、任务等待结果等。
Python 的 threading.Condition 类提供了对条件变量
的支持。常用方法有:
下面是一个简单的生产者-消费者示例:
import threading import time import random共享资源
items = [] condition = threading.Condition()
def producer(): global items for i in range(5): condition.acquire() item = random.randint(1, 100) items.append(item) print(f"生产者: 添加 {item}") condition.notify() # 唤醒一个消费者 condition.release() time.sleep(1)
def consumer(): global items while True: condition.acquire() if not items: print("消费者: 暂无数据,等待...") condition.wait() # 释放锁并等待 item = items.pop(0) print(f"消费者: 取出 {item}") condition.release() time.sleep(2)
启动线程
t1 = threading.Thread(target=producer) t2 = threading.Thread(target=consumer)
t1.start() t2.start()
t1.join() t2.join()
为了避免忘记释放锁,推荐使用 with 语句操作条件变量。它会自动处理 acquire 和 release:
def consumer_with():
global items
while True:
with condition:
if not items:
print("等待新数据...")
condition.wait()
item = items.pop(0)
print(f"消费: {item}")
time.sleep(2) # 不在锁内休眠,提高并发性
注意:wait() 调用前必须先持有锁,而 wait 执行时会自动释放锁,被唤醒后重新获取锁再返回。
使用条件变量时容易犯几个错误:
正确做法是:
with condition:
while not some_condition:
condition.wait()
# 执行操作
do_something()
循环检查可以防止虚假唤醒问题,确保条件真正满足后再继续。
基本上就这些。条件变量是实现复杂线程同步的关键工具,掌握它有助于构建更高效的并发程序。