greenplumn CMemoryPoolPalloc 源码
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 源码
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
6、 golang
-
7、 openharmony