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
@ -37,18 +37,19 @@ using namespace idbdatafile;
|
||||
|
||||
namespace BRM
|
||||
{
|
||||
|
||||
TableLockServer::TableLockServer(SessionManagerServer* sm) : sms(sm)
|
||||
{
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
config::Config* config = config::Config::makeConfig();
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
config::Config* config = config::Config::makeConfig();
|
||||
|
||||
filename = config->getConfig("SystemConfig", "TableLockSaveFile");
|
||||
filename = config->getConfig("SystemConfig", "TableLockSaveFile");
|
||||
|
||||
if (filename == "")
|
||||
throw invalid_argument("TableLockServer: Need to define SystemConfig/TableLockSaveFile in config file"); // todo, replace this
|
||||
if (filename == "")
|
||||
throw invalid_argument(
|
||||
"TableLockServer: Need to define SystemConfig/TableLockSaveFile in config file"); // todo, replace
|
||||
// this
|
||||
|
||||
load();
|
||||
load();
|
||||
}
|
||||
|
||||
TableLockServer::~TableLockServer()
|
||||
@ -58,264 +59,259 @@ TableLockServer::~TableLockServer()
|
||||
// call with lock held
|
||||
void TableLockServer::save()
|
||||
{
|
||||
lit_t it;
|
||||
uint32_t count = locks.size();
|
||||
lit_t it;
|
||||
uint32_t count = locks.size();
|
||||
|
||||
const char* filename_p = filename.c_str();
|
||||
const char* filename_p = filename.c_str();
|
||||
|
||||
scoped_ptr<IDBDataFile> out(IDBDataFile::open(
|
||||
IDBPolicy::getType(filename_p, IDBPolicy::WRITEENG),
|
||||
filename_p, "wb", 0));
|
||||
scoped_ptr<IDBDataFile> out(
|
||||
IDBDataFile::open(IDBPolicy::getType(filename_p, IDBPolicy::WRITEENG), filename_p, "wb", 0));
|
||||
|
||||
if (!out)
|
||||
throw runtime_error("TableLockServer::save(): could not open save file");
|
||||
|
||||
out->write((char*)&count, 4);
|
||||
|
||||
for (it = locks.begin(); it != locks.end(); ++it)
|
||||
{
|
||||
if (!out)
|
||||
throw runtime_error("TableLockServer::save(): could not open save file");
|
||||
throw runtime_error("TableLockServer::save(): could not write save file");
|
||||
|
||||
out->write((char*) &count, 4);
|
||||
|
||||
for (it = locks.begin(); it != locks.end(); ++it)
|
||||
{
|
||||
if (!out)
|
||||
throw runtime_error("TableLockServer::save(): could not write save file");
|
||||
|
||||
it->second.serialize(out.get());
|
||||
}
|
||||
it->second.serialize(out.get());
|
||||
}
|
||||
}
|
||||
|
||||
// call with lock held
|
||||
void TableLockServer::load()
|
||||
{
|
||||
uint32_t size;
|
||||
uint32_t i = 0;
|
||||
TableLockInfo tli;
|
||||
uint32_t size;
|
||||
uint32_t i = 0;
|
||||
TableLockInfo tli;
|
||||
|
||||
/* Need to standardize the file error handling */
|
||||
const char* filename_p = filename.c_str();
|
||||
scoped_ptr<IDBDataFile> in(IDBDataFile::open(
|
||||
IDBPolicy::getType(filename_p, IDBPolicy::WRITEENG),
|
||||
filename_p, "rb", 0));
|
||||
/* Need to standardize the file error handling */
|
||||
const char* filename_p = filename.c_str();
|
||||
scoped_ptr<IDBDataFile> in(
|
||||
IDBDataFile::open(IDBPolicy::getType(filename_p, IDBPolicy::WRITEENG), filename_p, "rb", 0));
|
||||
|
||||
if (!in)
|
||||
if (!in)
|
||||
{
|
||||
ostringstream os;
|
||||
os << "TableLockServer::load(): could not open the save file" << filename;
|
||||
log(os.str(), logging::LOG_TYPE_DEBUG);
|
||||
return;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
in->read((char*)&size, 4);
|
||||
|
||||
for (i = 0; i < size; i++)
|
||||
{
|
||||
ostringstream os;
|
||||
os << "TableLockServer::load(): could not open the save file"
|
||||
<< filename;
|
||||
log(os.str(), logging::LOG_TYPE_DEBUG);
|
||||
return;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
in->read((char*) &size, 4);
|
||||
|
||||
for (i = 0; i < size; i++)
|
||||
{
|
||||
tli.deserialize(in.get());
|
||||
tli.id = sms->getUnique64(); // Need new #s...
|
||||
|
||||
if (tli.id == 0) // 0 is an error code
|
||||
tli.id = sms->getUnique64();
|
||||
|
||||
locks[tli.id] = tli;
|
||||
}
|
||||
}
|
||||
catch (std::exception& e)
|
||||
{
|
||||
ostringstream os;
|
||||
os << "TableLockServer::load(): could not load save file " << filename <<
|
||||
" loaded " << i << "/" << size << " entries\n";
|
||||
log(os.str(), logging::LOG_TYPE_DEBUG);
|
||||
throw;
|
||||
tli.deserialize(in.get());
|
||||
tli.id = sms->getUnique64(); // Need new #s...
|
||||
|
||||
if (tli.id == 0) // 0 is an error code
|
||||
tli.id = sms->getUnique64();
|
||||
|
||||
locks[tli.id] = tli;
|
||||
}
|
||||
}
|
||||
catch (std::exception& e)
|
||||
{
|
||||
ostringstream os;
|
||||
os << "TableLockServer::load(): could not load save file " << filename << " loaded " << i << "/" << size
|
||||
<< " entries\n";
|
||||
log(os.str(), logging::LOG_TYPE_DEBUG);
|
||||
throw;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
// throws on a failed save()
|
||||
uint64_t TableLockServer::lock(TableLockInfo* tli)
|
||||
{
|
||||
set<uint32_t> dbroots;
|
||||
lit_t it;
|
||||
uint32_t i;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
set<uint32_t> dbroots;
|
||||
lit_t it;
|
||||
uint32_t i;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
|
||||
for (i = 0; i < tli->dbrootList.size(); i++)
|
||||
dbroots.insert(tli->dbrootList[i]);
|
||||
for (i = 0; i < tli->dbrootList.size(); i++)
|
||||
dbroots.insert(tli->dbrootList[i]);
|
||||
|
||||
for (it = locks.begin(); it != locks.end(); ++it)
|
||||
for (it = locks.begin(); it != locks.end(); ++it)
|
||||
{
|
||||
if (it->second.overlaps(*tli, dbroots))
|
||||
{
|
||||
if (it->second.overlaps(*tli, dbroots))
|
||||
{
|
||||
tli->ownerName = it->second.ownerName;
|
||||
tli->ownerPID = it->second.ownerPID;
|
||||
tli->ownerSessionID = it->second.ownerSessionID;
|
||||
tli->ownerTxnID = it->second.ownerTxnID;
|
||||
return false;
|
||||
}
|
||||
tli->ownerName = it->second.ownerName;
|
||||
tli->ownerPID = it->second.ownerPID;
|
||||
tli->ownerSessionID = it->second.ownerSessionID;
|
||||
tli->ownerTxnID = it->second.ownerTxnID;
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
tli->id = sms->getUnique64();
|
||||
|
||||
if (tli->id == 0) // 0 is an error code
|
||||
tli->id = sms->getUnique64();
|
||||
|
||||
if (tli->id == 0) // 0 is an error code
|
||||
tli->id = sms->getUnique64();
|
||||
locks[tli->id] = *tli;
|
||||
|
||||
locks[tli->id] = *tli;
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
locks.erase(tli->id);
|
||||
throw;
|
||||
}
|
||||
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
locks.erase(tli->id);
|
||||
throw;
|
||||
}
|
||||
|
||||
return tli->id;
|
||||
return tli->id;
|
||||
}
|
||||
|
||||
bool TableLockServer::unlock(uint64_t id)
|
||||
{
|
||||
std::map<uint64_t, TableLockInfo>::iterator it;
|
||||
TableLockInfo tli;
|
||||
std::map<uint64_t, TableLockInfo>::iterator it;
|
||||
TableLockInfo tli;
|
||||
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
it = locks.find(id);
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
it = locks.find(id);
|
||||
|
||||
if (it != locks.end())
|
||||
if (it != locks.end())
|
||||
{
|
||||
tli = it->second;
|
||||
locks.erase(it);
|
||||
|
||||
try
|
||||
{
|
||||
tli = it->second;
|
||||
locks.erase(it);
|
||||
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
locks[tli.id] = tli;
|
||||
throw;
|
||||
}
|
||||
|
||||
return true;
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
locks[tli.id] = tli;
|
||||
throw;
|
||||
}
|
||||
|
||||
return false;
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
bool TableLockServer::changeState(uint64_t id, LockState state)
|
||||
{
|
||||
lit_t it;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
LockState old;
|
||||
lit_t it;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
LockState old;
|
||||
|
||||
it = locks.find(id);
|
||||
it = locks.find(id);
|
||||
|
||||
if (it == locks.end())
|
||||
return false;
|
||||
if (it == locks.end())
|
||||
return false;
|
||||
|
||||
old = it->second.state;
|
||||
it->second.state = state;
|
||||
old = it->second.state;
|
||||
it->second.state = state;
|
||||
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
it->second.state = old;
|
||||
throw;
|
||||
}
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
it->second.state = old;
|
||||
throw;
|
||||
}
|
||||
|
||||
return true;
|
||||
return true;
|
||||
}
|
||||
|
||||
bool TableLockServer::changeOwner(uint64_t id, const string& ownerName, uint32_t pid, int32_t session,
|
||||
int32_t txnID)
|
||||
{
|
||||
lit_t it;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
string oldName;
|
||||
uint32_t oldPID;
|
||||
int32_t oldSession;
|
||||
int32_t oldTxnID;
|
||||
lit_t it;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
string oldName;
|
||||
uint32_t oldPID;
|
||||
int32_t oldSession;
|
||||
int32_t oldTxnID;
|
||||
|
||||
it = locks.find(id);
|
||||
it = locks.find(id);
|
||||
|
||||
if (it == locks.end())
|
||||
return false;
|
||||
if (it == locks.end())
|
||||
return false;
|
||||
|
||||
oldName = it->second.ownerName;
|
||||
oldPID = it->second.ownerPID;
|
||||
oldSession = it->second.ownerSessionID;
|
||||
oldTxnID = it->second.ownerTxnID;
|
||||
it->second.ownerName = ownerName;
|
||||
it->second.ownerPID = pid;
|
||||
it->second.ownerSessionID = session;
|
||||
it->second.ownerTxnID = txnID;
|
||||
oldName = it->second.ownerName;
|
||||
oldPID = it->second.ownerPID;
|
||||
oldSession = it->second.ownerSessionID;
|
||||
oldTxnID = it->second.ownerTxnID;
|
||||
it->second.ownerName = ownerName;
|
||||
it->second.ownerPID = pid;
|
||||
it->second.ownerSessionID = session;
|
||||
it->second.ownerTxnID = txnID;
|
||||
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
it->second.ownerName = oldName;
|
||||
it->second.ownerPID = oldPID;
|
||||
it->second.ownerSessionID = oldSession;
|
||||
it->second.ownerTxnID = oldTxnID;
|
||||
throw;
|
||||
}
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
it->second.ownerName = oldName;
|
||||
it->second.ownerPID = oldPID;
|
||||
it->second.ownerSessionID = oldSession;
|
||||
it->second.ownerTxnID = oldTxnID;
|
||||
throw;
|
||||
}
|
||||
|
||||
return true;
|
||||
return true;
|
||||
}
|
||||
|
||||
vector<TableLockInfo> TableLockServer::getAllLocks() const
|
||||
{
|
||||
vector<TableLockInfo> ret;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
constlit_t it;
|
||||
vector<TableLockInfo> ret;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
constlit_t it;
|
||||
|
||||
for (it = locks.begin(); it != locks.end(); ++it)
|
||||
ret.push_back(it->second);
|
||||
for (it = locks.begin(); it != locks.end(); ++it)
|
||||
ret.push_back(it->second);
|
||||
|
||||
return ret;
|
||||
return ret;
|
||||
}
|
||||
|
||||
void TableLockServer::releaseAllLocks()
|
||||
{
|
||||
std::map<uint64_t, TableLockInfo> tmp;
|
||||
std::map<uint64_t, TableLockInfo> tmp;
|
||||
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
tmp.swap(locks);
|
||||
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
tmp.swap(locks);
|
||||
|
||||
try
|
||||
{
|
||||
save();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
tmp.swap(locks);
|
||||
throw;
|
||||
}
|
||||
throw;
|
||||
}
|
||||
}
|
||||
|
||||
bool TableLockServer::getLockInfo(uint64_t id, TableLockInfo* out) const
|
||||
{
|
||||
constlit_t it;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
constlit_t it;
|
||||
boost::mutex::scoped_lock lk(mutex);
|
||||
|
||||
it = locks.find(id);
|
||||
it = locks.find(id);
|
||||
|
||||
if (out == NULL)
|
||||
return (it != locks.end());
|
||||
if (out == NULL)
|
||||
return (it != locks.end());
|
||||
|
||||
if (it != locks.end())
|
||||
{
|
||||
*out = it->second;
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
if (it != locks.end())
|
||||
{
|
||||
*out = it->second;
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
} // namespace BRM
|
||||
|
||||
Reference in New Issue
Block a user