greenplumn CMemoryPoolPalloc 代码
文件路径：/src/backend/gpopt/utils/CMemoryPoolPalloc.cpp
/---------------------------------------------------------------------------
/ Greenplum Database
/ Copyright (C) 2019 VMware, Inc. or its affiliates.
/
/ @filename:
/ CMemoryPoolPalloc.cpp
/
/ @doc:
/ CMemoryPool implementation that uses PostgreSQL memory
/ contexts.
/
/---------------------------------------------------------------------------
extern "C" {
#include "postgres.h"
#include "utils/memutils.h"
}
#include "gpos/memory/CMemoryPool.h"
#include "gpopt/gpdbwrappers.h"
#include "gpopt/utils/CMemoryPoolPalloc.h"
using namespace gpos;
/ ctor
CMemoryPoolPalloc::CMemoryPoolPalloc()
{
m_cxt = gpdb::GPDBAllocSetContextCreate();
}
void *
CMemoryPoolPalloc::NewImpl(const ULONG bytes, const CHAR *, const ULONG,
CMemoryPool::EAllocationType eat)
{
/ if it's a singleton allocation, allocate requested memory
if (CMemoryPool::EatSingleton == eat)
{
return gpdb::GPDBMemoryContextAlloc(m_cxt, bytes);
}
/ if it's an array allocation, allocate header + requested memory
else
{
ULONG alloc_size = GPOS_MEM_ALIGNED_STRUCT_SIZE(SArrayAllocHeader) +
GPOS_MEM_ALIGNED_SIZE(bytes);
void *ptr = gpdb::GPDBMemoryContextAlloc(m_cxt, alloc_size);
if (nullptr == ptr)
{
return nullptr;
}
SArrayAllocHeader *header = static_cast<SArrayAllocHeader *>(ptr);
header->m_user_size = bytes;
return static_cast<BYTE *>(ptr) +
GPOS_MEM_ALIGNED_STRUCT_SIZE(SArrayAllocHeader);
}
}
void
CMemoryPoolPalloc::DeleteImpl(void *ptr, CMemoryPool::EAllocationType eat)
{
if (CMemoryPool::EatSingleton == eat)
{
gpdb::GPDBFree(ptr);
}
else
{
void *header = static_cast<BYTE *>(ptr) -
GPOS_MEM_ALIGNED_STRUCT_SIZE(SArrayAllocHeader);
gpdb::GPDBFree(header);
}
}
/ Prepare the memory pool to be deleted
void
CMemoryPoolPalloc::TearDown()
{
gpdb::GPDBMemoryContextDelete(m_cxt);
}
/ Total allocated size including management overheads
ULLONG
CMemoryPoolPalloc::TotalAllocatedSize() const
{
return MemoryContextGetCurrentSpace(m_cxt);
}
/ get user requested size of array allocation. Note: this is ONLY called for arrays
ULONG
CMemoryPoolPalloc::UserSizeOfAlloc(const void *ptr)
{
GPOS_ASSERT(ptr != nullptr);
void *void_header = static_cast<BYTE *>(const_cast<void *>(ptr)) -
GPOS_MEM_ALIGNED_STRUCT_SIZE(SArrayAllocHeader);
const SArrayAllocHeader *header =
static_cast<SArrayAllocHeader *>(void_header);
return header->m_user_size;
}
/ EOF
相关信息
相关文章
greenplumn CConstExprEvaluatorProxy 源码