request_am.h

 
#pragma once
#include <cstddef>
#include <memory>
#include <string>
#include <utility>
#include <vector>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/delayed_submission.h>
#include <ucxx/request.h>
#include <ucxx/request_am_builder.h>
#include <ucxx/typedefs.h>
 
namespace ucxx {
 
class Buffer;
 
namespace internal {
class RecvAmMessage;
}  // namespace internal
 
class RequestAm : public Request {
 private:
  friend class internal::RecvAmMessage;
 
  std::vector<std::byte> _header{};  
 
  RequestAm(std::shared_ptr<Component> endpointOrWorker,
            const std::variant<data::AmSend, data::AmReceive> 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;
 
  void cancel() override;
 
  void request();
 
  [[nodiscard]] static ucs_status_t recvCallback(void* arg,
                                                 const void* header,
                                                 size_t header_length,
                                                 void* data,
                                                 size_t length,
                                                 const ucp_am_recv_param_t* param);
 
  [[nodiscard]] std::shared_ptr<Buffer> getRecvBuffer() override;
 
  [[nodiscard]] std::string getRecvHeader() override;
};
 
}  // namespace ucxx