request_am.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/typedefs.h>
 
namespace ucxx {
 
class Buffer;
 
namespace internal {
class RecvAmMessage;
}  // namespace internal
 
class RequestAm : public Request {
 private:
  friend class internal::RecvAmMessage;
 
  RequestAm(std::shared_ptr<Component> endpointOrWorker,
            const std::variant<data::AmSend, data::AmReceive> requestData,
            const std::string operationName,
            const bool enablePythonFuture                = false,
            RequestCallbackUserFunction callbackFunction = nullptr,
            RequestCallbackUserData callbackData         = nullptr);
 
 public:
  [[nodiscard]] friend std::shared_ptr<RequestAm> createRequestAm(
    std::shared_ptr<Endpoint> endpoint,
    const std::variant<data::AmSend, data::AmReceive> requestData,
    const bool enablePythonFuture,
    RequestCallbackUserFunction callbackFunction,
    RequestCallbackUserData callbackData);
 
  void cancel() override;
 
  void populateDelayedSubmission() 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;
};
 
}  // namespace ucxx