request_mem.h

#pragma once
#include <memory>
#include <string>
#include <utility>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/delayed_submission.h>
#include <ucxx/request.h>
#include <ucxx/request_mem_builder.h>
#include <ucxx/typedefs.h>
 
namespace ucxx {
 
class Buffer;
 
namespace internal {
class RecvMemMessage;
}  // namespace internal

class RequestMem : public Request {
 private:
  friend class internal::RecvMemMessage;
 
  uint64_t _remote_addr{};
  ucp_rkey_h _rkey{};

  RequestMem(std::shared_ptr<Endpoint> endpoint,
             const std::variant<data::MemPut, data::MemGet> requestData,
             std::string operationName,
             const bool enablePythonFuture                = false,
             RequestCallbackUserFunction callbackFunction = nullptr,
             RequestCallbackUserData callbackData         = nullptr);
 
  void populateDelayedSubmissionImpl() override;
 
 public:
  // Allow internal construction without exposing factory functions.
  friend class detail::ConstructorFactory;

  static void memPutCallback(void* request, ucs_status_t status, void* arg);

  static void memGetCallback(void* request, ucs_status_t status, void* arg);

  void request();
};
 
}  // namespace ucxx