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