request_tag.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 RequestTag : public Request {
 private:
  RequestTag(std::shared_ptr<Component> endpointOrWorker,
             const std::variant<data::TagSend, data::TagReceive> requestData,
             const std::string operationName,
             const bool enablePythonFuture                = false,
             RequestCallbackUserFunction callbackFunction = nullptr,
             RequestCallbackUserData callbackData         = nullptr);
 
 public:
  [[nodiscard]] friend std::shared_ptr<RequestTag> createRequestTag(
    std::shared_ptr<Component> endpointOrWorker,
    const std::variant<data::TagSend, data::TagReceive> requestData,
    const bool enablePythonFuture,
    RequestCallbackUserFunction callbackFunction,
    RequestCallbackUserData callbackData);
 
  virtual void populateDelayedSubmission();
 
  void request();
 
  static void tagSendCallback(void* request, ucs_status_t status, void* arg);
 
  static void tagRecvCallback(void* request,
                              ucs_status_t status,
                              const ucp_tag_recv_info_t* info,
                              void* arg);
 
  void callback(void* request, ucs_status_t status, const ucp_tag_recv_info_t* info);
};
 
}  // namespace ucxx