owning_wrapper.hpp

 
#pragma once
 
#include <memory>
 
namespace rapidsmpf {
 
class OwningWrapper {
  public:
    using deleter_type = void (*)(void*);
 
    OwningWrapper() = default;
 
    explicit OwningWrapper(void* obj, deleter_type deleter)
        : obj_{owning_type(obj, deleter)} {}
 
    [[nodiscard]] void* release() noexcept {
        return obj_.release();
    }
 
    [[nodiscard]] void* get() const noexcept {
        return obj_.get();
    }
 
  private:
    using owning_type = std::unique_ptr<void, deleter_type>;
    owning_type obj_{nullptr, [](void*) {}};
};
}  // namespace rapidsmpf