C/C++ 读取配置文件的几种方式-1. protobuf

  • 将配置信息读取到string流中
/**
 * @brief 将配置文件读取到string流中
 * 
 * @param file_name :配置文件名称
 * @param content: 读取后的string配置信息流
 * @return true :成功读取
 * @return false:打开文件失败
 */
bool GetContent(const std::string &file_name, std::string *content) {
  // 打开文件流
  std::ifstream fin(file_name);
  if (!fin) {
    return false;
  }
  // 获取流缓冲区 
  std::stringstream str_stream;
  str_stream << fin.rdbuf();
  *content = str_stream.str();
  return true;
}
  • 创建proto格式配置文件(dag_config.proto)
message DAGConfig {
    enum SubnodeType {
        SUBNODE_IN = 1; 
        SUBNODE_OUT = 2;
        SUBNODE_NORMAL = 3;
    };

    // Subnode instance.
    message Subnode {
        required int32 id = 1;
        required string name = 2;
        // node private data.节点私有数据
        optional string reserve = 3;
        optional SubnodeType type = 4 [default = SUBNODE_NORMAL];
    };

    message SubnodeConfig {
        repeated Subnode subnodes = 1;
    };

    message Event {
        required int32 id = 1;
        optional string name = 2;
    };

    message Edge {
        required int32 id = 1;
        required int32 from_node = 2;
        required int32 to_node = 3;
        repeated Event events = 4;
    };

    message EdgeConfig {
        repeated Edge edges = 1;
    }

    message SharedData {
        required int32 id = 1;
        required string name = 2;
    };

    message SharedDataConfig {
        repeated SharedData datas = 1;
    }

    required SubnodeConfig subnode_config = 1;
    required EdgeConfig edge_config = 2;
    required SharedDataConfig data_config = 3;
};
  • 将proto文件编译成.cc和.h文件
  • 使用protobuf的序列化进行配置文件解析
  DAGConfig dag_config;
  string content;
  // 获取内容
  if (!GetContent(dag_config_path, &content)) {
    return false;
  }
  // 解析字符串:使用protobuf的序列化进行配置文件解析,将解析结果放置在dag_config中
  if (!TextFormat::ParseFromString(content, &dag_config)) {
    return false;
  }
上一篇:milvus的compaction机制


下一篇:Python电能质量扰动信号分类(七)基于CNN-TCN-Attention的扰动信号识别模型