C++标准库对条件变量有两套实现:std::condition_variable和std::condition_variable_any,其中std::condition_variable_any的通用性更强,但是占用系统资源更多。std::condition_variable需要搭配std::unique_lock使用。
首先介绍如何使用std::condition_variable:
案例1:条件变量condition_variable的使用
#include <mutex> //std::mutex std::lock_guard
#include <condition_variable> //std::condition_variable
std::mutex mut; //互斥量
std::queue<data_chunk> data_queue; //队列
std::conditon_variable data_cond; //条件变量
void data_prparation_thread()
{
while(more_data_to_prepare())
{
data_chunk const data = prepare_data(); //获取数据
std::lock_guard<std::mutex> lk(mut); //锁管理器
data_queue.push(data); //数据写队列
data_cond.notify_one(); //通知等待线程
}
}
void data_processing_thread()
{
while(true)
{
std::unique_lock<std::mutex> lk(mut);
data_cond.wait(lk, []{return !data_queue.empty();});//条件变量会首先判断队列是否为空,若为空则进行阻塞,等待通知。得到通知后,会再次判断队列是否为空,若不为空才会获取锁才会接触阻塞,获取锁并进行后续程序。
data_chunk data = data_queue.front();
data_queue.pop();
lk.unlock();
process(data);
if (is_last_chunk(data))
break;
}
}
上述代码有两部分内容:一是数据准备线程,每次运行都会往队列里面添加元素并且通知数据处理线程;另一个是数据处理线程,在数据处理完之前一直循环运行,获取数据-等待通知-获取数据,按照这样的规律进行。
案列2:
#include <mutex> //std::mutex std::lock_guard
#include <condition_variable> //std::condition_variable
bool ready = false;
std::mutex mut; //互斥量
std::conditon_variable data_cond; //条件变量
void process_thread1() //等待线程1
{
std::unique_lock<std::mutex> lk(mut);
data_cond.wait(lk, ready);
lk.unlock();
process1();
}
void process_thread2() //等待线程2
{
std::unique_lock<std::mutex> lk(mut);
data.cond.wait(lk, ready);
lk.unlock();
process2();
}
void data_init_thread() //通知线程
{
std::lock_guard<std::mutex> lk(mut);
ready = true;
data_cond.notify_all();
}
上述代码中共有三个线程,等待线程1、等待线程2和等待线程3。等待线程1和等待线程2会阻塞,等待通知线程的通知才会继续运行。
案例三:有超时功能的条件变量
#include <mutex>
#include <condition_variable>
#include <chrono>
std::condition_variable cv;
bool done;
std::mutex m;
bool wait_loop()
{
auto const timeout = std::chrono::steady_clock::now() + std::chrono::milliseconds(500); //设置超时时间点
std::unique_lock<std::mutex> lk(m);
while(!done)
{
if(cv.wait_until(lk, timeout)==std::cv_status::timeout)//当超过某个时间点timeout时还无法获得锁,则会返回状态值std::cv_status::timeout。还有一种std::condition_variable::wait_for(lk, duar)则是超过了时间段duar还无法获得锁就会返回状态值
break;
}
return done;
}
条件变量实例cv会一直等待互斥量,直到超时