在 yaml-cpp 中操作节点

Manipulating nodes in yaml-cpp

我目前正在开发一个项目,我想使用库 yaml-cpp 进行配置。现在我正在尝试编写一个函数,它被赋予一个向量 tags/strings(例如,"A"、"B"、"C")和一些参数值(例如,31)。然后该函数应为每个标签插入一个映射节点(如果这样的节点尚不存在),将其映射到后续节点,最后创建一个具有参数值的标量节点。在所述示例调用之后,配置应为

A:
  B:
    C: 31

你可以在下面看到一个干净简单的实现的尝试:

Config.hpp

class Config {

public:

    template<typename T>
    static void setConfigParam(const std::vector<std::string> &tags,
            T value) {
        assert(tags.size() > 0);
        assert(config);
        YAML::Node parentNode = config;
        for(int i=0; i < tags.size()-1; i++) {
            const std::string & tag = tags.at(i);
            if(parentNode[tag]) {
                YAML::Node childNode = parentNode[tag];
                parentNode = childNode;
            }
            else {
                YAML::Node childNode(YAML::NodeType::Map);
                parentNode[tag] = childNode;
                parentNode = childNode;
            }
        }
        parentNode[tags.back()] = value;
        std::cout << "Config:\n" << config << "\n\n";
    }

private:

    static YAML::Node config;

};

YAML::Node Config::config(YAML::NodeType::Map);

Test.cpp

#include "Config.hpp"

int main(int argc, char **argv) {
    int val1 = 3;
    vector<string> tags1 = {"A1","B1","C1","D1"};
    Config::setConfigParam(tags1, val1);
    string val2 = "test";
    vector<string> tags2 = {"A1","B1","C1","D2"};
    Config::setConfigParam(tags2, val2);
    bool val3 = true;
    vector<string> tags3 = {"A1","B1","C1"};
    Config::setConfigParam(tags3, val3);
    double val4 = 2.385;
    vector<string> tags4 = {"A1","B2","C1"};
    Config::setConfigParam(tags4, val4);
}

函数的问题setConfigParam()好像是行

YAML::Node parentNode = config;

parentNode = childNode

根据我对 tutorial 的理解,第一行显然创建了节点 parentNode 作为 config 的别名。在每次函数调用时,第二行的第一次执行会用新创建的 childNode 替换现有的 config 对象。然而,在剩余的迭代中,config 不再完全重置,而是按预期进行调整。

如果一切正常 运行,输出应该是:

Config:
A1:
  B1:
    C1:
      D1: 3

Config:
A1:
  B1:
    C1:
      D1: 3
      D2: test

Config:
A1:
  B1:
    C1: true

Config:
A1:
  B1:
    C1: true
  B2:
    C1: 2.385

上面代码的实际输出是

Config:
B1:
  C1:
    D1: 3

Config:
B1:
  C1:
    D2: test

Config:
B1:
  C1: true

Config:
B2:
  C1: 2.385

我试图通过使用指针来防止提到的问题,并最终设法提出以下实现,它至少在测试代码中可以正常工作:

template<typename T>
static void setConfigParam(const std::vector<std::string> &tags,
        T value) {
    int numTags = tags.size();
    // Reads root node.
    assert(numTags > 0);
    assert(config);
    YAML::Node* parentNode = &config;
    // Determines index of first non-existing node.
    int i=0;
    for(; i < numTags-1; i++) {
        const std::string & tag = tags.at(i);
        if((*parentNode)[tag]) {
            auto childNode = (*parentNode)[tag];
            parentNode = &childNode;
        }
        else {
            break;
        }
    }
    // Note: necessary because *parentNode will later point
    // to a different node due to lib behavior .
    YAML::Node lastExistingNode = *parentNode;
    // Sets node value and creates missing nodes.
    if(i == numTags - 1) {
        lastExistingNode[tags.back()] = value;
    }
    else {
        YAML::Node newNode;
        newNode = value;
        for(int j = tags.size()-1; j>i; j--) {
            YAML::Node tmpNode;
            tmpNode[tags.at(j)] = newNode;
            newNode = tmpNode;
        }
        // Inserts missing nodes.
        lastExistingNode[tags.at(i)] = newNode;
    }
    std::cout << "Config:\n" << config << "\n\n";
}

但是,我相信一定有更简单、更清洁的解决方案。因此,我非常感谢任何有关如何通过更简单的实现使函数正常工作的想法,特别是因为库中没有太多文档。

在您的第一个示例中,分配给现有节点的行为与您预期的不同。有关详细信息,请参阅 this issue

你的指针示例解决了这个问题,但它还有另一个问题:它取消引用堆栈变量。

相反,使用递归解决方案:

template<typename T, typename Iter>
void setConfigParam(YAML::Node node, Iter begin, Iter end, T value) {
  if (begin == end) {
    return;
  }
  const auto& tag = *begin;
  if (std::next(begin) == end) {
    node[tag] = value;
    return;
  }
  if (!node[tag]) {
    node[tag] = YAML::Node(YAML::NodeType::Map);
  }
  setConfigParam(node[tag], std::next(begin), end, value);
}