解析YAML并返回行号

Question

我正在从Go中的YAML输入创建一个文档生成器。它需要指定每个项目/节点是从YAML文件的哪一行生成的。有没有办法在Go中实现？

例如，这是一个YAML文件

- key1: item 1
  key2: item 2
- key1: another item 1
  key2: another item 2

我希望看到以下内容

[
     {'__line__': 1, 'key1': 'item 1', 'key2': 'item 2'},
     {'__line__': 3, 'key1': 'another item 1', 'key2': 'another item 2'},
]

我看到一个类似的问题回答了Python解析YAML，返回行号，但我不知道如何利用https://pkg.go.dev/gopkg.in/yaml.v3

Answer 1

在 Go 中，这可以通过实现自定义解编拆器并手动设置行号来实现。

因此，让我们构建文件所需的数据结构：

type ListItem struct {
    Line int

    ListItemData
}

type ListItemData struct {
    Key1 string `yaml:"key1"`
    Key2 string `yaml:"key2"`
}

现在，我们通过在< code>ListItem类型上创建一个方法来实现解组器接口:

golang prettyprint-override">func (li *ListItem) UnmarshalYAML(value *yaml.Node) error {
    err := value.Decode(&li.ListItemData)
    if err != nil {
        return err
    }
     
    // Save the line number
    li.Line = value.Line

    return nil
}

您会注意到我创建了一个内部结构，并且只将数据解码为该结构。这是因为如果我们在 li 本身上调用 Decode，我们会得到一个堆栈溢出，因为解码会反复调用 UnmarshalYAML 方法。

你可以在网上试试，看看行号设置是否正确。