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>
 
namespace ucxx {
 
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:
  friend std::shared_ptr<RemoteKey> createRemoteKeyFromMemoryHandle(
    std::shared_ptr<MemoryHandle> memoryHandle);
 
  friend std::shared_ptr<RemoteKey> createRemoteKeyFromSerialized(
    std::shared_ptr<Endpoint> endpoint, SerializedRemoteKey serializedRemoteKey);
 
  ~RemoteKey();
 
  [[nodiscard]] ucp_rkey_h getHandle();
 
  [[nodiscard]] size_t getSize() const;
 
  [[nodiscard]] uint64_t getBaseAddress();
 
  [[nodiscard]] SerializedRemoteKey serialize() const;
};
 
}  // namespace ucxx