waitcondition.h

Go to the documentation of this file.

 
#pragma once
 
 
#include <promeki/config.h>
#if PROMEKI_ENABLE_CORE
#include <chrono>
#include <type_traits>
#include <condition_variable>
#include <promeki/namespace.h>
#include <promeki/error.h>
#include <promeki/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
 
#endif // PROMEKI_ENABLE_CORE