我正在使用内存池来实现更快的分配和释放。内存池的Release()
操作需要释放的指针和分配的字节数。
现在我想将从内存池获得的数组指针存储在unique_ptr
. 数组大小可变,所以我需要一个有状态的自定义删除器来存储数组的大小。
您能否提供有关如何声明此类的示例代码unique_ptr
?
更新:在 Artyer 的提示让我清楚之后,这是我尝试过的:
template<typename T> struct MPSingleDeleter {
void operator()(T *p) {
p->~T();
MemPool::Instance().Release(p, sizeof(T));
}
};
template<typename T> struct MPArrayDeleter {
size_t _nItems;
explicit MPArrayDeleter(const size_t nItems) : _nItems(nItems) {
}
void operator()(T *p) {
for (size_t i = 0; i < _nItems; i++) {
p[i].~T();
}
MemPool::Instance().Release(p, _nItems * sizeof(T));
}
};
template <typename T> using SingleUP = std::unique_ptr<T, MPSingleDeleter<T>>;
template <typename T> using ArrayUP = std::unique_ptr<T[], MPArrayDeleter<T>>;
struct MemHelper {
template<typename T, typename ...Args> static T* NewSingle(Args&&... args) {
void * const p = MemPool::Instance().Acquire(sizeof(T));
::new (p) T(std::forward<Args>(args)...);
return static_cast<T*>(p);
}
template<typename T, typename ...Args> static T* NewArray(const size_t nItems, Args&&... args) {
T *p = static_cast<T*>(MemPool::Instance().Acquire(nItems * sizeof(T)));
for (size_t i = 0; i < nItems; i++) {
void * const pv = static_cast<void *>(p + i);
::new (pv) T(std::forward<Args>(args)...);
}
return p;
}
template<typename T, typename ...Args> static SingleUP<T> MakeSUP(Args&&... args) {
return SingleUP<T>(NewSingle<T>(std::forward<Args>(args)...));
}
template<typename T, typename ...Args> static ArrayUP<T> MakeAUP(const size_t nItems, Args&&... args) {
return ArrayUP<T>(NewArray<T>(nItems, std::forward<Args>(args)...), MPArrayDeleter<T>(nItems));
}
};
现在unique_ptr
变量的声明很简单:
// Array of double
ArrayUP<double> pBuffer = MemHelper::MakeAUP<double>(2ui64 * nItems);
// Single Connection
SingleUP<Connection> pConn = MemHelper::MakeSUP<Connection>(ioContext);
您通常会在内存池中分配数据之前存储大小,因此您只需传递一个从指针获取大小的无状态删除器。
不过,您可以轻松地执行以下操作:
#include <memory>
struct pool_deleter {
std::size_t size;
template<class T>
void operator()(T* ptr) noexcept {
std::destroy_n(ptr, size);
Release(ptr, size * sizeof(T));
}
};
template<class T>
using pool_ptr = std::unique_ptr<T, pool_deleter>;
// Used like:
std::size_t n = /* ... */;
T* ptr_ = /* Allocate and construct n objects from pool */;
pool_ptr<T> ptr{ptr_, pool_deleter{n}}; // Pass the custom deleter as an argument
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句