温馨提示×

如何在C++中使用notify通知机制

c++
小樊
82
2024-08-29 07:24:36
栏目: 编程语言

在 C++ 中,std::condition_variable 提供了一种线程间的通信机制,可以用来实现线程同步

下面是一个简单的示例,展示了如何使用 std::condition_variablestd::mutex 实现多线程之间的通知机制:

#include<iostream>
#include<thread>
#include <mutex>
#include<condition_variable>

std::mutex mtx;
std::condition_variable cv;
bool ready = false;

void print_id(int id) {
    std::unique_lock<std::mutex> lck(mtx);
    while (!ready) {  // 如果 ready 为 false, 则等待
        cv.wait(lck);  // 当前线程被阻塞, 当全局变量 ready 变成 true 之后,
    }
    // 现在线程继续执行
    std::cout << "thread " << id << '\n';
}

void go() {
    std::unique_lock<std::mutex> lck(mtx);
    ready = true;  // 修改全局变量
    cv.notify_all();  // 唤醒所有线程
}

int main() {
    std::thread threads[10];
    // spawn 10 threads:
    for (int i = 0; i < 10; ++i)
        threads[i] = std::thread(print_id, i);

    std::cout << "10 threads ready to race...\n";
    go();  // go!

    for (auto &th : threads) th.join();

    return 0;
}

在这个示例中,我们创建了 10 个线程,每个线程都会执行 print_id 函数。这些线程在开始时会获取互斥锁并检查全局变量 ready 是否为 true。如果 ready 不是 true,线程将调用 cv.wait(lck) 进入等待状态,同时释放互斥锁以便其他线程可以获取该锁。

主线程调用 go 函数,该函数获取互斥锁、设置全局变量 readytrue,然后调用 cv.notify_all() 来唤醒所有等待的线程。注意,在调用 notify_all() 之前必须拥有互斥锁,以确保唤醒的线程能够立即获取互斥锁并继续执行。

这个示例展示了如何使用条件变量和互斥锁来实现线程间的通知机制。当然,在实际应用中,你可能需要根据具体情况对代码进行调整。

0