python_future.h

#pragma once
 
#include <functional>
#include <memory>
 
#include <Python.h>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/future.h>
#include <ucxx/notifier.h>
#include <ucxx/python/future.h>
 
namespace ucxx {
 
namespace python {

class Future : public ::ucxx::Future {
 private:
  PyObject* _asyncioEventLoop{nullptr};  
  PyObject* _handle{nullptr};            

  explicit Future(PyObject* asyncioEventLoop, std::shared_ptr<::ucxx::Notifier> notifier);
 
 public:
  Future()                         = delete;
  Future(const Future&)            = delete;
  Future& operator=(Future const&) = delete;
  Future(Future&& o)               = delete;
  Future& operator=(Future&& o)    = delete;

  friend std::shared_ptr<::ucxx::Future> createFuture(std::shared_ptr<::ucxx::Notifier> notifier);

  friend std::shared_ptr<::ucxx::Future> createFutureWithEventLoop(
    PyObject* asyncioEventLoop, std::shared_ptr<::ucxx::Notifier> notifier);

  virtual ~Future();

  void notify(ucs_status_t status);

  void set(ucs_status_t status);

  [[nodiscard]] void* getHandle();

  [[nodiscard]] void* release();
};
 
}  // namespace python
 
}  // namespace ucxx