105 lines
2.8 KiB
C++
105 lines
2.8 KiB
C++
//内存设计有问题，溜了溜了。还是先学习mimalloc吧
|
||
#include "memory_pool_debug.h"
|
||
namespace pmr {
|
||
inline MemoryPoolManager* pMainPool;
|
||
inline BlockAllocator* pMetaAlloc;
|
||
void CleanMainMemPool() {
|
||
if (!pMainPool) return;
|
||
dumpMemoryPoolLeaks(pMainPool);
|
||
pMainPool->~MemoryPoolManager();
|
||
dumpMetaMemoryLeaks(pMetaAlloc);
|
||
pMetaAlloc->~BlockAllocator();
|
||
}
|
||
void* meta_malloc(size_t bytes, size_t alignment)
|
||
{
|
||
if (bytes > META_BLOCK_EMEM_SIZE * MAX_BLOCK_ELEM_GROUP_N || !pMetaAlloc->try_allocate(bytes)) {
|
||
return pMainPool->do_allocate(bytes, alignment);
|
||
}
|
||
using MemBlock = BlockAllocator::MemBlock;
|
||
bytes += sizeof(MemBlock);
|
||
MemBlock* pBlock = (MemBlock*)pMetaAlloc->do_allocate(bytes, alignment);
|
||
new(pBlock)MemBlock(bytes);
|
||
return (char*)pBlock + sizeof(MemBlock);
|
||
}
|
||
void meta_free(void* ptr)
|
||
{
|
||
if (!ptr) {
|
||
return;
|
||
}
|
||
using MemBlock = BlockAllocator::MemBlock;
|
||
MemBlock* pBlock = (MemBlock*)((char*)ptr - sizeof(MemBlock));
|
||
if (!*pBlock) {
|
||
pMainPool->do_deallocate(ptr);
|
||
return;
|
||
}
|
||
pMetaAlloc->do_deallocate(ptr, pBlock->size, 0);
|
||
}
|
||
}
|
||
namespace pmr {
|
||
struct MemPoolWrap {
|
||
inline static char AllocData[sizeof(BlockAllocator)];
|
||
char pMemory[sizeof(MemoryPoolManager)];
|
||
bool isMainTread;
|
||
MemPoolWrap() {
|
||
isMainTread = !pMainPool;
|
||
if (isMainTread) {
|
||
pMainPool = (MemoryPoolManager*)&pMemory;
|
||
pMetaAlloc = (BlockAllocator*)&AllocData;
|
||
std::construct_at(pMetaAlloc, MEMORY_BLOCK_SIZE, META_BLOCK_EMEM_SIZE);
|
||
std::atexit(CleanMainMemPool);
|
||
}
|
||
std::construct_at((MemoryPoolManager*)&pMemory);
|
||
}
|
||
~MemPoolWrap() {
|
||
auto pool = (MemoryPoolManager*)&pMemory;
|
||
if (!isMainTread) {//主线程最后析构
|
||
pool->~MemoryPoolManager();
|
||
}
|
||
}
|
||
MemoryPoolManager* operator->() {
|
||
return (MemoryPoolManager*)&pMemory;
|
||
}
|
||
};
|
||
}
|
||
thread_local pmr::MemPoolWrap MemPool;
|
||
|
||
MEMALLOC_API void* me_malloc(size_t size)
|
||
{
|
||
return MemPool->do_allocate(size, MEMORY_ALIGN_N);
|
||
}
|
||
|
||
MEMALLOC_API void* me_malloc_nothrow(size_t size) noexcept
|
||
{
|
||
return MemPool->do_allocate(size, MEMORY_ALIGN_N);
|
||
}
|
||
|
||
MEMALLOC_API void me_free(void* p) noexcept
|
||
{
|
||
MemPool->do_deallocate(p);
|
||
}
|
||
#if (__cplusplus >= 201402L || _MSC_VER >= 1916)
|
||
MEMALLOC_API void me_free_size(void* p, size_t size) noexcept
|
||
{
|
||
MemPool->do_deallocate(p);
|
||
}
|
||
#endif
|
||
#if (__cplusplus > 201402L || defined(__cpp_aligned_new))
|
||
MEMALLOC_API void* me_malloc_aligned(size_t size, size_t alignment)
|
||
{
|
||
return MemPool->do_allocate(size, alignment);
|
||
}
|
||
|
||
MEMALLOC_API void* me_malloc_aligned_nothrow(size_t size, size_t alignment) noexcept
|
||
{
|
||
return MemPool->do_allocate(size, alignment);
|
||
}
|
||
MEMALLOC_API void me_free_aligned(void* p, size_t alignment) noexcept
|
||
{
|
||
return MemPool->do_deallocate(p);
|
||
}
|
||
MEMALLOC_API void me_free_size_aligned(void* p, size_t size, size_t alignment) noexcept
|
||
{
|
||
MemPool->do_deallocate(p);
|
||
}
|
||
|
||
#endif |