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