waitcondition.h

Go to the documentation of this file.

 
#pragma once
 
#include <chrono>
#include <type_traits>
#include <condition_variable>
#include <promeki/core/namespace.h>
#include <promeki/core/error.h>
#include <promeki/core/mutex.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class WaitCondition {
        public:
                WaitCondition() = default;
 
                ~WaitCondition() = default;
 
                WaitCondition(const WaitCondition &) = delete;
                WaitCondition &operator=(const WaitCondition &) = delete;
                WaitCondition(WaitCondition &&) = delete;
                WaitCondition &operator=(WaitCondition &&) = delete;
 
                Error wait(Mutex &mutex, unsigned int timeoutMs = 0) {
                        std::unique_lock<std::mutex> lock(mutex._mutex, std::adopt_lock);
                        if(timeoutMs == 0) {
                                _cv.wait(lock);
                                lock.release();
                                return Error::Ok;
                        }
                        std::cv_status status = _cv.wait_for(lock,
                                std::chrono::milliseconds(timeoutMs));
                        lock.release();
                        return status == std::cv_status::no_timeout
                                ? Error::Ok : Error::Timeout;
                }
 
                template <typename Predicate,
                          typename = std::enable_if_t<std::is_invocable_r_v<bool, Predicate>>>
                Error wait(Mutex &mutex, Predicate pred, unsigned int timeoutMs = 0) {
                        std::unique_lock<std::mutex> lock(mutex._mutex, std::adopt_lock);
                        if(timeoutMs == 0) {
                                _cv.wait(lock, pred);
                                lock.release();
                                return Error::Ok;
                        }
                        bool result = _cv.wait_for(lock,
                                std::chrono::milliseconds(timeoutMs), pred);
                        lock.release();
                        return result ? Error::Ok : Error::Timeout;
                }
 
                void wakeOne() {
                        _cv.notify_one();
                }
 
                void wakeAll() {
                        _cv.notify_all();
                }
 
        private:
                std::condition_variable _cv;
};
 
PROMEKI_NAMESPACE_END