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());
        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 < )
            {
                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, ));
    //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();