当前位置: 首页 > 工具软件 > mprpc > 使用案例 >

Mprpc框架的配置文件

隆飞驰
2023-12-01

在mprpcApplication类中,初始化函数Init需要加载配置文件,即rpc节点的IP和端口,zookeeper的IP和端口。所以写一个MprpcConfig类。

mprpcConfig.h

#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

mprpcConfig.cc

#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);
    }
}
 类似资料: