条件变量是多线程编程中用于同步的一个工具,它允许线程挂起,直到满足某种条件。在 C++ 中,条件变量定义在 <condition_variable>
的头文件中,并配合互斥锁使用。以下是一个详细的 C++ 条件变量使用的例子:
假设有一个生产者-消费者问题,其中生产者生成数据并将其添加到共享缓冲区中,消费者从中取出数据。共享缓冲区满时,生产者应等待;共享缓冲区空时,消费者应等待。
#include <iostream>
#include <thread>
#include <mutex>
#include <condition_variable>
#include <queue>
std::mutex mtx; // 互斥锁
std::condition_variable cv; // 条件变量
std::queue<int> data_queue; // 共享数据队列
bool done = false; // 用来通知消费者生产者已经完成生产
// 生产者函数
void producer() {
for (int i = 0; i < 10; ++i) {
std::unique_lock<std::mutex> lk(mtx); // 锁定互斥量
data_queue.push(i); // 生产数据并放入队列
std::cout << "Produced: " << i << std::endl;
lk.unlock(); // 解锁互斥量
cv.notify_one(); // 通知一个等待的线程
}
// 生产完成
std::unique_lock<std::mutex> lk(mtx);
done = true; // 设置生产完成标志
lk.unlock();
cv.notify_all(); // 通知所有等待的线程
}
// 消费者函数
void consumer() {
while (true) {
std::unique_lock<std::mutex> lk(mtx); // 锁定互斥量
// 检查缓冲区是否为空或是否完成生产
cv.wait(lk, []{ return !data_queue.empty() || done; });
if (!data_queue.empty()) {
int data = data_queue.front(); // 获取数据
data_queue.pop();
lk.unlock(); // 解锁互斥量
std::cout << "Consumed: " << data << std::endl;
} else if (done) {
// 如果没有数据但已完成生产,跳出循环
lk.unlock();
break;
}
}
}
int main() {
std::thread p(producer); // 生产者线程
std::thread c(consumer); // 消费者线程
p.join(); // 等待生产者完成
c.join(); // 等待消费者完成
std::cout << "End of production and consumption." << std::endl;
return 0;
}
在这个例子中,生产者线程生成数据并将其放入共享队列中。每次放入数据后,生产者使用条件变量的 notify_one
方法唤醒一个可能在等待(即调用 wait
方法)的消费者线程。
消费者线程等待条件变量,直到队列中有数据可取。它使用了 wait
方法的重载版本,此版本接受一个谓词参数,并且只有当共享队列不为空或 done
为 true
时,才会从等待中返回。
最后,当生产者生产完数据后,它将 done
标志设置为 true
并调用 notify_all
方法,通知所有等待的消费者线程生产已完成,这样消费者线程就能退出循环。
记住在使用生产者-消费者或类似的模式时,总是使用条件变量和互斥锁来确保同步和数据的安全。
这个例子中的消费者线程能够避免虚假唤醒。虚假唤醒是指线程在没有收到实际通知的情况下被唤醒。为避免因虚假唤醒而导致的潜在问题,std::condition_variable
的 wait
函数通常与一个谓词(Predicate)一起使用。
这个谓词是一个返回布尔值的函数或者 lambda 表达式,作为 wait
函数的第二个参数。仅当谓词的结果为 true
时,等待操作才会完成。这样就确保了即使是由于虚假唤醒,线程醒来后也会检查条件是否满足,如果不满足,它会再次进入等待状态。
在我们之前的例子中,消费者线程使用 wait
函数时传入了一个谓词:
cv.wait(lk, []{ return !data_queue.empty() || done; });
这个谓词检查队列是否为空或生产是否已完成。只有在队列不为空或生产完成标志 done
设置为 true
时,谓词才返回 true
。这样即便是虚假唤醒发生,消费者线程在没有数据可用时也不会处理队列,而是再次进入等待状态,直到条件真正满足。
因此,通过正确使用带谓词的 wait
函数可以确保线程在满足特定条件时才能继续运行,避免了虚假唤醒的问题。