nearby/cpp/platform_v2/impl/g3/count_down_latch.h

#ifndef PLATFORM_V2_IMPL_G3_COUNT_DOWN_LATCH_H_
#define PLATFORM_V2_IMPL_G3_COUNT_DOWN_LATCH_H_

#include "platform_v2/api/count_down_latch.h"
#include "absl/base/thread_annotations.h"
#include "absl/synchronization/mutex.h"
#include "absl/time/clock.h"

namespace location {
namespace nearby {
namespace g3 {

// A synchronization aid that allows one or more threads to wait until a set of
// operations being performed in other threads completes.
//
// https://docs.oracle.com/javase/8/docs/api/java/util/concurrent/CountDownLatch.html
class CountDownLatch final : public api::CountDownLatch {
 public:
  explicit CountDownLatch(int count) : count_(count) {}
  CountDownLatch(const CountDownLatch&) = delete;
  CountDownLatch& operator=(const CountDownLatch&) = delete;
  CountDownLatch(CountDownLatch&&) = delete;
  CountDownLatch& operator=(CountDownLatch&&) = delete;
  ExceptionOr<bool> Await(absl::Duration timeout) override {
    absl::MutexLock lock(&mutex_);
    absl::Time deadline = absl::Now() + timeout;
    while (count_ > 0) {
      if (cond_.WaitWithDeadline(&mutex_, deadline)) {
        return ExceptionOr<bool>(false);
      }
    }
    return ExceptionOr<bool>(true);
  }
  Exception Await() override {
    absl::MutexLock lock(&mutex_);
    while (count_ > 0) {
      cond_.Wait(&mutex_);
    }
    return {Exception::kSuccess};
  }
  void CountDown() override {
    absl::MutexLock lock(&mutex_);
    if (count_ > 0 && --count_ == 0) {
      cond_.SignalAll();
    }
  }

 private:
  absl::Mutex mutex_;   // Mutex to be used with cond_.Wait...() method family.
  absl::CondVar cond_;  // Condition to synchronize up to N waiting threads.
  int count_
      ABSL_GUARDED_BY(mutex_);  // When zero, latch should release all waiters.
};

}  // namespace g3
}  // namespace nearby
}  // namespace location

#endif  // PLATFORM_V2_IMPL_G3_COUNT_DOWN_LATCH_H_