request_stream.h

 
#pragma once
#include <memory>
#include <string>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/delayed_submission.h>
#include <ucxx/request.h>
#include <ucxx/request_data.h>
#include <ucxx/typedefs.h>
 
namespace ucxx {
 
class RequestStream : public Request {
 private:
  RequestStream(std::shared_ptr<Endpoint> endpoint,
                const std::variant<data::StreamSend, data::StreamReceive> requestData,
                const std::string operationName,
                const bool enablePythonFuture = false);
 
 public:
  friend std::shared_ptr<RequestStream> createRequestStream(
    std::shared_ptr<Endpoint> endpoint,
    const std::variant<data::StreamSend, data::StreamReceive> requestData,
    const bool enablePythonFuture);
 
  virtual void populateDelayedSubmission();
 
  void request();
 
  void callback(void* request, ucs_status_t status, size_t length);
 
  static void streamSendCallback(void* request, ucs_status_t status, void* arg);
 
  static void streamRecvCallback(void* request, ucs_status_t status, size_t length, void* arg);
};
 
}  // namespace ucxx