Replaced ACE_Configuration_Heap based config file handling with boost::property_tree::ini_parser

This commit is contained in:
leak
2014-06-22 15:45:54 +02:00
parent 7dd6f0f1d8
commit ca3327dbed
2 changed files with 53 additions and 104 deletions
+42 -78
View File
@@ -16,57 +16,41 @@
* with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <algorithm>
#include <mutex>
#include <boost/property_tree/ptree.hpp>
#include <boost/property_tree/ini_parser.hpp>
#include "Config.h"
#include "Errors.h"
// Defined here as it must not be exposed to end-users.
bool ConfigMgr::GetValueHelper(const char* name, ACE_TString &result)
{
GuardType guard(_configLock);
if (_config.get() == 0)
return false;
ACE_TString section_name;
ACE_Configuration_Section_Key section_key;
const ACE_Configuration_Section_Key &root_key = _config->root_section();
int i = 0;
while (_config->enumerate_sections(root_key, i, section_name) == 0)
{
_config->open_section(root_key, section_name.c_str(), 0, section_key);
if (_config->get_string_value(section_key, name, result) == 0)
return true;
++i;
}
return false;
}
using namespace boost::property_tree;
bool ConfigMgr::LoadInitial(char const* file)
{
ASSERT(file);
GuardType guard(_configLock);
std::lock_guard<std::mutex> lock(_configLock);
_filename = file;
_config.reset(new ACE_Configuration_Heap());
if (_config->open() == 0)
if (LoadData(_filename.c_str()))
return true;
_config.reset();
return false;
}
try
{
ptree temp;
boost::property_tree::ini_parser::read_ini(file, temp);
bool ConfigMgr::LoadMore(char const* file)
{
ASSERT(file);
ASSERT(_config);
GuardType guard(_configLock);
for (auto bla : temp)
{
_config = bla.second;
break;
}
}
catch (std::exception const& /*ex*/)
{
return false;
}
return LoadData(file);
return true;
}
bool ConfigMgr::Reload()
@@ -74,76 +58,56 @@ bool ConfigMgr::Reload()
return LoadInitial(_filename.c_str());
}
bool ConfigMgr::LoadData(char const* file)
{
ACE_Ini_ImpExp config_importer(*_config.get());
if (config_importer.import_config(file) == 0)
return true;
return false;
}
std::string ConfigMgr::GetStringDefault(const char* name, const std::string &def)
{
ACE_TString val;
return GetValueHelper(name, val) ? val.c_str() : def;
std::string value = _config.get<std::string>(ptree::path_type(name,'/'), def);
value.erase(std::remove(value.begin(), value.end(), '"'), value.end());
return value;
}
bool ConfigMgr::GetBoolDefault(const char* name, bool def)
{
ACE_TString val;
if (!GetValueHelper(name, val))
try
{
std::string val = _config.get<std::string>(name);
val.erase(std::remove(val.begin(), val.end(), '"'), val.end());
return (val == "true" || val == "TRUE" || val == "yes" || val == "YES" || val == "1");
}
catch (std::exception const& /*ex*/)
{
return def;
return (val == "true" || val == "TRUE" || val == "yes" || val == "YES" ||
val == "1");
}
}
int ConfigMgr::GetIntDefault(const char* name, int def)
{
ACE_TString val;
return GetValueHelper(name, val) ? atoi(val.c_str()) : def;
return _config.get<int>(name, def);
}
float ConfigMgr::GetFloatDefault(const char* name, float def)
{
ACE_TString val;
return GetValueHelper(name, val) ? (float)atof(val.c_str()) : def;
return _config.get<float>(name, def);
}
std::string const& ConfigMgr::GetFilename()
{
GuardType guard(_configLock);
std::lock_guard<std::mutex> lock(_configLock);
return _filename;
}
std::list<std::string> ConfigMgr::GetKeysByString(std::string const& name)
{
GuardType guard(_configLock);
std::lock_guard<std::mutex> lock(_configLock);
std::list<std::string> keys;
if (_config.get() == 0)
return keys;
ACE_TString section_name;
ACE_Configuration_Section_Key section_key;
const ACE_Configuration_Section_Key &root_key = _config->root_section();
int i = 0;
while (_config->enumerate_sections(root_key, i++, section_name) == 0)
for (const ptree::value_type& child : _config)
{
_config->open_section(root_key, section_name.c_str(), 0, section_key);
ACE_TString key_name;
ACE_Configuration::VALUETYPE type;
int j = 0;
while (_config->enumerate_values(section_key, j++, key_name, type) == 0)
if (child.first.compare(0, name.length(), name) == 0)
{
std::string temp = key_name.c_str();
if (!temp.find(name))
keys.push_back(temp);
keys.push_back(child.first);
}
}
+10 -25
View File
@@ -21,18 +21,11 @@
#include <string>
#include <list>
#include <ace/Singleton.h>
#include <ace/Configuration_Import_Export.h>
#include <ace/Thread_Mutex.h>
#include <AutoPtr.h>
typedef Trinity::AutoPtr<ACE_Configuration_Heap, ACE_Null_Mutex> Config;
#include <mutex>
#include <boost/property_tree/ptree.hpp>
class ConfigMgr
{
friend class ACE_Singleton<ConfigMgr, ACE_Null_Mutex>;
friend class ConfigLoader;
ConfigMgr() { }
~ConfigMgr() { }
@@ -40,13 +33,11 @@ public:
/// Method used only for loading main configuration files (authserver.conf and worldserver.conf)
bool LoadInitial(char const* file);
/**
* This method loads additional configuration files
* It is recommended to use this method in WorldScript::OnConfigLoad hooks
*
* @return true if loading was successful
*/
bool LoadMore(char const* file);
static ConfigMgr* instance()
{
static ConfigMgr *instance = new ConfigMgr();
return instance;
}
bool Reload();
@@ -59,20 +50,14 @@ public:
std::list<std::string> GetKeysByString(std::string const& name);
private:
bool GetValueHelper(const char* name, ACE_TString &result);
bool LoadData(char const* file);
typedef ACE_Thread_Mutex LockType;
typedef ACE_Guard<LockType> GuardType;
std::string _filename;
Config _config;
LockType _configLock;
boost::property_tree::ptree _config;
std::mutex _configLock;
ConfigMgr(ConfigMgr const&);
ConfigMgr& operator=(ConfigMgr const&);
};
#define sConfigMgr ACE_Singleton<ConfigMgr, ACE_Null_Mutex>::instance()
#define sConfigMgr ConfigMgr::instance()
#endif