You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-07-27 21:01:50 +03:00
clang format apply
This commit is contained in:
Leonid Fedorov
@ -40,199 +40,204 @@ namespace bi = boost::interprocess;
|
||||
|
||||
namespace BRM
|
||||
{
|
||||
|
||||
BRMShmImpl::BRMShmImpl(unsigned key, off_t size, bool readOnly) :
|
||||
fKey(key), fSize(size), fReadOnly(readOnly)
|
||||
BRMShmImpl::BRMShmImpl(unsigned key, off_t size, bool readOnly) : fKey(key), fSize(size), fReadOnly(readOnly)
|
||||
{
|
||||
string keyName = ShmKeys::keyToName(fKey);
|
||||
string keyName = ShmKeys::keyToName(fKey);
|
||||
|
||||
if (fSize == 0)
|
||||
{
|
||||
unsigned tries = 0;
|
||||
again:
|
||||
|
||||
try
|
||||
{
|
||||
bi::shared_memory_object shm(bi::open_only, keyName.c_str(), bi::read_write);
|
||||
off_t curSize = 0;
|
||||
#ifdef _MSC_VER
|
||||
bi::offset_t tmp = 0;
|
||||
shm.get_size(tmp);
|
||||
curSize = static_cast<off_t>(tmp);
|
||||
#else
|
||||
shm.get_size(curSize);
|
||||
#endif
|
||||
|
||||
if (curSize == 0) throw
|
||||
bi::interprocess_exception("shm size is zero");
|
||||
}
|
||||
catch (bi::interprocess_exception&)
|
||||
{
|
||||
if (++tries > 10)
|
||||
{
|
||||
log("BRMShmImpl::BRMShmImpl(): retrying on size==0");
|
||||
throw;
|
||||
}
|
||||
|
||||
cerr << "BRMShmImpl::BRMShmImpl(): retrying on size==0" << endl;
|
||||
usleep(500 * 1000);
|
||||
goto again;
|
||||
}
|
||||
}
|
||||
if (fSize == 0)
|
||||
{
|
||||
unsigned tries = 0;
|
||||
again:
|
||||
|
||||
try
|
||||
{
|
||||
bi::permissions perms;
|
||||
perms.set_unrestricted();
|
||||
bi::shared_memory_object shm(bi::create_only, keyName.c_str(), bi::read_write, perms);
|
||||
idbassert(fSize > 0);
|
||||
shm.truncate(fSize);
|
||||
fShmobj.swap(shm);
|
||||
}
|
||||
catch (bi::interprocess_exception &b)
|
||||
{
|
||||
if (b.get_error_code() != bi::already_exists_error) {
|
||||
ostringstream o;
|
||||
o << "BRM caught an exception creating a shared memory segment: " << b.what();
|
||||
log(o.str());
|
||||
throw;
|
||||
}
|
||||
bi::shared_memory_object *shm = NULL;
|
||||
try {
|
||||
shm = new bi::shared_memory_object(bi::open_only, keyName.c_str(), bi::read_write);
|
||||
}
|
||||
catch (exception &e) {
|
||||
ostringstream o;
|
||||
o << "BRM caught an exception attaching to a shared memory segment (" << keyName << "): " << b.what();
|
||||
log(o.str());
|
||||
throw;
|
||||
}
|
||||
off_t curSize = 0;
|
||||
bi::shared_memory_object shm(bi::open_only, keyName.c_str(), bi::read_write);
|
||||
off_t curSize = 0;
|
||||
#ifdef _MSC_VER
|
||||
bi::offset_t tmp = 0;
|
||||
shm->get_size(tmp);
|
||||
curSize = static_cast<off_t>(tmp);
|
||||
bi::offset_t tmp = 0;
|
||||
shm.get_size(tmp);
|
||||
curSize = static_cast<off_t>(tmp);
|
||||
#else
|
||||
shm->get_size(curSize);
|
||||
shm.get_size(curSize);
|
||||
#endif
|
||||
idbassert(curSize > 0);
|
||||
idbassert(curSize >= fSize);
|
||||
fShmobj.swap(*shm);
|
||||
delete shm;
|
||||
fSize = curSize;
|
||||
}
|
||||
|
||||
if (fReadOnly)
|
||||
{
|
||||
bi::mapped_region ro_region(fShmobj, bi::read_only);
|
||||
fMapreg.swap(ro_region);
|
||||
if (curSize == 0)
|
||||
throw bi::interprocess_exception("shm size is zero");
|
||||
}
|
||||
else
|
||||
catch (bi::interprocess_exception&)
|
||||
{
|
||||
bi::mapped_region region(fShmobj, bi::read_write);
|
||||
fMapreg.swap(region);
|
||||
if (++tries > 10)
|
||||
{
|
||||
log("BRMShmImpl::BRMShmImpl(): retrying on size==0");
|
||||
throw;
|
||||
}
|
||||
|
||||
cerr << "BRMShmImpl::BRMShmImpl(): retrying on size==0" << endl;
|
||||
usleep(500 * 1000);
|
||||
goto again;
|
||||
}
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
bi::permissions perms;
|
||||
perms.set_unrestricted();
|
||||
bi::shared_memory_object shm(bi::create_only, keyName.c_str(), bi::read_write, perms);
|
||||
idbassert(fSize > 0);
|
||||
shm.truncate(fSize);
|
||||
fShmobj.swap(shm);
|
||||
}
|
||||
catch (bi::interprocess_exception& b)
|
||||
{
|
||||
if (b.get_error_code() != bi::already_exists_error)
|
||||
{
|
||||
ostringstream o;
|
||||
o << "BRM caught an exception creating a shared memory segment: " << b.what();
|
||||
log(o.str());
|
||||
throw;
|
||||
}
|
||||
bi::shared_memory_object* shm = NULL;
|
||||
try
|
||||
{
|
||||
shm = new bi::shared_memory_object(bi::open_only, keyName.c_str(), bi::read_write);
|
||||
}
|
||||
catch (exception& e)
|
||||
{
|
||||
ostringstream o;
|
||||
o << "BRM caught an exception attaching to a shared memory segment (" << keyName << "): " << b.what();
|
||||
log(o.str());
|
||||
throw;
|
||||
}
|
||||
off_t curSize = 0;
|
||||
#ifdef _MSC_VER
|
||||
bi::offset_t tmp = 0;
|
||||
shm->get_size(tmp);
|
||||
curSize = static_cast<off_t>(tmp);
|
||||
#else
|
||||
shm->get_size(curSize);
|
||||
#endif
|
||||
idbassert(curSize > 0);
|
||||
idbassert(curSize >= fSize);
|
||||
fShmobj.swap(*shm);
|
||||
delete shm;
|
||||
fSize = curSize;
|
||||
}
|
||||
|
||||
if (fReadOnly)
|
||||
{
|
||||
bi::mapped_region ro_region(fShmobj, bi::read_only);
|
||||
fMapreg.swap(ro_region);
|
||||
}
|
||||
else
|
||||
{
|
||||
bi::mapped_region region(fShmobj, bi::read_write);
|
||||
fMapreg.swap(region);
|
||||
}
|
||||
}
|
||||
|
||||
int BRMShmImpl::grow(unsigned newKey, off_t newSize)
|
||||
{
|
||||
idbassert(newKey != fKey);
|
||||
idbassert(newSize >= fSize);
|
||||
idbassert(newKey != fKey);
|
||||
idbassert(newSize >= fSize);
|
||||
|
||||
string oldName = fShmobj.get_name();
|
||||
string oldName = fShmobj.get_name();
|
||||
|
||||
string keyName = ShmKeys::keyToName(newKey);
|
||||
bi::permissions perms;
|
||||
perms.set_unrestricted();
|
||||
bi::shared_memory_object shm(bi::create_only, keyName.c_str(), bi::read_write, perms);
|
||||
shm.truncate(newSize);
|
||||
string keyName = ShmKeys::keyToName(newKey);
|
||||
bi::permissions perms;
|
||||
perms.set_unrestricted();
|
||||
bi::shared_memory_object shm(bi::create_only, keyName.c_str(), bi::read_write, perms);
|
||||
shm.truncate(newSize);
|
||||
|
||||
bi::mapped_region region(shm, bi::read_write);
|
||||
bi::mapped_region region(shm, bi::read_write);
|
||||
|
||||
//Copy old data into new region
|
||||
memcpy(region.get_address(), fMapreg.get_address(), fSize);
|
||||
//clear new region
|
||||
//make some versions of gcc happier...
|
||||
memset(reinterpret_cast<void*>(reinterpret_cast<ptrdiff_t>(region.get_address()) + fSize), 0, newSize - fSize);
|
||||
// Copy old data into new region
|
||||
memcpy(region.get_address(), fMapreg.get_address(), fSize);
|
||||
// clear new region
|
||||
// make some versions of gcc happier...
|
||||
memset(reinterpret_cast<void*>(reinterpret_cast<ptrdiff_t>(region.get_address()) + fSize), 0,
|
||||
newSize - fSize);
|
||||
|
||||
fShmobj.swap(shm);
|
||||
fMapreg.swap(region);
|
||||
fShmobj.swap(shm);
|
||||
fMapreg.swap(region);
|
||||
|
||||
if (!oldName.empty()) bi::shared_memory_object::remove(oldName.c_str());
|
||||
if (!oldName.empty())
|
||||
bi::shared_memory_object::remove(oldName.c_str());
|
||||
|
||||
fKey = newKey;
|
||||
fSize = newSize;
|
||||
fKey = newKey;
|
||||
fSize = newSize;
|
||||
|
||||
if (fReadOnly)
|
||||
{
|
||||
bi::mapped_region ro_region(fShmobj, bi::read_only);
|
||||
fMapreg.swap(ro_region);
|
||||
}
|
||||
if (fReadOnly)
|
||||
{
|
||||
bi::mapped_region ro_region(fShmobj, bi::read_only);
|
||||
fMapreg.swap(ro_region);
|
||||
}
|
||||
|
||||
return 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int BRMShmImpl::clear(unsigned newKey, off_t newSize)
|
||||
{
|
||||
idbassert(newKey != fKey);
|
||||
idbassert(newKey != fKey);
|
||||
|
||||
string oldName = fShmobj.get_name();
|
||||
string oldName = fShmobj.get_name();
|
||||
|
||||
string keyName = ShmKeys::keyToName(newKey);
|
||||
bi::permissions perms;
|
||||
perms.set_unrestricted();
|
||||
bi::shared_memory_object shm(bi::create_only, keyName.c_str(), bi::read_write, perms);
|
||||
shm.truncate(newSize);
|
||||
string keyName = ShmKeys::keyToName(newKey);
|
||||
bi::permissions perms;
|
||||
perms.set_unrestricted();
|
||||
bi::shared_memory_object shm(bi::create_only, keyName.c_str(), bi::read_write, perms);
|
||||
shm.truncate(newSize);
|
||||
|
||||
bi::mapped_region region(shm, bi::read_write);
|
||||
bi::mapped_region region(shm, bi::read_write);
|
||||
|
||||
//clear new region
|
||||
memset(region.get_address(), 0, newSize);
|
||||
// clear new region
|
||||
memset(region.get_address(), 0, newSize);
|
||||
|
||||
fShmobj.swap(shm);
|
||||
fMapreg.swap(region);
|
||||
fShmobj.swap(shm);
|
||||
fMapreg.swap(region);
|
||||
|
||||
if (!oldName.empty()) bi::shared_memory_object::remove(oldName.c_str());
|
||||
if (!oldName.empty())
|
||||
bi::shared_memory_object::remove(oldName.c_str());
|
||||
|
||||
fKey = newKey;
|
||||
fSize = newSize;
|
||||
fKey = newKey;
|
||||
fSize = newSize;
|
||||
|
||||
if (fReadOnly)
|
||||
{
|
||||
bi::mapped_region ro_region(fShmobj, bi::read_only);
|
||||
fMapreg.swap(ro_region);
|
||||
}
|
||||
if (fReadOnly)
|
||||
{
|
||||
bi::mapped_region ro_region(fShmobj, bi::read_only);
|
||||
fMapreg.swap(ro_region);
|
||||
}
|
||||
|
||||
return 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
void BRMShmImpl::setReadOnly()
|
||||
{
|
||||
if (fReadOnly) return;
|
||||
if (fReadOnly)
|
||||
return;
|
||||
|
||||
bi::mapped_region ro_region(fShmobj, bi::read_only);
|
||||
fMapreg.swap(ro_region);
|
||||
bi::mapped_region ro_region(fShmobj, bi::read_only);
|
||||
fMapreg.swap(ro_region);
|
||||
|
||||
fReadOnly = true;
|
||||
fReadOnly = true;
|
||||
}
|
||||
|
||||
void BRMShmImpl::swap(BRMShmImpl& rhs)
|
||||
{
|
||||
fShmobj.swap(rhs.fShmobj);
|
||||
fMapreg.swap(rhs.fMapreg);
|
||||
std::swap(fKey, rhs.fKey);
|
||||
std::swap(fSize, rhs.fSize);
|
||||
std::swap(fReadOnly, rhs.fReadOnly);
|
||||
fShmobj.swap(rhs.fShmobj);
|
||||
fMapreg.swap(rhs.fMapreg);
|
||||
std::swap(fKey, rhs.fKey);
|
||||
std::swap(fSize, rhs.fSize);
|
||||
std::swap(fReadOnly, rhs.fReadOnly);
|
||||
}
|
||||
|
||||
void BRMShmImpl::destroy()
|
||||
{
|
||||
string oldName = fShmobj.get_name();
|
||||
string oldName = fShmobj.get_name();
|
||||
|
||||
if (!oldName.empty()) bi::shared_memory_object::remove(oldName.c_str());
|
||||
if (!oldName.empty())
|
||||
bi::shared_memory_object::remove(oldName.c_str());
|
||||
}
|
||||
|
||||
} //namespace
|
||||
} // namespace BRM
|
||||
|
||||
// vim:ts=4 sw=4:
|
||||
|
||||
|
||||
Reference in New Issue
Block a user