endpoint_builder.h

 
#pragma once
 
#include <cstdint>
#include <memory>
#include <string>
#include <utility>
 
#include <ucp/api/ucp.h>
 
namespace ucxx {
 
class Address;
class Endpoint;
class Listener;
class Worker;
 
class EndpointBuilder final {
 public:
  EndpointBuilder(std::shared_ptr<Worker> worker, std::string ipAddress, uint16_t port);
 
  EndpointBuilder(std::shared_ptr<Listener> listener, ucp_conn_request_h connRequest);
 
  EndpointBuilder(std::shared_ptr<Worker> worker, std::shared_ptr<Address> address);
 
  ~EndpointBuilder();
 
  EndpointBuilder(const EndpointBuilder& other);
  EndpointBuilder& operator=(const EndpointBuilder& other);
  EndpointBuilder(EndpointBuilder&&) noexcept;
  EndpointBuilder& operator=(EndpointBuilder&&) noexcept;
 
  operator std::shared_ptr<Endpoint>();
 
  EndpointBuilder& endpointErrorHandling(bool enable = true);
 
  [[nodiscard]] std::shared_ptr<Endpoint> build();
 
 private:
  struct Impl;
  std::unique_ptr<Impl> _impl;
};
 
[[nodiscard]] inline EndpointBuilder endpointBuilder(std::shared_ptr<Worker> worker,
                                                     std::string ipAddress,
                                                     uint16_t port)
{
  return EndpointBuilder(std::move(worker), std::move(ipAddress), port);
}
 
[[nodiscard]] inline EndpointBuilder endpointBuilder(std::shared_ptr<Listener> listener,
                                                     ucp_conn_request_h connRequest)
{
  return EndpointBuilder(std::move(listener), connRequest);
}
 
[[nodiscard]] inline EndpointBuilder endpointBuilder(std::shared_ptr<Worker> worker,
                                                     std::shared_ptr<Address> address)
{
  return EndpointBuilder(std::move(worker), std::move(address));
}
 
}  // namespace ucxx