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

#ifndef PLATFORM_V2_IMPL_G3_SETTABLE_FUTURE_ANY_H_
#define PLATFORM_V2_IMPL_G3_SETTABLE_FUTURE_ANY_H_

#include <utility>

#include "platform_v2/api/platform.h"
#include "platform_v2/api/settable_future.h"
#include "absl/synchronization/mutex.h"
#include "absl/time/clock.h"
#include "absl/types/any.h"

namespace location {
namespace nearby {
namespace g3 {

class SettableFutureAny : public api::SettableFuture<absl::any> {
 public:
  SettableFutureAny() = default;
  ~SettableFutureAny() override = default;

  bool Set(const absl::any& value) override {
    absl::MutexLock lock(&mutex_);
    if (!done_) {
      value_ = value;
      done_ = true;
      exception_ = {Exception::kSuccess};
      completed_.SignalAll();
    }
    return true;
  }

  bool Set(absl::any&& value) override {
    absl::MutexLock lock(&mutex_);
    if (!done_) {
      value_ = std::move(value);
      done_ = true;
      exception_ = {Exception::kSuccess};
      completed_.SignalAll();
    }
    return true;
  }

  bool SetException(Exception exception) override {
    absl::MutexLock lock(&mutex_);
    return SetExceptionLocked(exception);
  }

  void AddListener(Runnable runnable, api::Executor* executor) override {}

  ExceptionOr<std::any> Get() override {
    absl::MutexLock lock(&mutex_);
    while (!done_) {
      completed_.Wait(&mutex_);
    }
    return exception_.value != Exception::kSuccess
               ? ExceptionOr<std::any>{exception_.value}
               : ExceptionOr<std::any>{value_};
  }

  ExceptionOr<std::any> Get(absl::Duration timeout) override {
    absl::MutexLock lock(&mutex_);
    while (!done_) {
      absl::Time start_time = absl::Now();
      if (completed_.WaitWithTimeout(&mutex_, timeout)) {
        SetExceptionLocked({Exception::kTimeout});
        break;
      }
      absl::Duration spent = absl::Now() - start_time;
      if (spent < timeout) {
        timeout -= spent;
      } else if (!done_) {
        SetExceptionLocked({Exception::kTimeout});
        break;
      }
    }
    return exception_.value != Exception::kSuccess
               ? ExceptionOr<std::any>{exception_.value}
               : ExceptionOr<std::any>{value_};
  }

 private:
  bool SetExceptionLocked(Exception exception) {
    if (!done_) {
      exception_ = exception.value != Exception::kSuccess
                       ? exception
                       : Exception{Exception::kFailed};
      done_ = true;
      completed_.SignalAll();
    }
    return true;
  }

  absl::Mutex mutex_;
  absl::CondVar completed_;
  bool done_{false};
  absl::any value_;
  Exception exception_{Exception::kFailed};
};

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

#endif  // PLATFORM_V2_IMPL_G3_SETTABLE_FUTURE_ANY_H_