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;
 
  [[nodiscard]] friend std::shared_ptr<::ucxx::Future> createFuture(
    std::shared_ptr<::ucxx::Notifier> notifier);
 
  [[nodiscard]] 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