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

条件变量是多线程编程中用于同步的一个工具,它允许线程挂起,直到满足某种条件。在 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 方法的重载版本,此版本接受一个谓词参数,并且只有当共享队列不为空或 donetrue 时,才会从等待中返回。

最后,当生产者生产完数据后,它将 done 标志设置为 true 并调用 notify_all 方法,通知所有等待的消费者线程生产已完成,这样消费者线程就能退出循环。

记住在使用生产者-消费者或类似的模式时,总是使用条件变量和互斥锁来确保同步和数据的安全。

这个例子中的消费者线程能够避免虚假唤醒。虚假唤醒是指线程在没有收到实际通知的情况下被唤醒。为避免因虚假唤醒而导致的潜在问题,std::condition_variablewait 函数通常与一个谓词(Predicate)一起使用。

这个谓词是一个返回布尔值的函数或者 lambda 表达式,作为 wait 函数的第二个参数。仅当谓词的结果为 true 时,等待操作才会完成。这样就确保了即使是由于虚假唤醒,线程醒来后也会检查条件是否满足,如果不满足,它会再次进入等待状态。

在我们之前的例子中,消费者线程使用 wait 函数时传入了一个谓词:

cv.wait(lk, []{ return !data_queue.empty() || done; });

这个谓词检查队列是否为空或生产是否已完成。只有在队列不为空或生产完成标志 done 设置为 true 时,谓词才返回 true。这样即便是虚假唤醒发生,消费者线程在没有数据可用时也不会处理队列,而是再次进入等待状态,直到条件真正满足。

因此,通过正确使用带谓词的 wait 函数可以确保线程在满足特定条件时才能继续运行,避免了虚假唤醒的问题。

  • 7
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值