C++条件变量避免虚假唤醒的例子

最新推荐文章于 2025-05-26 11:02:46 发布

happyblreay

最新推荐文章于 2025-05-26 11:02:46 发布

阅读量502

点赞数 7

CC 4.0 BY-SA版权

分类专栏： C++ 文章标签： c++ 开发语言

本文链接：https://blog.csdn.net/happyblreay/article/details/138538526

C++ 专栏收录该内容

8 篇文章

订阅专栏

条件变量是多线程编程中用于同步的一个工具，它允许线程挂起，直到满足某种条件。在 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 函数可以确保线程在满足特定条件时才能继续运行，避免了虚假唤醒的问题。