You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-06-13 16:01:32 +03:00
Reformat all code to coding standard
This commit is contained in:
Andrew Hutchings
@ -34,7 +34,7 @@ using namespace std;
|
||||
#include <boost/filesystem.hpp>
|
||||
#include <boost/unordered_map.hpp>
|
||||
using namespace boost;
|
||||
namespace fs=boost::filesystem;
|
||||
namespace fs = boost::filesystem;
|
||||
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
@ -76,549 +76,609 @@ boost::mutex Config::fWriteXmlLock;
|
||||
|
||||
Config* Config::makeConfig(const string& cf)
|
||||
{
|
||||
return makeConfig(cf.c_str());
|
||||
return makeConfig(cf.c_str());
|
||||
}
|
||||
|
||||
Config* Config::makeConfig(const char* cf)
|
||||
{
|
||||
mutex::scoped_lock lk(fInstanceMapMutex);
|
||||
mutex::scoped_lock lk(fInstanceMapMutex);
|
||||
|
||||
static string installDir;
|
||||
|
||||
static string installDir;
|
||||
if (installDir.empty())
|
||||
installDir = startup::StartUp::installDir();
|
||||
|
||||
if (cf == 0)
|
||||
{
|
||||
if (cf == 0)
|
||||
{
|
||||
#ifdef _MSC_VER
|
||||
string cfStr = IDBreadRegistry("ConfigFile");
|
||||
if (!cfStr.empty())
|
||||
cf = cfStr.c_str();
|
||||
string cfStr = IDBreadRegistry("ConfigFile");
|
||||
|
||||
if (!cfStr.empty())
|
||||
cf = cfStr.c_str();
|
||||
|
||||
#else
|
||||
cf = getenv("CALPONT_CONFIG_FILE");
|
||||
cf = getenv("CALPONT_CONFIG_FILE");
|
||||
#endif
|
||||
if (cf == 0 || *cf == 0)
|
||||
{
|
||||
|
||||
if (cf == 0 || *cf == 0)
|
||||
{
|
||||
static string defaultFilePath;
|
||||
|
||||
if (defaultFilePath.empty())
|
||||
{
|
||||
fs::path configFilePath;
|
||||
configFilePath = fs::path(installDir) / fs::path("etc") / defaultCalpontConfigFile;
|
||||
defaultFilePath = configFilePath.string();
|
||||
}
|
||||
|
||||
if (fInstanceMap.find(defaultFilePath) == fInstanceMap.end())
|
||||
{
|
||||
Config* instance = new Config(defaultFilePath, installDir);
|
||||
fInstanceMap[defaultFilePath] = instance;
|
||||
}
|
||||
{
|
||||
Config* instance = new Config(defaultFilePath, installDir);
|
||||
fInstanceMap[defaultFilePath] = instance;
|
||||
}
|
||||
|
||||
return fInstanceMap[defaultFilePath];
|
||||
}
|
||||
}
|
||||
string configFile(cf);
|
||||
}
|
||||
}
|
||||
|
||||
if (fInstanceMap.find(configFile) == fInstanceMap.end())
|
||||
{
|
||||
Config* instance = new Config(configFile, installDir);
|
||||
fInstanceMap[configFile] = instance;
|
||||
}
|
||||
string configFile(cf);
|
||||
|
||||
return fInstanceMap[configFile];
|
||||
if (fInstanceMap.find(configFile) == fInstanceMap.end())
|
||||
{
|
||||
Config* instance = new Config(configFile, installDir);
|
||||
fInstanceMap[configFile] = instance;
|
||||
}
|
||||
|
||||
return fInstanceMap[configFile];
|
||||
}
|
||||
|
||||
Config::Config(const string& configFile, const string& installDir) :
|
||||
fDoc(0), fConfigFile(configFile), fMtime(0), fInstallDir(installDir), fParser(fInstallDir)
|
||||
fDoc(0), fConfigFile(configFile), fMtime(0), fInstallDir(installDir), fParser(fInstallDir)
|
||||
{
|
||||
for ( int i = 0 ; i < 20 ; i++ )
|
||||
{
|
||||
if (access(fConfigFile.c_str(), R_OK) != 0) {
|
||||
if ( i >= 15 )
|
||||
throw runtime_error("Config::Config: error accessing config file " + fConfigFile);
|
||||
sleep (1);
|
||||
}
|
||||
else
|
||||
break;
|
||||
}
|
||||
for ( int i = 0 ; i < 20 ; i++ )
|
||||
{
|
||||
if (access(fConfigFile.c_str(), R_OK) != 0)
|
||||
{
|
||||
if ( i >= 15 )
|
||||
throw runtime_error("Config::Config: error accessing config file " + fConfigFile);
|
||||
|
||||
struct stat statbuf;
|
||||
if (stat(configFile.c_str(), &statbuf) == 0)
|
||||
fMtime = statbuf.st_mtime;
|
||||
sleep (1);
|
||||
}
|
||||
else
|
||||
break;
|
||||
}
|
||||
|
||||
parseDoc();
|
||||
struct stat statbuf;
|
||||
|
||||
if (stat(configFile.c_str(), &statbuf) == 0)
|
||||
fMtime = statbuf.st_mtime;
|
||||
|
||||
parseDoc();
|
||||
}
|
||||
|
||||
Config::~Config()
|
||||
{
|
||||
if (fDoc != 0)
|
||||
closeConfig();
|
||||
if (fDoc != 0)
|
||||
closeConfig();
|
||||
}
|
||||
|
||||
void Config::parseDoc(void)
|
||||
{
|
||||
struct flock fl;
|
||||
int fd;
|
||||
struct flock fl;
|
||||
int fd;
|
||||
|
||||
memset(&fl, 0, sizeof(fl));
|
||||
fl.l_type = F_RDLCK; // read lock
|
||||
fl.l_whence = SEEK_SET;
|
||||
fl.l_start = 0;
|
||||
fl.l_len = 0; //lock whole file
|
||||
memset(&fl, 0, sizeof(fl));
|
||||
fl.l_type = F_RDLCK; // read lock
|
||||
fl.l_whence = SEEK_SET;
|
||||
fl.l_start = 0;
|
||||
fl.l_len = 0; //lock whole file
|
||||
|
||||
// lock file if exist
|
||||
if ((fd = open(fConfigFile.c_str(), O_RDONLY)) >= 0)
|
||||
{
|
||||
if (fcntl(fd, F_SETLKW, &fl) != 0)
|
||||
{
|
||||
ostringstream oss;
|
||||
oss << "Config::parseDoc: error locking file " <<
|
||||
fConfigFile <<
|
||||
": " <<
|
||||
strerror(errno) <<
|
||||
", proceding anyway.";
|
||||
cerr << oss.str() << endl;
|
||||
}
|
||||
// lock file if exist
|
||||
if ((fd = open(fConfigFile.c_str(), O_RDONLY)) >= 0)
|
||||
{
|
||||
if (fcntl(fd, F_SETLKW, &fl) != 0)
|
||||
{
|
||||
ostringstream oss;
|
||||
oss << "Config::parseDoc: error locking file " <<
|
||||
fConfigFile <<
|
||||
": " <<
|
||||
strerror(errno) <<
|
||||
", proceding anyway.";
|
||||
cerr << oss.str() << endl;
|
||||
}
|
||||
|
||||
fXmlLock.lock();
|
||||
fDoc = xmlParseFile(fConfigFile.c_str());
|
||||
fXmlLock.lock();
|
||||
fDoc = xmlParseFile(fConfigFile.c_str());
|
||||
fXmlLock.unlock();
|
||||
|
||||
fl.l_type = F_UNLCK; //unlock
|
||||
fcntl(fd, F_SETLK, &fl);
|
||||
fl.l_type = F_UNLCK; //unlock
|
||||
fcntl(fd, F_SETLK, &fl);
|
||||
|
||||
close(fd);
|
||||
}
|
||||
else
|
||||
{
|
||||
ostringstream oss;
|
||||
oss << "Config::parseDoc: error opening file " <<
|
||||
fConfigFile <<
|
||||
": " <<
|
||||
strerror(errno);
|
||||
throw runtime_error(oss.str());
|
||||
}
|
||||
close(fd);
|
||||
}
|
||||
else
|
||||
{
|
||||
ostringstream oss;
|
||||
oss << "Config::parseDoc: error opening file " <<
|
||||
fConfigFile <<
|
||||
": " <<
|
||||
strerror(errno);
|
||||
throw runtime_error(oss.str());
|
||||
}
|
||||
|
||||
|
||||
if (fDoc == 0 ) {
|
||||
throw runtime_error("Config::parseDoc: error parsing config file " + fConfigFile);
|
||||
}
|
||||
if (fDoc == 0 )
|
||||
{
|
||||
throw runtime_error("Config::parseDoc: error parsing config file " + fConfigFile);
|
||||
}
|
||||
|
||||
xmlNodePtr cur = xmlDocGetRootElement(fDoc);
|
||||
xmlNodePtr cur = xmlDocGetRootElement(fDoc);
|
||||
|
||||
if (cur == NULL)
|
||||
{
|
||||
xmlFreeDoc(fDoc);
|
||||
fDoc = 0;
|
||||
throw runtime_error("Config::parseDoc: error parsing config file " + fConfigFile);
|
||||
}
|
||||
if (cur == NULL)
|
||||
{
|
||||
xmlFreeDoc(fDoc);
|
||||
fDoc = 0;
|
||||
throw runtime_error("Config::parseDoc: error parsing config file " + fConfigFile);
|
||||
}
|
||||
|
||||
if (xmlStrcmp(cur->name, (const xmlChar *)"Columnstore"))
|
||||
{
|
||||
xmlFreeDoc(fDoc);
|
||||
fDoc = 0;
|
||||
throw runtime_error("Config::parseDoc: error parsing config file " + fConfigFile);
|
||||
}
|
||||
if (xmlStrcmp(cur->name, (const xmlChar*)"Columnstore"))
|
||||
{
|
||||
xmlFreeDoc(fDoc);
|
||||
fDoc = 0;
|
||||
throw runtime_error("Config::parseDoc: error parsing config file " + fConfigFile);
|
||||
}
|
||||
|
||||
return;
|
||||
return;
|
||||
}
|
||||
|
||||
void Config::closeConfig(void)
|
||||
{
|
||||
xmlFreeDoc(fDoc);
|
||||
fDoc = 0;
|
||||
xmlFreeDoc(fDoc);
|
||||
fDoc = 0;
|
||||
}
|
||||
|
||||
const string Config::getConfig(const string& section, const string& name)
|
||||
{
|
||||
mutex::scoped_lock lk(fLock);
|
||||
mutex::scoped_lock lk(fLock);
|
||||
|
||||
if (section.length() == 0 || name.length() == 0)
|
||||
throw invalid_argument("Config::getConfig: both section and name must have a length");
|
||||
if (section.length() == 0 || name.length() == 0)
|
||||
throw invalid_argument("Config::getConfig: both section and name must have a length");
|
||||
|
||||
if (fDoc == 0){
|
||||
throw runtime_error("Config::getConfig: no XML document!");
|
||||
}
|
||||
if (fDoc == 0)
|
||||
{
|
||||
throw runtime_error("Config::getConfig: no XML document!");
|
||||
}
|
||||
|
||||
struct stat statbuf;
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
struct stat statbuf;
|
||||
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
|
||||
return fParser.getConfig(fDoc, section, name);
|
||||
}
|
||||
|
||||
void Config::getConfig(const string& section, const string& name, vector<string>& values)
|
||||
{
|
||||
mutex::scoped_lock lk(fLock);
|
||||
mutex::scoped_lock lk(fLock);
|
||||
|
||||
if (section.length() == 0)
|
||||
throw invalid_argument("Config::getConfig: section must have a length");
|
||||
if (section.length() == 0)
|
||||
throw invalid_argument("Config::getConfig: section must have a length");
|
||||
|
||||
if (fDoc == 0)
|
||||
throw runtime_error("Config::getConfig: no XML document!");
|
||||
if (fDoc == 0)
|
||||
throw runtime_error("Config::getConfig: no XML document!");
|
||||
|
||||
struct stat statbuf;
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
struct stat statbuf;
|
||||
|
||||
fParser.getConfig(fDoc, section, name, values);
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
|
||||
fParser.getConfig(fDoc, section, name, values);
|
||||
}
|
||||
|
||||
void Config::setConfig(const string& section, const string& name, const string& value)
|
||||
{
|
||||
mutex::scoped_lock lk(fLock);
|
||||
mutex::scoped_lock lk(fLock);
|
||||
|
||||
if (section.length() == 0 || name.length() == 0 )
|
||||
throw invalid_argument("Config::setConfig: all of section and name must have a length");
|
||||
if (section.length() == 0 || name.length() == 0 )
|
||||
throw invalid_argument("Config::setConfig: all of section and name must have a length");
|
||||
|
||||
if (fDoc == 0) {
|
||||
throw runtime_error("Config::setConfig: no XML document!");
|
||||
}
|
||||
if (fDoc == 0)
|
||||
{
|
||||
throw runtime_error("Config::setConfig: no XML document!");
|
||||
}
|
||||
|
||||
struct stat statbuf;
|
||||
memset(&statbuf, 0, sizeof(statbuf));
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
struct stat statbuf;
|
||||
|
||||
fParser.setConfig(fDoc, section, name, value);
|
||||
return;
|
||||
memset(&statbuf, 0, sizeof(statbuf));
|
||||
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
|
||||
fParser.setConfig(fDoc, section, name, value);
|
||||
return;
|
||||
}
|
||||
|
||||
void Config::delConfig(const string& section, const string& name)
|
||||
{
|
||||
mutex::scoped_lock lk(fLock);
|
||||
mutex::scoped_lock lk(fLock);
|
||||
|
||||
if (section.length() == 0 || name.length() == 0)
|
||||
throw invalid_argument("Config::delConfig: both section and name must have a length");
|
||||
if (section.length() == 0 || name.length() == 0)
|
||||
throw invalid_argument("Config::delConfig: both section and name must have a length");
|
||||
|
||||
if (fDoc == 0){
|
||||
throw runtime_error("Config::delConfig: no XML document!");
|
||||
}
|
||||
if (fDoc == 0)
|
||||
{
|
||||
throw runtime_error("Config::delConfig: no XML document!");
|
||||
}
|
||||
|
||||
struct stat statbuf;
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
struct stat statbuf;
|
||||
|
||||
fParser.delConfig(fDoc, section, name);
|
||||
return;
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
|
||||
fParser.delConfig(fDoc, section, name);
|
||||
return;
|
||||
}
|
||||
|
||||
void Config::writeConfig(const string& configFile) const
|
||||
{
|
||||
mutex::scoped_lock lk(fLock);
|
||||
FILE *fi;
|
||||
if (fDoc == 0)
|
||||
throw runtime_error("Config::writeConfig: no XML document!");
|
||||
mutex::scoped_lock lk(fLock);
|
||||
FILE* fi;
|
||||
|
||||
if (fDoc == 0)
|
||||
throw runtime_error("Config::writeConfig: no XML document!");
|
||||
|
||||
#ifdef _MSC_VER
|
||||
fs::path configFilePth(configFile);
|
||||
fs::path outFilePth(configFilePth);
|
||||
outFilePth.replace_extension("temp");
|
||||
if ((fi = fopen(outFilePth.string().c_str(), "wt")) == NULL)
|
||||
throw runtime_error("Config::writeConfig: error opening config file for write " + outFilePth.string());
|
||||
int rc = -1;
|
||||
rc = xmlDocDump(fi, fDoc);
|
||||
if (rc < 0) {
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + outFilePth.string());
|
||||
}
|
||||
fclose(fi);
|
||||
if (fs::exists(configFilePth))
|
||||
fs::remove(configFilePth);
|
||||
fs::rename(outFilePth, configFilePth);
|
||||
fs::path configFilePth(configFile);
|
||||
fs::path outFilePth(configFilePth);
|
||||
outFilePth.replace_extension("temp");
|
||||
|
||||
if ((fi = fopen(outFilePth.string().c_str(), "wt")) == NULL)
|
||||
throw runtime_error("Config::writeConfig: error opening config file for write " + outFilePth.string());
|
||||
|
||||
int rc = -1;
|
||||
rc = xmlDocDump(fi, fDoc);
|
||||
|
||||
if (rc < 0)
|
||||
{
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + outFilePth.string());
|
||||
}
|
||||
|
||||
fclose(fi);
|
||||
|
||||
if (fs::exists(configFilePth))
|
||||
fs::remove(configFilePth);
|
||||
|
||||
fs::rename(outFilePth, configFilePth);
|
||||
#else
|
||||
|
||||
const fs::path defaultCalpontConfigFileTemp("Columnstore.xml.temp");
|
||||
const fs::path saveCalpontConfigFileTemp("Columnstore.xml.columnstoreSave");
|
||||
const fs::path tmpCalpontConfigFileTemp("Columnstore.xml.temp1");
|
||||
const fs::path defaultCalpontConfigFileTemp("Columnstore.xml.temp");
|
||||
const fs::path saveCalpontConfigFileTemp("Columnstore.xml.columnstoreSave");
|
||||
const fs::path tmpCalpontConfigFileTemp("Columnstore.xml.temp1");
|
||||
|
||||
fs::path etcdir = fs::path(fInstallDir) / fs::path("etc");
|
||||
fs::path etcdir = fs::path(fInstallDir) / fs::path("etc");
|
||||
|
||||
fs::path dcf = etcdir / fs::path(defaultCalpontConfigFile);
|
||||
fs::path dcft = etcdir / fs::path(defaultCalpontConfigFileTemp);
|
||||
fs::path scft = etcdir / fs::path(saveCalpontConfigFileTemp);
|
||||
fs::path tcft = etcdir / fs::path(tmpCalpontConfigFileTemp);
|
||||
//perform a temp write first if Columnstore.xml file to prevent possible corruption
|
||||
if ( configFile == dcf ) {
|
||||
fs::path dcf = etcdir / fs::path(defaultCalpontConfigFile);
|
||||
fs::path dcft = etcdir / fs::path(defaultCalpontConfigFileTemp);
|
||||
fs::path scft = etcdir / fs::path(saveCalpontConfigFileTemp);
|
||||
fs::path tcft = etcdir / fs::path(tmpCalpontConfigFileTemp);
|
||||
|
||||
if (exists(dcft)) fs::remove(dcft);
|
||||
if ((fi = fopen(dcft.string().c_str(), "w+")) == NULL)
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + configFile);
|
||||
//perform a temp write first if Columnstore.xml file to prevent possible corruption
|
||||
if ( configFile == dcf )
|
||||
{
|
||||
|
||||
int rc;
|
||||
rc = xmlDocDump(fi, fDoc);
|
||||
if ( rc < 0) {
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + configFile);
|
||||
//cout << "xmlDocDump " << rc << " " << errno << endl;
|
||||
}
|
||||
if (exists(dcft)) fs::remove(dcft);
|
||||
|
||||
fclose(fi);
|
||||
if ((fi = fopen(dcft.string().c_str(), "w+")) == NULL)
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + configFile);
|
||||
|
||||
//check temp file
|
||||
try {
|
||||
Config* c1 = makeConfig(dcft.string().c_str());
|
||||
int rc;
|
||||
rc = xmlDocDump(fi, fDoc);
|
||||
|
||||
string value;
|
||||
value = c1->getConfig("SystemConfig", "SystemName");
|
||||
if ( rc < 0)
|
||||
{
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + configFile);
|
||||
//cout << "xmlDocDump " << rc << " " << errno << endl;
|
||||
}
|
||||
|
||||
//good read, save copy, copy temp file tp tmp then to Columnstore.xml
|
||||
//move to /tmp to get around a 'same file error' in mv command
|
||||
try {
|
||||
if (exists(scft)) fs::remove(scft);
|
||||
} catch (fs::filesystem_error&) { }
|
||||
fs::copy_file(dcf, scft, fs::copy_option::overwrite_if_exists);
|
||||
try {
|
||||
fs::permissions(scft, fs::add_perms | fs::owner_read | fs::owner_write |
|
||||
fs::group_read | fs::group_write |
|
||||
fs::others_read | fs::others_write);
|
||||
} catch (fs::filesystem_error&) { }
|
||||
fclose(fi);
|
||||
|
||||
if (exists(tcft)) fs::remove(tcft);
|
||||
fs::rename(dcft, tcft);
|
||||
//check temp file
|
||||
try
|
||||
{
|
||||
Config* c1 = makeConfig(dcft.string().c_str());
|
||||
|
||||
if (exists(dcf)) fs::remove(dcf);
|
||||
fs::rename(tcft, dcf);
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + configFile);
|
||||
}
|
||||
}
|
||||
else
|
||||
{ // non Columnstore.xml, perform update
|
||||
if ((fi = fopen(configFile.c_str(), "w")) == NULL)
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + configFile);
|
||||
string value;
|
||||
value = c1->getConfig("SystemConfig", "SystemName");
|
||||
|
||||
xmlDocDump(fi, fDoc);
|
||||
//good read, save copy, copy temp file tp tmp then to Columnstore.xml
|
||||
//move to /tmp to get around a 'same file error' in mv command
|
||||
try
|
||||
{
|
||||
if (exists(scft)) fs::remove(scft);
|
||||
}
|
||||
catch (fs::filesystem_error&) { }
|
||||
|
||||
fs::copy_file(dcf, scft, fs::copy_option::overwrite_if_exists);
|
||||
|
||||
try
|
||||
{
|
||||
fs::permissions(scft, fs::add_perms | fs::owner_read | fs::owner_write |
|
||||
fs::group_read | fs::group_write |
|
||||
fs::others_read | fs::others_write);
|
||||
}
|
||||
catch (fs::filesystem_error&) { }
|
||||
|
||||
if (exists(tcft)) fs::remove(tcft);
|
||||
|
||||
fs::rename(dcft, tcft);
|
||||
|
||||
if (exists(dcf)) fs::remove(dcf);
|
||||
|
||||
fs::rename(tcft, dcf);
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + configFile);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
// non Columnstore.xml, perform update
|
||||
if ((fi = fopen(configFile.c_str(), "w")) == NULL)
|
||||
throw runtime_error("Config::writeConfig: error writing config file " + configFile);
|
||||
|
||||
xmlDocDump(fi, fDoc);
|
||||
|
||||
fclose(fi);
|
||||
}
|
||||
|
||||
fclose(fi);
|
||||
}
|
||||
#endif
|
||||
return;
|
||||
return;
|
||||
}
|
||||
|
||||
void Config::write(void) const
|
||||
{
|
||||
mutex::scoped_lock lk(fWriteXmlLock);
|
||||
mutex::scoped_lock lk(fWriteXmlLock);
|
||||
#ifdef _MSC_VER
|
||||
writeConfig(fConfigFile);
|
||||
writeConfig(fConfigFile);
|
||||
#else
|
||||
write(fConfigFile);
|
||||
write(fConfigFile);
|
||||
#endif
|
||||
}
|
||||
|
||||
void Config::write(const string& configFile) const
|
||||
{
|
||||
struct flock fl;
|
||||
int fd;
|
||||
struct flock fl;
|
||||
int fd;
|
||||
|
||||
fl.l_type = F_WRLCK; // write lock
|
||||
fl.l_whence = SEEK_SET;
|
||||
fl.l_start = 0;
|
||||
fl.l_len = 0;
|
||||
fl.l_pid = getpid();
|
||||
fl.l_type = F_WRLCK; // write lock
|
||||
fl.l_whence = SEEK_SET;
|
||||
fl.l_start = 0;
|
||||
fl.l_len = 0;
|
||||
fl.l_pid = getpid();
|
||||
|
||||
// lock file if it exists
|
||||
if ((fd = open(configFile.c_str(), O_WRONLY)) >= 0)
|
||||
{
|
||||
if (fcntl(fd, F_SETLKW, &fl) == -1)
|
||||
throw runtime_error("Config::write: file lock error " + configFile);
|
||||
// lock file if it exists
|
||||
if ((fd = open(configFile.c_str(), O_WRONLY)) >= 0)
|
||||
{
|
||||
if (fcntl(fd, F_SETLKW, &fl) == -1)
|
||||
throw runtime_error("Config::write: file lock error " + configFile);
|
||||
|
||||
|
||||
try
|
||||
{
|
||||
writeConfig(configFile);
|
||||
}
|
||||
catch(...)
|
||||
catch (...)
|
||||
{
|
||||
fl.l_type = F_UNLCK; //unlock
|
||||
|
||||
if (fcntl(fd, F_SETLK, &fl) == -1)
|
||||
throw runtime_error("Config::write: file unlock error after exception in writeConfig " + configFile);
|
||||
|
||||
throw;
|
||||
}
|
||||
|
||||
fl.l_type = F_UNLCK; //unlock
|
||||
if (fcntl(fd, F_SETLK, &fl) == -1)
|
||||
throw runtime_error("Config::write: file unlock error " + configFile);
|
||||
fl.l_type = F_UNLCK; //unlock
|
||||
|
||||
close(fd);
|
||||
}
|
||||
else
|
||||
{
|
||||
writeConfig(configFile);
|
||||
}
|
||||
if (fcntl(fd, F_SETLK, &fl) == -1)
|
||||
throw runtime_error("Config::write: file unlock error " + configFile);
|
||||
|
||||
close(fd);
|
||||
}
|
||||
else
|
||||
{
|
||||
writeConfig(configFile);
|
||||
}
|
||||
}
|
||||
|
||||
void Config::writeConfigFile(messageqcpp::ByteStream msg) const
|
||||
{
|
||||
struct flock fl;
|
||||
int fd;
|
||||
struct flock fl;
|
||||
int fd;
|
||||
|
||||
//get config file name being udated
|
||||
string fileName;
|
||||
msg >> fileName;
|
||||
//get config file name being udated
|
||||
string fileName;
|
||||
msg >> fileName;
|
||||
|
||||
fl.l_type = F_WRLCK; // write lock
|
||||
fl.l_whence = SEEK_SET;
|
||||
fl.l_start = 0;
|
||||
fl.l_len = 0;
|
||||
fl.l_pid = getpid();
|
||||
fl.l_type = F_WRLCK; // write lock
|
||||
fl.l_whence = SEEK_SET;
|
||||
fl.l_start = 0;
|
||||
fl.l_len = 0;
|
||||
fl.l_pid = getpid();
|
||||
|
||||
// lock file if it exists
|
||||
if ((fd = open(fileName.c_str(), O_WRONLY)) >= 0)
|
||||
{
|
||||
if (fcntl(fd, F_SETLKW, &fl) == -1)
|
||||
throw runtime_error("Config::write: file lock error " + fileName);
|
||||
// lock file if it exists
|
||||
if ((fd = open(fileName.c_str(), O_WRONLY)) >= 0)
|
||||
{
|
||||
if (fcntl(fd, F_SETLKW, &fl) == -1)
|
||||
throw runtime_error("Config::write: file lock error " + fileName);
|
||||
|
||||
ofstream out(fileName.c_str());
|
||||
out << msg;
|
||||
ofstream out(fileName.c_str());
|
||||
out << msg;
|
||||
|
||||
fl.l_type = F_UNLCK; //unlock
|
||||
if (fcntl(fd, F_SETLK, &fl) == -1)
|
||||
throw runtime_error("Config::write: file unlock error " + fileName);
|
||||
fl.l_type = F_UNLCK; //unlock
|
||||
|
||||
close(fd);
|
||||
}
|
||||
else
|
||||
{
|
||||
ofstream out(fileName.c_str());
|
||||
out << msg;
|
||||
}
|
||||
if (fcntl(fd, F_SETLK, &fl) == -1)
|
||||
throw runtime_error("Config::write: file unlock error " + fileName);
|
||||
|
||||
close(fd);
|
||||
}
|
||||
else
|
||||
{
|
||||
ofstream out(fileName.c_str());
|
||||
out << msg;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/* static */
|
||||
void Config::deleteInstanceMap()
|
||||
{
|
||||
mutex::scoped_lock lk(fInstanceMapMutex);
|
||||
for (Config::configMap_t::iterator iter = fInstanceMap.begin();
|
||||
iter != fInstanceMap.end(); ++iter)
|
||||
{
|
||||
Config* instance = iter->second;
|
||||
delete instance;
|
||||
}
|
||||
fInstanceMap.clear();
|
||||
mutex::scoped_lock lk(fInstanceMapMutex);
|
||||
|
||||
for (Config::configMap_t::iterator iter = fInstanceMap.begin();
|
||||
iter != fInstanceMap.end(); ++iter)
|
||||
{
|
||||
Config* instance = iter->second;
|
||||
delete instance;
|
||||
}
|
||||
|
||||
fInstanceMap.clear();
|
||||
}
|
||||
|
||||
/* static */
|
||||
int64_t Config::fromText(const std::string& text)
|
||||
{
|
||||
if (text.length() == 0) return 0;
|
||||
if (text.length() == 0) return 0;
|
||||
|
||||
int64_t val = 0;
|
||||
char* ctext = static_cast<char*>(alloca(text.length() + 1));
|
||||
strcpy(ctext, text.c_str());
|
||||
char* cptr;
|
||||
int64_t val = 0;
|
||||
char* ctext = static_cast<char*>(alloca(text.length() + 1));
|
||||
strcpy(ctext, text.c_str());
|
||||
char* cptr;
|
||||
|
||||
val = strtoll(ctext, &cptr, 0);
|
||||
val = strtoll(ctext, &cptr, 0);
|
||||
|
||||
switch (*cptr)
|
||||
{
|
||||
case 'T':
|
||||
case 't':
|
||||
val *= 1024;
|
||||
/* fallthru */
|
||||
case 'G':
|
||||
case 'g':
|
||||
val *= 1024;
|
||||
/* fallthru */
|
||||
case 'M':
|
||||
case 'm':
|
||||
val *= 1024;
|
||||
/* fallthru */
|
||||
case 'K':
|
||||
case 'k':
|
||||
val *= 1024;
|
||||
/* fallthru */
|
||||
case '\0':
|
||||
break;
|
||||
default:
|
||||
ostringstream oss;
|
||||
oss << "Invalid character '" << *cptr << "' found in numeric parameter '" << text <<
|
||||
"'. Since this will not do what you want it is fatal." << endl;
|
||||
throw runtime_error(oss.str());
|
||||
break;
|
||||
}
|
||||
switch (*cptr)
|
||||
{
|
||||
case 'T':
|
||||
case 't':
|
||||
val *= 1024;
|
||||
|
||||
return val;
|
||||
/* fallthru */
|
||||
case 'G':
|
||||
case 'g':
|
||||
val *= 1024;
|
||||
|
||||
/* fallthru */
|
||||
case 'M':
|
||||
case 'm':
|
||||
val *= 1024;
|
||||
|
||||
/* fallthru */
|
||||
case 'K':
|
||||
case 'k':
|
||||
val *= 1024;
|
||||
|
||||
/* fallthru */
|
||||
case '\0':
|
||||
break;
|
||||
|
||||
default:
|
||||
ostringstream oss;
|
||||
oss << "Invalid character '" << *cptr << "' found in numeric parameter '" << text <<
|
||||
"'. Since this will not do what you want it is fatal." << endl;
|
||||
throw runtime_error(oss.str());
|
||||
break;
|
||||
}
|
||||
|
||||
return val;
|
||||
}
|
||||
|
||||
time_t Config::getCurrentMTime()
|
||||
{
|
||||
mutex::scoped_lock lk(fLock);
|
||||
mutex::scoped_lock lk(fLock);
|
||||
|
||||
struct stat statbuf;
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
return statbuf.st_mtime;
|
||||
else
|
||||
return 0;
|
||||
struct stat statbuf;
|
||||
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
return statbuf.st_mtime;
|
||||
else
|
||||
return 0;
|
||||
}
|
||||
|
||||
const vector<string> Config::enumConfig()
|
||||
{
|
||||
mutex::scoped_lock lk(fLock);
|
||||
mutex::scoped_lock lk(fLock);
|
||||
|
||||
if (fDoc == 0){
|
||||
throw runtime_error("Config::getConfig: no XML document!");
|
||||
}
|
||||
if (fDoc == 0)
|
||||
{
|
||||
throw runtime_error("Config::getConfig: no XML document!");
|
||||
}
|
||||
|
||||
struct stat statbuf;
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
struct stat statbuf;
|
||||
|
||||
return fParser.enumConfig(fDoc);
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
|
||||
return fParser.enumConfig(fDoc);
|
||||
}
|
||||
|
||||
const vector<string> Config::enumSection(const string& section)
|
||||
{
|
||||
mutex::scoped_lock lk(fLock);
|
||||
mutex::scoped_lock lk(fLock);
|
||||
|
||||
if (fDoc == 0){
|
||||
throw runtime_error("Config::getConfig: no XML document!");
|
||||
}
|
||||
if (fDoc == 0)
|
||||
{
|
||||
throw runtime_error("Config::getConfig: no XML document!");
|
||||
}
|
||||
|
||||
struct stat statbuf;
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
struct stat statbuf;
|
||||
|
||||
return fParser.enumSection(fDoc, section);
|
||||
if (stat(fConfigFile.c_str(), &statbuf) == 0)
|
||||
{
|
||||
if (statbuf.st_mtime != fMtime)
|
||||
{
|
||||
closeConfig();
|
||||
fMtime = statbuf.st_mtime;
|
||||
parseDoc();
|
||||
}
|
||||
}
|
||||
|
||||
return fParser.enumSection(fDoc, section);
|
||||
}
|
||||
|
||||
} //namespace config
|
||||
|
||||
Reference in New Issue
Block a user