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/request_stream_builder.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,
                std::string operationName,
                const bool enablePythonFuture = false);
 
  void populateDelayedSubmissionImpl() override;
 
 public:
  // Allow internal construction without exposing factory functions.
  friend class detail::ConstructorFactory;

  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