latch.h

Go to the documentation of this file.

 
#ifndef CPPCOMMON_THREADS_LATCH_H
#define CPPCOMMON_THREADS_LATCH_H
 
#include "time/timestamp.h"
 
#include <cassert>
#include <condition_variable>
#include <memory>
#include <mutex>
 
namespace CppCommon {
 
 
class Latch
{
public:
 
    explicit Latch(int threads) noexcept;
    Latch(const Latch&) = delete;
    Latch(Latch&&) = delete;
    ~Latch() = default;
 
    Latch& operator=(const Latch&) = delete;
    Latch& operator=(Latch&&) = delete;
 
    int threads() const noexcept { return _threads; }
 
 
    void Reset(int threads) noexcept;
 
 
    void CountDown() noexcept;
 
 
    void CountDownAndWait() noexcept;
 
 
    void Wait() noexcept;
 
 
    bool TryWait() noexcept;
 
 
    bool TryWaitFor(const Timespan& timespan) noexcept;
 
    bool TryWaitUntil(const Timestamp& timestamp) noexcept;
 
private:
    std::mutex _mutex;
    std::condition_variable _cond;
    int _generation;
    int _threads;
 
    bool CountDown(std::unique_lock<std::mutex>& lock) noexcept;
};
 
} // namespace CppCommon
 
#include "latch.inl"
 
#endif // CPPCOMMON_THREADS_LATCH_H