去掉字符串前后的空格
在使用c++一行一行地处理配置文件的时候,我发现用户写的配置文件中的每一行都由可能输入有的没有的空格,如下:
xxx = xx
去掉字符串前后的空格:
config.h
#pragma once
#include <unordered_map>
#include <string>
// rpcserverip rpcserverport zookeeperip zookeeperport
// 框架读取配置文件类
class MprpcConfig
{
public:
// 负责解析加载配置文件
void LoadConfigFile(const char *config_file);
// 查询配置项信息
std::string Load(const std::string &key); // 传入key,返回value
private:
std::unordered_map<std::string, std::string> m_configMap;
// 去掉字符串前后的空格
void Trim(std::string &src_buf);
};
config.cc
#include "mprpcconfig.h"
#include <iostream>
#include <string>
// 负责解析加载配置文件
void MprpcConfig::LoadConfigFile(const char *config_file)
{
FILE *pf = fopen(config_file, "r");
if (nullptr == pf)
{
std::cout << config_file << " is note exist!" << std::endl;
exit(EXIT_FAILURE);
}
// 需要处理的东西: 1.注释 2.正确的配置项 = 3.去掉开头的多余的空格
while(!feof(pf))
{
char buf[512] = {0};
fgets(buf, 512, pf); // 一行超过513个字节,是不是就出问题了
// 去掉字符串前面多余的空格
std::string read_buf(buf);
Trim(read_buf);
// 判断#的注释
if (read_buf[0] == '#' || read_buf.empty()) // read_buf.empty()代表去除空格以后得到的字符串是空的
{
continue;
}
// 解析配置项
int idx = read_buf.find('=');
if (idx == -1)
{
// 配置项不合法
continue;
}
std::string key;
std::string value;
key = read_buf.substr(0, idx);
Trim(key);
// rpcserverip=127.0.0.1\n
int endidx = read_buf.find('\n', idx);
value = read_buf.substr(idx+1, endidx-idx-1); // 去掉换行
Trim(value);
m_configMap.insert({key, value});
}
fclose(pf);
}
// 查询配置项信息
std::string MprpcConfig::Load(const std::string &key)
{
auto it = m_configMap.find(key);
if (it == m_configMap.end())
{
return "";
}
return it->second;
// 这里不能使用return m_configMap[key],因为m_configMap[key]会向m_configMap中添加键值对
}
// 去掉字符串前后的空格
void MprpcConfig::Trim(std::string &src_buf)
{
int idx = src_buf.find_first_not_of(' '); // 返回第一个不为空格的,下标
if (idx != -1)
{
// 说明字符串前面有空格
src_buf = src_buf.substr(idx, src_buf.size()-idx);
}
// 去掉字符串后面多余的空格
idx = src_buf.find_last_not_of(' ');
if (idx != -1)
{
// 说明字符串后面有空格
src_buf = src_buf.substr(0, idx+1);
}
}