C++ condition_variable 学习笔记
基本概念
std::condition_variable
是 C++ 标准库中用于多线程同步的重要工具之一。它可以与 std::mutex
一起使用,实现线程之间的等待和唤醒机制。通过 std::condition_variable
,一个线程可以等待另一个线程满足某个特定条件,而不需要使用轮询等待的方式,这样有助于节省 CPU 资源,提高程序的性能。
内部原理
std::condition_variable
的内部原理通常是基于操作系统提供的条件变量或信号量实现的。在大多数情况下,它使用了操作系统底层的原子操作和同步机制,以确保线程的安全等待和唤醒。具体的实现细节可能会因操作系统和编译器而异。
成员函数
std::condition_variable
类提供了几个重要的成员函数,用于线程的等待和唤醒:
名称 | 描述 |
---|---|
notify_one() | 唤醒一个正在等待的线程。 |
notify_all() | 唤醒所有正在等待的线程。 |
wait(lock) | 在持有 lock 的情况下等待,直到被唤醒。 |
wait_for(lock, timeout) | 在持有 lock 的情况下等待一段时间,直到被唤醒或超时。 |
wait_until(lock, time_point) | 在持有 lock 的情况下等待到指定时间点,直到被唤醒或超时。 |
使用技巧
避免虚假唤醒
在使用 std::condition_variable
进行等待时,存在虚假唤醒的可能性,即即使没有调用 notify_one()
或 notify_all()
,线程也有可能被唤醒。为了避免这种情况,建议在等待条件时使用 predicate
进行判断。predicate
是一个函数或函数对象,用于检查条件是否满足。例如:
std::condition_variable cv;
std::mutex mtx;
bool data_ready = false;
void consumer() {
std::unique_lock<std::mutex> lock(mtx);
cv.wait(lock, []{ return data_ready; }); // 使用 predicate 避免虚假唤醒
// 执行消费操作
}
精确的通知
在进行通知时,确保在释放锁之前完成通知操作,以避免通知操作在条件变量被销毁之后导致程序崩溃的情况。例如:
std::condition_variable cv;
std::mutex mtx;
bool data_ready = false;
void producer() {
{
std::lock_guard<std::mutex> lock(mtx);
data_ready = true;
}
cv.notify_one(); // 确保在释放锁之前完成通知操作
}
实战案例
场景一:多线程等待数据准备就绪
#include <iostream>
#include <thread>
#include <mutex>
#include <condition_variable>
std::mutex mtx;
std::condition_variable cv;
bool data_ready = false;
void worker(int id) {
std::unique_lock<std::mutex> lock(mtx);
cv.wait(lock, []{ return data_ready; }); // 等待数据准备就绪
std::cout << "Worker thread " << id << ": Data is ready." << std::endl;
}
void prepare_data() {
std::this_thread::sleep_for(std::chrono::seconds(2)); // 模拟生产数据的耗时操作
{
std::lock_guard<std::mutex> lock(mtx);
data_ready = true;
}
cv.notify_all(); // 通知所有等待的线程数据已经准备就绪
}
int main() {
const int num_threads = 5;
std::thread workers[num_threads];
for (int i = 0
; i < num_threads; ++i) {
workers[i] = std::thread(worker, i);
}
std::thread data_preparer(prepare_data);
for (int i = 0; i < num_threads; ++i) {
workers[i].join();
}
data_preparer.join();
return 0;
}
在这个示例中,有 5 个工作线程等待数据准备就绪。一旦数据准备就绪,调用 notify_all()
唤醒所有等待的线程。
注意事项
- 虚假唤醒:
std::condition_variable
中的等待方法可能会出现虚假唤醒,即即使没有调用notify_one()
或notify_all()
,线程也有可能被唤醒。因此,建议在等待条件时使用predicate
进行判断,避免虚假唤醒带来的问题。
总结
std::condition_variable
是 C++ 并发编程中重要的同步原语之一,用于线程间的通信和同步。通过合理地使用 std::condition_variable
,可以有效地实现多线程之间的协调和同步,提高程序的并发性能和可靠性。