温馨提示×

c++ wait能自定义吗

c++
小樊
81
2024-11-24 04:38:41
栏目: 编程语言

在C++中,wait()函数通常与条件变量一起使用,用于让当前线程等待某个条件成立。wait()函数本身不能直接自定义,但你可以通过条件变量来实现自定义的等待逻辑。

以下是一个简单的示例,展示了如何使用条件变量和wait()函数实现自定义等待逻辑:

#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); // 当前线程被阻塞,直到 condition 变量变为 true
    }
    std::cout << "thread " << id << '\n';
}

void go() {
    std::this_thread::sleep_for(std::chrono::seconds(1));
    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;
}

在这个示例中,我们创建了一个名为ready的共享变量,用于表示条件是否满足。我们还创建了一个条件变量cv和一个互斥锁mtxprint_id函数中的线程会等待ready变量变为true,然后继续执行。go函数中的线程会将ready变量设置为true,并通过cv.notify_all()唤醒所有等待的线程。

这个示例展示了如何使用条件变量和wait()函数实现自定义等待逻辑。你可以根据自己的需求修改这个示例,以实现更复杂的同步和通信场景。

0