各类资料学习下载合集
链接:https://pan.quark.cn/s/7c8c391011eb
在上一篇博客中,我们已经初步实现了基于条件变量的生产者-消费者模型。然而,当涉及到多消费者场景时,我们对代码的严谨性提出了更高的要求。本文将详细讲解pthread_cond_wait的正确使用姿势,特别是为什么它必须与while循环结合,以及pthread_cond_signal可能带来的“意外”行为。
一、 回顾生产者-消费者模型核心流程
我们再次梳理一下生产者和消费者使用条件变量进行协作的基本流程。
1. 生产者流程
- 生成数据:创建新的产品(例如链表节点)。
- 加锁:
pthread_mutex_lock(&mutex),保护公共区。 - 放入公共区:将新产品添加到链表(如头插法)。
- 解锁:
pthread_mutex_unlock(&mutex)。 - 通知消费者:
pthread_cond_signal(&cond),告知有新数据了。 - 循环生产:持续进行。
2. 消费者流程
- 加锁:
pthread_mutex_lock(&mutex),准备检查公共区。 - 条件等待:
pthread_cond_wait(&cond, &mutex)。这是一个“三合一”操作:- 阻塞等待:如果条件不满足(如公共区为空),线程进入阻塞状态。
- 自动解锁:在阻塞的同时,原子性地释放互斥锁,让生产者有机会进入临界区。
- 重新加锁:被生产者唤醒后,自动重新获取互斥锁,然后
wait函数返回。
- 数据消费:从公共区取出数据。
- 解锁:
pthread_mutex_unlock(&mutex)。 - 循环消费:持续进行。
二、 消费者代码中的关键修正:从if到while
现在,我们来重点关注消费者线程中的一个关键点:条件判断。在单消费者场景下,我们可能习惯用if (head == NULL)来判断是否需要等待。但在多消费者场景下,这将是灾难性的!
1. 错误示例:使用if判断
假设我们有多个消费者,代码片段如下:
// 消费者线程函数 (错误示例 - 使用 if)void*consumer_bad(void*arg){while(1){pthread_mutex_lock(&mutex);// 错误!这里使用 if 判断if(head==NULL){pthread_cond_wait(&has_data,&mutex);}// ... (取数据、解锁、消费)// 假设这里会取走 head 指向的数据// ...pthread_mutex_unlock(&mutex);// ...}returnNULL;}问题分析:
- 多个消费者阻塞:当
head == NULL时,所有消费者都会进入if语句块,并调用pthread_cond_wait阻塞。 - 生产者
signal:生产者生产一个数据,并调用pthread_cond_signal(&has_data)。 signal的“意外”行为:尽管官方文档说pthread_cond_signal唤醒“至少一个”线程,但在许多 Linux/POSIX 实现中,它实际上可能唤醒所有等待在该条件变量上的线程(或者唤醒多个,数量不确定)。- 竞争条件:假设
signal唤醒了三个消费者 A、B、C。- 它们会竞争
mutex。假设 A 抢到了锁,它会跳过if语句(因为wait返回时head可能不再为NULL),取出数据并消费。 - A 释放锁后,B 抢到锁。此时
head再次变为NULL(已经被 A 取走了)。但 B 之前已经通过了if (head == NULL)的判断,它不会再次检查条件,而是直接尝试去取数据。 - 结果:B 会尝试访问一个空的
head指针,导致程序崩溃或数据损坏!
- 它们会竞争
2. 正确姿势:使用while循环判断
为了彻底避免上述问题,我们必须将条件判断放在while循环中:
// 消费者线程函数 (正确示例 - 使用 while)void*consumer(void