在mprpcApplication类中,初始化函数Init需要加载配置文件,即rpc节点的IP和端口,zookeeper的IP和端口。所以写一个MprpcConfig类。
#pragma once
#include <unordered_map>
#include <string>
// rpcserver_ip= rpcserver_port= zookeeper_ip= zookeeper_port=?
// 框架读取配置文件类
class MprpcConfig
{
public:
// 负责解析加载配置文件
void LoadConfigFile(const char* config_file);
// 查询配置项信息
std::string Load(const std::string& key);
private:
std::unordered_map<std::string, std::string> m_configMap;
// 去掉字符串前后的空格
void Trim(std::string& src_buf);
};
test.conf
# rpc节点的IP地址
rpcserverip=127.0.0.1
#rpc节点的port端口号
rpcserverport=6500
# zk的IP地址
zookeeperip=127.0.0.1
# zk的port端口号
zookeeperport=2181
#include "mprpcConfig.h"
#include <iostream>
#include <string>
// 负责解析加载配置文件
void MprpcConfig::LoadConfigFile(const char* config_file)
{
FILE* fp = fopen(config_file, "r");
if (nullptr == fp)
{
std::cout << config_file << " is not exist!" << std::endl;
exit(EXIT_FAILURE);
}
// 注释,正确的匹配项 = 去掉开头多余的空格
while (!feof(fp))
{
char buf[512] = {0};
fgets(buf, 512, fp);
// 去掉字符串前面多余的空格
std::string read_buf(buf);
Trim(read_buf);
// 判断#的注释
if (read_buf[0] == '#' || 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);
int endidx = read_buf.find('\n', idx);
value = read_buf.substr(idx + 1, endidx - idx - 1);
Trim(value);
m_configMap.insert({key, value});
}
}
// 查询配置项信息
std::string MprpcConfig::Load(const std::string& key)
{
auto it = m_configMap.find(key);
if (it == m_configMap.end())
{
return "";
}
return it->second;
}
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);
}
}