c++实现数据存储程序第一天—读取配置文件
2019年4月11日晚上,脑子里突然间冒出个想法(用c++撸一个数据存储服务的程序),之前有用GO写过一个简单的可分布式扩展的图片存储程序。虽然还没有具体思路,但是仍然打算开始动手,走一步算一步(做着做着思路就来了呢也说不定)。
一、开发编辑器
打算用Qt Creator,之前电脑有装这个IDE,感觉还行,就继续用它吧。
二、程序取名
随便取个名字就叫daobu吧,总不能叫test吧...
ok,开始建工程了
三、第一天主要内容
- 主要内容是c++实现读取配置文件(读取的部分用了某位大佬的源码,博客上看到的还不清楚真正出处)。源码主要如下:
- configcore.h
#ifndef CONFIGCORE_H
#define CONFIGCORE_H
#pragma once
#include
#include
- configcore.cpp
#include "configcore.h"
using namespace std;
ConfigCore::ConfigCore( string filename, string delimiter,
string comment )
: m_Delimiter(delimiter), m_Comment(comment)
{
// Construct a Config, getting keys and values from given file
std::ifstream in( filename.c_str() );
if( !in ) throw File_not_found( filename );
in >> (*this);
}
ConfigCore::ConfigCore()
: m_Delimiter( string(1,'=') ), m_Comment( string(1,'#') )
{
// Construct a Config without a file; empty
}
bool ConfigCore::KeyExists( const string& key ) const
{
// Indicate whether key is found
mapci p = m_Contents.find( key );
return ( p != m_Contents.end() );
}
/* static */
void ConfigCore::Trim( string& inout_s )
{
// Remove leading and trailing whitespace
static const char whitespace[] = " \n\t\v\r\f";
inout_s.erase( 0, inout_s.find_first_not_of(whitespace) );
inout_s.erase( inout_s.find_last_not_of(whitespace) + 1U );
}
std::ostream& operator<<( std::ostream& os, const ConfigCore& cf )
{
// Save a Config to os
for( ConfigCore::mapci p = cf.m_Contents.begin();
p != cf.m_Contents.end();
++p )
{
os << p->first << " " << cf.m_Delimiter << " ";
os << p->second << std::endl;
}
return os;
}
void ConfigCore::Remove( const string& key )
{
// Remove key and its value
m_Contents.erase( m_Contents.find( key ) );
return;
}
std::istream& operator>>( std::istream& is, ConfigCore& cf )
{
// Load a Config from is
// Read in keys and values, keeping internal whitespace
typedef string::size_type pos;
const string& delim = cf.m_Delimiter; // separator
const string& comm = cf.m_Comment; // comment
const pos skip = delim.length(); // length of separator
string nextline = ""; // might need to read ahead to see where value ends
while( is || nextline.length() > 0 )
{
// Read an entire line at a time
string line;
if( nextline.length() > 0 )
{
line = nextline; // we read ahead; use it now
nextline = "";
}
else
{
std::getline( is, line );
}
// Ignore comments
line = line.substr( 0, line.find(comm) );
// Parse the line if it contains a delimiter
pos delimPos = line.find( delim );
if( delimPos < string::npos )
{
// Extract the key
string key = line.substr( 0, delimPos );
line.replace( 0, delimPos+skip, "" );
// See if value continues on the next line
// Stop at blank line, next line with a key, end of stream,
// or end of file sentry
bool terminate = false;
while( !terminate && is )
{
std::getline( is, nextline );
terminate = true;
string nlcopy = nextline;
ConfigCore::Trim(nlcopy);
if( nlcopy == "" ) continue;
nextline = nextline.substr( 0, nextline.find(comm) );
if( nextline.find(delim) != string::npos )
continue;
nlcopy = nextline;
ConfigCore::Trim(nlcopy);
if( nlcopy != "" ) line += "\n";
line += nextline;
terminate = false;
}
// Store key and value
ConfigCore::Trim(key);
ConfigCore::Trim(line);
cf.m_Contents[key] = line; // overwrites if key is repeated
}
}
return is;
}
bool ConfigCore::FileExist(std::string filename)
{
bool exist= false;
std::ifstream in( filename.c_str() );
if( in )
exist = true;
return exist;
}
void ConfigCore::ReadFile( string filename, string delimiter,
string comment )
{
m_Delimiter = delimiter;
m_Comment = comment;
std::ifstream in( filename.c_str() );
if( !in ) throw File_not_found( filename );
in >> (*this);
}
- config.h
#ifndef CONFIG_H
#define CONFIG_H
#include "configcore.h"
#include
- config.cpp
#include "config.h"
#include
#include
#include
#include
using namespace std;
Config::Config()
{
// // 默认配置文件路径
// this->confDirect = "../../conf";
// // 设置配置文件后缀名
// this->confSuffix = ".conf";
}
/**
* @brief config::config
* @param direct
* @param suffix
*/
Config::Config(string direct,string suffix)
{
// 设置配置文件路径
this->confDirect = direct;
// 设置配置文件后缀名
this->confSuffix = suffix;
}
/**
* 获取目录下所有符合后缀的文件名
*
* @brief Config::getFilesFromDirect
*/
void Config::getFilesFromDirect() {
struct dirent *ptr;
DIR *dir = opendir(this->confDirect.c_str());
string::size_type idx;
string dirName;
while( (ptr=readdir(dir)) != NULL ) {
// 4 表示目录; 8 表示文件; 0 表示未知
if(ptr->d_type == 8) {
//跳过'.'和'..'两个目录
if(ptr->d_name[0] == '.') {
continue;
}
// 符合后缀名的文件名加入到迭代器中去
dirName = ptr->d_name;
idx = dirName.find(this->confSuffix);
if(idx != string::npos) {
this->allFiles.push_back(dirName);
}
}
}
closedir(dir);
}
/**
* 读取文件内容到内存
*
* @brief config::readFileToMemory
* @return
*/
bool Config::readFileToMemory() {
this->getFilesFromDirect();
int fileSize = this->allFiles.size();
if (fileSize == 0) {
throw "No right config file!";
} else if (fileSize == 1) {
this->Config_Core.ReadFile(this->confDirect +"/"+ this->allFiles[0]);
} else if (fileSize > 1) {
for (int i=0; iConfig_Core.ReadFile(this->confDirect +"/"+ this->allFiles[i]);
}
}
return true;
}
- 测试入口main.cpp
#include
#include
#include
#include
using namespace std;
#define MAX_PATH_LENGTH 150
int main()
{
// 获取当前路径
char cwd[MAX_PATH_LENGTH];
getcwd(cwd, MAX_PATH_LENGTH);
string cwdString = cwd;
cwdString += "/conf";
Config conf = Config(cwdString,".conf");
try {
conf.readFileToMemory();
} catch (char const* e) {
cout << e << endl;
return 0;
}
conf.Config_Core.Add("test",1099999);
cout << "add:" << conf.Config_Core.Read("add") << endl;
cout << "ipAddress:" << conf.Config_Core.Read("ipAddress") << endl;
cout << "username:" << conf.Config_Core.Read("username") << endl;
cout << "test:" << conf.Config_Core.Read("test") << endl;
return 0;
}
运行结果
- .conf配置文件的内容如下:
add=1234
username1=3333
ipAddress=10.10.90.125
port=3001
username=mark
password=2d2df5a
- 程序输出
add:1234
ipAddress:10.10.90.125
username:mark
test:1099999