notifier.h

#pragma once
 
#include <condition_variable>
#include <memory>
#include <mutex>
#include <utility>
#include <vector>
 
#include <ucxx/future.h>
#include <ucxx/notifier.h>
 
namespace ucxx {
 
namespace python {

class Notifier : public ::ucxx::Notifier {
 private:
  std::mutex _notifierThreadMutex{};  
  std::vector<std::pair<std::shared_ptr<::ucxx::Future>, ucs_status_t>>
    _notifierThreadFutureStatus{};               
  bool _notifierThreadFutureStatusReady{false};  
  RequestNotifierThreadState _notifierThreadFutureStatusFinished{
    RequestNotifierThreadState::NotRunning};  
  std::condition_variable
    _notifierThreadConditionVariable{};  

  Notifier() = default;

  [[nodiscard]] RequestNotifierWaitState waitRequestNotifierWithoutTimeout();

  [[nodiscard]] RequestNotifierWaitState waitRequestNotifierWithTimeout(uint64_t period);
 
 public:
  Notifier(const Notifier&)            = delete;
  Notifier& operator=(Notifier const&) = delete;
  Notifier(Notifier&& o)               = delete;
  Notifier& operator=(Notifier&& o)    = delete;

  friend std::shared_ptr<::ucxx::Notifier> createNotifier();

  virtual ~Notifier();

  void scheduleFutureNotify(std::shared_ptr<::ucxx::Future> future, ucs_status_t status) override;

  [[nodiscard]] RequestNotifierWaitState waitRequestNotifier(uint64_t period) override;

  void runRequestNotifier() override;

  void stopRequestNotifierThread() override;

  [[nodiscard]] bool isRunning() const override;
};
 
}  // namespace python
 
}  // namespace ucxx