latch.hpp

浏览该文件的文档.

#ifndef NEFORCE_CORE_ASYNC_LATCH_HPP__
#define NEFORCE_CORE_ASYNC_LATCH_HPP__

 
#include "NeForce/core/async/atomic_base.hpp"
#include "NeForce/core/numeric/numeric_traits.hpp"
NEFORCE_BEGIN_NAMESPACE__



class latch {
private:
    alignas(alignof(platform_wait_t)) platform_wait_t counter_; 
 
public:
    static constexpr platform_wait_t max() noexcept { return numeric_traits<platform_wait_t>::max(); }

    constexpr explicit latch(const platform_wait_t expected) noexcept :
    counter_(expected) {
        NEFORCE_CONSTEXPR_ASSERT(expected >= 0);
    }
 
    ~latch() = default;                      
    latch(const latch&) = delete;            
    latch& operator=(const latch&) = delete; 

    NEFORCE_ALWAYS_INLINE void count_down(const platform_wait_t update = 1) {
        auto const old_value = _NEFORCE atomic_fetch_sub(&counter_, update, memory_order_release);
        if (old_value == update) {
            _NEFORCE atomic_notify_address(&counter_, true);
        }
    }

    NEFORCE_NODISCARD NEFORCE_ALWAYS_INLINE bool try_wait() const noexcept {
        return _NEFORCE atomic_load(&counter_, memory_order_acquire) == 0;
    }

    NEFORCE_ALWAYS_INLINE void wait() const noexcept {
        auto const predicate = [this] { return this->try_wait(); };
        _NEFORCE atomic_wait_address(&counter_, predicate);
    }

    NEFORCE_ALWAYS_INLINE void arrive_and_wait(const platform_wait_t update = 1) noexcept {
        count_down(update);
        wait();
    }
};
 // ThreadSync
 // AsyncComponents
 
NEFORCE_END_NAMESPACE__
#endif // NEFORCE_CORE_ASYNC_LATCH_HPP__