使用Golang解析yaml文件并获取数据

欢迎各位小伙伴来到golang学习网，相聚于此都是缘哈哈哈！今天我给大家带来《使用Golang解析yaml文件并获取数据》，这篇文章主要讲到等等知识，如果你对Golang相关的知识非常感兴趣或者正在自学，都可以关注我，我会持续更新相关文章！当然，有什么建议也欢迎在评论留言提出！一起学习！

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 吗？

解决方案

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 锚点/别名），这会导致堆栈溢出，因为代码不会检查循环。

好了，本文到此结束，带大家了解了《使用Golang解析yaml文件并获取数据》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！