request_am_builder.h

 
#pragma once
 
#include <memory>
#include <optional>
#include <utility>
#include <variant>
 
#include <ucxx/request_builder_base.h>
#include <ucxx/request_data.h>
 
namespace ucxx {
 
// Forward declarations
class Endpoint;
class Request;
class RequestAm;
 
class RequestAmBuilder : public RequestCallbackBuilderBase<RequestAmBuilder> {
 private:
  std::shared_ptr<Endpoint> _endpoint;                       
  std::variant<data::AmSend, data::AmReceive> _requestData;  
 
 public:
  explicit RequestAmBuilder(std::shared_ptr<Endpoint> endpoint,
                            std::variant<data::AmSend, data::AmReceive> requestData);
 
  RequestAmBuilder& receiverCallbackInfo(std::optional<AmReceiverCallbackInfo> info) &;
 
  RequestAmBuilder&& receiverCallbackInfo(std::optional<AmReceiverCallbackInfo> info) &&;
 
  [[nodiscard]] std::shared_ptr<RequestAm> build();
 
  operator std::shared_ptr<RequestAm>();
 
  operator std::shared_ptr<Request>();
};
 
[[nodiscard]] inline RequestAmBuilder requestAmBuilder(
  std::shared_ptr<Endpoint> endpoint, std::variant<data::AmSend, data::AmReceive> requestData)
{
  return RequestAmBuilder(std::move(endpoint), std::move(requestData));
}
 
}  // namespace ucxx