remote_key.h

#pragma once
#include <memory>
#include <string>
#include <utility>
#include <vector>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/component.h>
#include <ucxx/endpoint.h>
#include <ucxx/memory_handle.h>
#include <ucxx/remote_key_builder.h>
 
namespace ucxx {
 
class RemoteKey;

typedef size_t SerializedRemoteKeyHash;

class RemoteKey : public Component {
 private:
  ucp_rkey_h _remoteKey{nullptr};  
  void* _packedRemoteKey{
    nullptr};  
  size_t _packedRemoteKeySize{0};              
  std::vector<char> _packedRemoteKeyVector{};  
  uint64_t _memoryBaseAddress{0};              
  size_t _memorySize{0};                       

  explicit RemoteKey(std::shared_ptr<MemoryHandle> memoryHandle);

  RemoteKey(std::shared_ptr<Endpoint> endpoint, SerializedRemoteKey serializedRemoteKey);

  void deserialize(const SerializedRemoteKey& serializedHeader);
 
 public:
  // Allow internal construction without exposing factory functions.
  friend class detail::ConstructorFactory;
 
  ~RemoteKey();

  [[nodiscard]] ucp_rkey_h getHandle();

  [[nodiscard]] size_t getSize() const;

  [[nodiscard]] uint64_t getBaseAddress();

  [[nodiscard]] SerializedRemoteKey serialize() const;
};
 
}  // namespace ucxx