request_mem_builder.h

#pragma once
 
#include <memory>
#include <utility>
#include <variant>
 
#include <ucxx/request_builder_base.h>
#include <ucxx/request_data.h>
 
namespace ucxx {
 
// Forward declarations
class Endpoint;
class Request;
class RequestMem;

class RequestMemBuilder : public RequestCallbackBuilderBase<RequestMemBuilder> {
 private:
  std::shared_ptr<Endpoint> _endpoint;                    
  std::variant<data::MemPut, data::MemGet> _requestData;  
  uint64_t _remoteAddressOffset{0};                       
 
 public:
  explicit RequestMemBuilder(std::shared_ptr<Endpoint> endpoint,
                             std::variant<data::MemPut, data::MemGet> requestData);

  RequestMemBuilder& remoteAddressOffset(uint64_t offset) &;

  RequestMemBuilder&& remoteAddressOffset(uint64_t offset) &&;

  [[nodiscard]] std::shared_ptr<RequestMem> build();

  operator std::shared_ptr<RequestMem>();

  operator std::shared_ptr<Request>();
};

[[nodiscard]] inline RequestMemBuilder requestMemBuilder(
  std::shared_ptr<Endpoint> endpoint, std::variant<data::MemPut, data::MemGet> requestData)
{
  return RequestMemBuilder(std::move(endpoint), std::move(requestData));
}
 
}  // namespace ucxx