C++并发编程 等待与唤醒
C++并发编程 等待与唤醒
条件变量
条件变量, 包括(std::condition_variable 和 std::condition_variable_any)
定义在 condition_variable 头文件中, 它们都需要与互斥量(作为同步工具)一起才能工作.
std::condition_variable 允许阻塞一个线程, 直到条件达成.
成员函数
void wait(std::unique_lock<std::mutex>& lock);
等待, 通过 notify_one(), notify_all()或伪唤醒结束等待
void wait(std::unique_lock<std::mutex>& lock, Predicate pred);
等待, 通过 notify_one(), notify_all()被调用, 并且谓词为 true 时结束等待.
pred 谓词必须是合法的, 并且需要返回一个值, 这个值可以和bool互相转化
cv_status wait_until(std::unique_lock<std::mutex>& lock, const std::chrono::time_point<Clock, Duration>& absolute_time);
调用 notify_one(), notify_all(), 超时或线程伪唤醒时, 结束等待.
返回值标识了是否超时.
bool wait_until(std::unique_lock<std::mutex>& lock, const std::chrono::time_point<Clock, Duration>& absolute_time, Predicate pred);
等待, 通过 notify_one(), notify_all(), 超时, 线程伪唤醒, 并且谓词为 true 时结束等待.
cv_status wait_for(std::unique_lock<std::mutex>& lock, const std::chrono::duration<Rep, Period>& relative_time);
调用 notify_one(), notify_all(), 指定时间内达成条件或线程伪唤醒时,结束等待
bool wait_for(std::unique_lock<std::mutex>& lock, const std::chrono::duration<Rep, Period>& relative_time, Predicate pred);
调用 notify_one(), notify_all(), 指定时间内达成条件或线程伪唤醒时,并且谓词为 true 时结束等待.
void notify_one() noexcept; 唤醒一个等待当前 std::condition_variable 实例的线程
void notify_all() noexcept; 唤醒所有等待当前 std::condition_variable 实例的线程
一个线程安全的队列设计:
#ifndef _THREAD_SAFE_QUEUE_ #define _THREAD_SAFE_QUEUE_ #include <condition_variable> #include <mutex> #include <queue> #include <memory> template<typename Ty, typename ConditionVar = std::condition_variable, typename Mutex = std::mutex> class ThreadSafeQueue { typedef std::queue<Ty> Queue; typedef std::shared_ptr<Ty> SharedPtr; typedef std::lock_guard<Mutex> MutexLockGuard; typedef std::unique_lock<Mutex> MutexUniqueLock; public: explicit ThreadSafeQueue() {} ~ThreadSafeQueue() {} ThreadSafeQueue(const ThreadSafeQueue&) = delete; ThreadSafeQueue& operator=(const ThreadSafeQueue&) = delete; bool IsEmpty() const { MutexLockGuard lk(mMutex); return mQueue.empty(); } void WaitAndPop(Ty& value) { MutexUniqueLock lk(mMutex); mConVar.wait(lk, [this] { return !mQueue.empty(); }); value = mQueue.front(); mQueue.pop(); } SharedPtr WaitAndPop() { MutexUniqueLock lk(mMutex); mConVar.wait(lk, [this] { return !mQueue.empty(); }); SharedPtr sp = std::make_shared<Ty>(mQueue.front()); mQueue.pop(); return sp; } bool TryPop(Ty& value) { MutexLockGuard lk(mMutex); if (mQueue.empty()) return false; value = mQueue.front(); mQueue.pop(); return true; } SharedPtr TryPop() { MutexLockGuard lk(mMutex); if (mQueue.empty()) return false; SharedPtr sp = std::make_shared<Ty>(mQueue.front()); mQueue.pop(); return sp; } void Push(const Ty& value) { MutexLockGuard lk(mMutex); mQueue.push(value); mConVar.notify_all(); } private: mutable Mutex mMutex; ConditionVar mConVar; Queue mQueue; }; #endif // _THREAD_SAFE_QUEUE_
另一个版本, 使用 shared_ptr 作为成员对队列的性能有很大的提升, 其在push时减少了互斥量持有的时间, 允许其它线程在分配内存的同时,对队列进行其它操作.
template<typename Ty, typename ConditionVar = std::condition_variable, typename Mutex = std::mutex> class ThreadSafeQueue { typedef std::shared_ptr<Ty> SharedPtr; typedef std::queue<SharedPtr> Queue; typedef std::shared_ptr<Ty> SharedPtr; typedef std::lock_guard<Mutex> MutexLockGuard; typedef std::unique_lock<Mutex> MutexUniqueLock; public: explicit ThreadSafeQueue() {} ~ThreadSafeQueue() {} ThreadSafeQueue(const ThreadSafeQueue&) = delete; ThreadSafeQueue& operator=(const ThreadSafeQueue&) = delete; bool IsEmpty() const { MutexLockGuard lk(mMutex); return mQueue.empty(); } void WaitAndPop(Ty& value) { MutexUniqueLock lk(mMutex); mConVar.wait(lk, [this] { return !mQueue.empty(); }); value = std::move(*mQueue.front()); mQueue.pop(); } SharedPtr WaitAndPop() { MutexUniqueLock lk(mMutex); mConVar.wait(lk, [this] { return !mQueue.empty(); }); SharedPtr sp = mQueue.front(); mQueue.pop(); return sp; } bool TryPop(Ty& value) { MutexLockGuard lk(mMutex); if (mQueue.empty()) return false; value = std::move(*mQueue.front()); mQueue.pop(); return true; } SharedPtr TryPop() { MutexLockGuard lk(mMutex); if (mQueue.empty()) return false; SharedPtr sp = mQueue.front(); mQueue.pop(); return sp; } void Push(const Ty& value) { SharedPtr p = std::make_shared<Ty>(value); MutexLockGuard lk(mMutex); mQueue.push(p); mConVar.notify_all(); } private: mutable Mutex mMutex; ConditionVar mConVar; Queue mQueue; };
std::future
期望(std::future)可以用来等待其他线程上的异步结果, 其实例可以在任意时间引用异步结果.
C++包括两种期望, std::future(唯一期望) 和 std::shared_future(共享期望)
std::future 的实例只能与一个指定事件相关联.
std::shared_future 的实例能关联多个事件, 它们同时变为就绪状态, 并且可以访问与事件相关的任何数据.
在与数据无关的地方,可以使用 std::future<void> 与 std::shared_future<void> 的特化模板.
期望对象本身并不提供同步访问, 如果多个线程要访问一个独立的期望对象, 需要使用互斥体进行保护.
std::packaged_task
std::packaged_task 可包装一个函数或可调用的对象, 并且允许异步获取该可调用对象产生的结果, 返回值通过 get_future 返回的 std::future 对象取得, 其返回的 std::future 的模板类型为 std::packaged_task 模板函数签名中的返回值类型.
std::packaged_task 对象被调用时, 就会调用相应的函数或可调用对象, 将期望置为就绪, 并存储返回值.
std::packaged_task 的模板参数是一个函数签名, 如 int(std::string&, double*), 构造 std::packaged_task 实例时必须传入一个可以匹配的函数或可调用对象, 也可以是隐藏转换能匹配的.
std::packaged_task<std::string(const std::string&)> task([](std::string str) { std::stringstream stm; stm << "tid:" << std::this_thread::get_id() << ", str:" << str << std::endl; std::cout << stm.str(); std::this_thread::sleep_for(std::chrono::seconds(1)); return std::string("MSG:Hello"); }); std::future<std::string> f = task.get_future(); std::thread t(std::move(task), std::string("package task test")); t.detach(); // 调用 f.get 返回结果, 但是须阻塞等到任务执行完成 std::cout << "main tid:" << std::this_thread::get_id() << ", result: " << f.get() << std::endl;
std::promise
std::promise 类型模板提供设置异步结果的方法, 这样其他线程就可以通过 std::future 实例来索引该结果.
class SquareRoot { std::promise<double>& prom; public: SquareRoot(std::promise<double>& p) : prom(p) {} ~SquareRoot() {} void operator()(double x) { if (x < 0) { prom.set_exception(std::make_exception_ptr(std::out_of_range("x<0"))); } else { double result = std::sqrt(x); prom.set_value(result); } } }; std::promise<double> prom; SquareRoot p(prom); std::thread t(std::bind(&SquareRoot::operator(), &p, 1)); //std::thread t(std::bind(&SquareRoot::operator(), &p, -1)); std::future<double> f = prom.get_future(); try { double v = f.get(); std::cout << "value:" << v << std::endl; } catch (std::exception& e) { std::cout << "exception:" << e.what() << std::endl; } t.join();