memory_handle.h

 
#pragma once
#include <memory>
#include <string>
#include <utility>
 
#include <ucp/api/ucp.h>
 
#include <ucxx/component.h>
#include <ucxx/context.h>
 
namespace ucxx {
 
class RemoteKey;
 
class MemoryHandle : public Component {
 private:
  ucp_mem_h _handle{};       
  size_t _size{0};           
  uint64_t _baseAddress{0};  
 
  MemoryHandle(std::shared_ptr<Context> context, const size_t size, void* buffer);
 
 public:
  MemoryHandle()                               = delete;
  MemoryHandle(const MemoryHandle&)            = delete;
  MemoryHandle& operator=(MemoryHandle const&) = delete;
  MemoryHandle(MemoryHandle&& o)               = delete;
  MemoryHandle& operator=(MemoryHandle&& o)    = delete;
 
  friend std::shared_ptr<MemoryHandle> createMemoryHandle(std::shared_ptr<Context> context,
                                                          const size_t size,
                                                          void* buffer);
 
  ~MemoryHandle();
 
  ucp_mem_h getHandle();
 
  size_t getSize() const;
 
  uint64_t getBaseAddress();
 
  std::shared_ptr<RemoteKey> createRemoteKey();
};
 
}  // namespace ucxx