mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-18 21:44:02 +03:00
103 lines
2.3 KiB
C++
103 lines
2.3 KiB
C++
/* Copyright (C) 2014 InfiniDB, Inc.
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; version 2 of
|
|
the License.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
MA 02110-1301, USA. */
|
|
|
|
/******************************************************************************************
|
|
* $Id$
|
|
*
|
|
******************************************************************************************/
|
|
|
|
#include <iostream>
|
|
//#define NDEBUG
|
|
#include <cassert>
|
|
|
|
|
|
#include "poolallocator.h"
|
|
|
|
using namespace std;
|
|
|
|
|
|
namespace utils
|
|
{
|
|
PoolAllocator& PoolAllocator::operator=(const PoolAllocator& v)
|
|
{
|
|
allocSize = v.allocSize;
|
|
tmpSpace = v.tmpSpace;
|
|
useLock = v.useLock;
|
|
deallocateAll();
|
|
return *this;
|
|
}
|
|
|
|
void PoolAllocator::deallocateAll()
|
|
{
|
|
capacityRemaining = 0;
|
|
nextAlloc = NULL;
|
|
memUsage = 0;
|
|
mem.clear();
|
|
oob.clear();
|
|
}
|
|
|
|
void PoolAllocator::newBlock()
|
|
{
|
|
std::shared_ptr<uint8_t[]> next;
|
|
|
|
capacityRemaining = allocSize;
|
|
|
|
if (!tmpSpace || mem.size() == 0)
|
|
{
|
|
next.reset(new uint8_t[allocSize]);
|
|
mem.push_back(next);
|
|
nextAlloc = next.get();
|
|
}
|
|
else
|
|
nextAlloc = mem.front().get();
|
|
}
|
|
|
|
void* PoolAllocator::allocOOB(uint64_t size)
|
|
{
|
|
OOBMemInfo memInfo;
|
|
|
|
memUsage += size;
|
|
memInfo.mem.reset(new uint8_t[size]);
|
|
memInfo.size = size;
|
|
void* ret = (void*)memInfo.mem.get();
|
|
oob[ret] = memInfo;
|
|
return ret;
|
|
}
|
|
|
|
void PoolAllocator::deallocate(void* p)
|
|
{
|
|
bool _false = false;
|
|
if (useLock)
|
|
while (!lock.compare_exchange_weak(_false, true, std::memory_order_acquire))
|
|
_false = false;
|
|
OutOfBandMap::iterator it = oob.find(p);
|
|
|
|
if (it == oob.end())
|
|
{
|
|
if (useLock)
|
|
lock.store(false, std::memory_order_release);
|
|
return;
|
|
}
|
|
|
|
memUsage -= it->second.size;
|
|
oob.erase(it);
|
|
if (useLock)
|
|
lock.store(false, std::memory_order_release);
|
|
}
|
|
|
|
} // namespace utils
|