在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
和一个互斥锁mtx
。print_id
函数中的线程会等待ready
变量变为true
,然后继续执行。go
函数中的线程会将ready
变量设置为true
,并通过cv.notify_all()
唤醒所有等待的线程。
这个示例展示了如何使用条件变量和wait()
函数实现自定义等待逻辑。你可以根据自己的需求修改这个示例,以实现更复杂的同步和通信场景。