muduo C++11 重写CountDownLatch.h

#ifndef MUDUO_BASE_COUNTDOWNLATCH_H
#define MUDUO_BASE_COUNTDOWNLATCH_H

#include "Condition.h"
#include "Mutex.h"

namespace muduo
{

class CountDownLatch : noncopyable
{
public:
    explicit CountDownLatch(int count)
        : mutex_(),
          condition_(mutex_),
          count_(count)
    {
    }

    void wait()
    {
        MutexLockGuard lock(mutex_);
        while (count_ > 0)
        {
            condition_.wait();
        }
    }

    void countDown()
    {
        MutexLockGuard lock(mutex_);
        --count_;
        if (count_ == 0)
        {
            condition_.notifyAll();
        }
    }

    int getCount() const
    {
        MutexLockGuard lock(mutex_);
        return count_;
    }

private:
    mutable MutexLock mutex_;
    Condition condition_;
    int count_;
};

} // namespace muduo

#endif // MUDUO_BASE_COUNTDOWNLATCH_H