request_tag_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 Component;
class Request;
class RequestTag;

class RequestTagBuilder : public RequestCallbackBuilderBase<RequestTagBuilder> {
 private:
  std::shared_ptr<Component> _endpointOrWorker;  
  std::variant<data::TagSend, data::TagReceive, data::TagReceiveWithHandle>
    _requestData;  
 
 public:
  explicit RequestTagBuilder(
    std::shared_ptr<Component> endpointOrWorker,
    std::variant<data::TagSend, data::TagReceive, data::TagReceiveWithHandle> requestData);

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

  operator std::shared_ptr<RequestTag>();

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

[[nodiscard]] inline RequestTagBuilder requestTagBuilder(
  std::shared_ptr<Component> endpointOrWorker,
  std::variant<data::TagSend, data::TagReceive, data::TagReceiveWithHandle> requestData)
{
  return RequestTagBuilder(std::move(endpointOrWorker), std::move(requestData));
}
 
}  // namespace ucxx