C++实现RPC分布式网络通信框架（六）-------配置文件及基础框架设计

在方法可以发布为RPC方法后我们需要将其放到我们的框架中，毕竟项目的初始目标是做一个框架而不是具体的业务代码，所以我们需要一个框架，在整个程序运行起来后先初始化这个框架，然后使用一个网络服务对象将服务对象发布到rpc节点上，再启动一个rpc服务节点，Run以后进程进入阻塞状态等待远程的RPC调用请求。现在我们就来实现基础的框架类。
由我们上面的分析可以知道，我们在运行起来的时候应该是下面这种逻辑：

int main(int argc,char **argv)
{
    
    
    //先调用框架的初始化操作，从init方法读取配置服务，比如IP地址和端口号
    MprpcApplication::Init(argc,argv);

    //服务提供者，让我们可以发布该服务
    RpcProvider provider;
    //把UserService对象发布到rpc节点上
    provider.NotifyService(new UserService());
    
    //启动一个rpc服务发布节点，run以后，进程进入阻塞状态，等待远程的rpc请求
    provider.Run();

    return 0;
}

所以我们现在对上面涉及的类和函数等进行实现。

一、MprpcConfig的实现

我们运行时需要打开通道，即读取配置文件，就是读取调用方和被调用方的IP地址和端口等，所以我们先对读取配置文件的功能进行实现以及配置文件的书写。

test.conf

#配置文件

#rpc节点的ip地址
rpcserverip=127.0.0.1

#rpc节点的端口号
rpcserverport=8000

#zk的ip地址
zookeeperip=127.0.0.1

#zk的端口号
zookeeperport=2181

接下来我们就需要对这个配置文件进行读取，理清思路后先书写头文件：

mprpcconfig.h

#pragma once
//配置文件为键值对，且不需要排序，所以使用无序哈希
#include<unordered_map>
#include<string>
//配置文件为：rpcserverip,rpcserverport,zookeeperip,zookeeperport

//框架读取配置文件类
class MprpcConfig
{
    
    
public:
    //负责解析加载配置文件
    void LoadConfigFile(const char *file);
    //查询配置项信息
    std::string Load(std::string const &key);
private:
    std::unordered_map<std::string,std::string> m_configMap;
    //去掉字符串前后的空格
    void Trim(std::string &src_buf);
};

然后对头文件中的对应方法进行实现：

mprpcconfig.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 << " not exists" << std::endl;
        exit(EXIT_FAILURE);
    }

    //文件读取成功
    //1.注释 2.正确的配置项= 3.去掉开头的多余空格
    while (!feof(pf))
    {
    
    
        char buff[512] = {
    
    0};
        fgets(buff, 512, pf);

        std::string read_buf(buff);
        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(std::string const &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);
    }
}

二、MprpcApplication的实现

从逻辑角度分析，我们的框架应该只有一个，所以需要使用单例模式来创建，单例模式简而言之就是让该类只能生成一个实例对象，具体不再赘述。然后我们在该类中需要提供Init()方法、以及获取单例的方法。

//mprpcapplication.h
#pragma once

#include"mprpcconfig.h"
#include"mprpccontroller.h"
#include"mprpcchannel.h"
//mprpc框架的基础类，负责框架的一些初始化操作，使用单例模式设计
class MprpcApplication
{
    
    
public:
    static void Init(int argc,char **argv);
    static MprpcApplication& GetInstance();//用于获取单例对象
private:
    static MprpcConfig m_config;//用于读取配置文件
    MprpcApplication(){
    
    };
    MprpcApplication(const MprpcApplication&)=delete;
    MprpcApplication(MprpcApplication&&)=delete;
};

在mprpcapplication.h中我们声明了MprpcApplication类和对应的一些方法，由于是单例模式，所以需要将拷贝构造有关的方法全部删除掉，防止重新构造新对象，并且为了读取配置文件，我们需要在这个类中声明一个MprpcConfig对象。其实现则在对应的.cc文件中：

//mprpcapplication.cc
#include"mprpcapplication.h"
#include<iostream>
#include<unistd.h>
#include<string>

//类外对静态对象初始化
MprpcConfig MprpcApplication::m_config;
//实现成全局的方法
void ShowArgHelp()
{
    
    
    std::cout<<"format:command -i <configfile>"<<std::endl;
}
void MprpcApplication::Init(int argc,char **argv)
{
    
    
    if(argc<2)//参数不足
    {
    
    
        ShowArgHelp();//打印参数的配置说明
        exit(EXIT_FAILURE);
    }

    //使用getopt读取参数
    int c=0;
    std::string config_file;//配置选项
    while((c=getopt(argc,argv,"i:"))!=-1)
    {
    
    
        switch (c)
        {
    
    
        //指定选项
        case 'i':
            config_file=optarg;
            break;
        //出现了不想要的参数
        case '?':
            std::cout<<"invalid args"<<std::endl;
            ShowArgHelp();
            exit(EXIT_FAILURE);
        case ':':
            std::cout<<"need <configfile>"<<std::endl;
            ShowArgHelp();
            exit(EXIT_FAILURE);
        default:
            break;
        }
    }
    //开始加载配置文件 加载IP，端口，zk的IP和端口
    m_config.LoadConfigFile(config_file.c_str());
    //测试是否成功加载
    std::cout<<"rpcserverip:"<<m_config.Load("rpcserverip")<<std::endl;
    std::cout<<"rpcserverport:"<<m_config.Load("rpcserverport")<<std::endl;
    std::cout<<"zookeeperip:"<<m_config.Load("zookeeperip")<<std::endl;
    std::cout<<"zookeeperport:"<<m_config.Load("zookeeperport")<<std::endl;
}
MprpcApplication& MprpcApplication::GetInstance()
{
    
    
    static MprpcApplication app;
    return app;
}

//获取配置项
MprpcConfig& MprpcApplication::GetConfig()
{
    
    
    return m_config;
}

现在我们的基础框架已经初见形状，即将为框架进行填充，使项目逐渐完善起来，项目持续更新中……