Golang 从 yaml 文件中提取值答案

【问题标题】：Golang extract value from yaml fileGolang 从 yaml 文件中提取值
【发布时间】：2021-02-25 22:17:46
【问题描述】：

jobs:
- name: test
 public: true
 plan:
 - try:
     task: task1
     file: test1.yaml
   on_success:
     in_parallel:
       steps:
       - task: task2
         file: test2.yaml
       - task: task3
         file: task3.yaml

我想从此 yaml 中提取 task 的值。这很棘手，因为它可以位于文件中的不同路径。除了上面的task 可以在 yaml 文件中找到之外，还有一些不同的路径。有没有一种简单的方法可以提取task 的所有值？我应该转换成json吗？

【问题讨论】：

task 的“the”值是什么这里有三个不同的 task 键；您认为其中哪些是“该”价值？到目前为止，您尝试过什么？
如果您知道如何使用json，您也可以这样做，但对于yaml（您的文件格式）：pkg.go.dev/gopkg.in/yaml.v2 是一个很好的起点
@Jorropo 我可以很容易地将其转换为 JSON。在 JSON 中你会怎么做？
@EliBendersky 想要所有这些 task 键的值。我尝试做一个 YAML 解组器，但遇到了问题，因为我无法定义解组器的结构，因为 task 键可能出现在文件中的许多不同位置。

标签： go yaml

【解决方案1】：

go-yaml 希望您为该类型定义一个目标类型和一个解组器。

如果您想使用键 task 提取某个映射中的所有值，您的类型将是任务名称列表。

对于该类型，您需要创建一个解组器函数，该函数采用 YAML 文件的根节点并将任务名称提取到您定义的类型的对象中。

这是一个最小的工作示例：

package main

import (
    "errors"
    "fmt"
    "gopkg.in/yaml.v3"
)

type Tasks struct {
    items []string
}

// descend implements recursive descent into YAML mapping and sequence structures
func (t *Tasks) descend(node *yaml.Node) error {
    switch node.Kind {
    case yaml.SequenceNode:
        for _, item := range(node.Content) {
            t.descend(item)
        }
    case yaml.MappingNode:
        for i := 0; i < len(node.Content); i += 2 {
            key := node.Content[i]
            value := node.Content[i+1]
            if key.Kind != yaml.ScalarNode ||
                key.Value != "task" {
                t.descend(value)
                continue
            }
            if value.Kind != yaml.ScalarNode {
                return errors.New("encountered non-scalar task")
            }
            t.items = append(t.items, value.Value)
        }
    }
    return nil
}

// UnmarshalYAML is the unmarshaler that will be called by the YAML processor.
func (t *Tasks) UnmarshalYAML(value *yaml.Node) error {
    t.items = nil
    return t.descend(value)
}


func main() {
    var t Tasks
    // I fixed some whitespace issues in your YAML input
    if err := yaml.Unmarshal([]byte(`jobs:
- name: test
  public: true
  plan:
  - try:
      task: task1
      file: test1.yaml
    on_success:
      in_parallel:
        steps:
        - task: task2
          file: test2.yaml
        - task: task3
          file: task3.yaml`), &t); err != nil {
        panic(err)
    }
    for _, item := range(t.items) {
        fmt.Println(item)
    }
}

输出：

task1
task2
task3

请注意，此解决方案通常不正确，因为 YAML 节点可能包含循环（由于 YAML 锚/别名），这会导致堆栈溢出，因为代码不检查循环。

【讨论】：