登录
首页 >  Golang >  Go问答

保持 Go 中读取 YAML 文件的顺序不变

来源:stackoverflow

时间:2024-02-07 08:57:24 222浏览 收藏

珍惜时间,勤奋学习!今天给大家带来《保持 Go 中读取 YAML 文件的顺序不变》,正文内容主要涉及到等等,如果你正在学习Golang,或者是对Golang有疑问,欢迎大家关注我!后面我会持续更新相关内容的,希望都能帮到正在学习的大家!

问题内容

我正在尝试读取 yaml 文件中的数据,如下所示:

category name:
  task 1:
    name: meeting
    pm: 1
    tc: 0
    stc: 1
    optional: false

一个类别中可以有多个任务,也可以有多个类别。我最初是这样阅读 yaml 文件的。

type task struct {
    name     string
    pm       string
    tc       string
    stc      string
    optional bool
}
type category struct {
    name  string
    tasks []task
}
func parseFile() map[string]map[string]map[string]string {
    file, err := ioutil.ReadFile("config.yaml")

    if err != nil {
        log.Fatal(err)
    }

    data := make(map[string]map[string]map[string]string)
    err1 := yaml.Unmarshal(file, &data)

    if err1 != nil {
        log.Fatal(err1)
    }

    return data
}

然后我只有一个函数来循环遍历在 parsefile() 中创建的映射,并创建一个包含填充信息的类别列表。这非常有效...除了由于地图未排序而未保留 yaml 文件中写入的类别/任务的顺序这一事实之外。

通过在线阅读,我发现您可以使用 gokpg.in/yaml.v2 中的 yaml.mapslice{} 保留订单。但是,我不知道如何处理生成的 mapslice。我可以迭代它来获取类别,但我无法迭代 .value,因为 mapitems 是接口,但 mapitems 会存储我的任务列表。我该怎么做才能获取任务并放入 []task?

感谢帮助:)


正确答案


我设法通过更改 yaml 和结构的格式,然后直接解组到结构中来解决这个问题。

新结构。

type categories struct {
    category []struct {
        name string `yaml:"categoryname"`
        task []struct {
            name     string  `yaml:"taskname"`
            pm       float64 `yaml:"pm"`
            tc       float64 `yaml:"tc"`
            stc      float64 `yaml:"stc"`
            optional bool    `yaml:"optional"`
        } `yaml:"tasks"`
    } `yaml:"categories"`
}

yaml 文件的新格式支持结构中的内容。

categories:
  - categoryname: category 1
    tasks:
      - taskname: test 1
        pm: 1
        tc: 0
        stc: 1
        optional: false
      - taskname: test 2
        pm: 2
        tc: 0
        stc: 2
        optional: false
  - categoryname: category 2
    tasks:
      - taskname: test 3
        pm: 1
        tc: 2
        stc: 3
        optional: true

将 yaml 文件直接解组到新结构中

var tasks categories
    file, err := ioutil.readfile("config.yaml")

    if err != nil {
        log.fatal(err)
    }

    yaml.unmarshal(file, &tasks)

这解决了我的问题 - 不再使用映射,因此保留了顺序,并以我认为易于循环的方式使用结构。

package main

import (
    "fmt"
    "gopkg.in/yaml.v2"
    "io/ioutil"
    "log"
)

type Config struct {
    CategoryName []map[string]Task `yaml:"CategoryName"`
}

type Task struct {
    Name string `yaml:"Name"`
    PM int `yaml:"PM"`
    TC int `yaml:"TC"`
    STC int `yaml:"STC"`
    Optional bool  `yaml:"Optional"`
}

func main() {
    yamlData := []byte(`
CategoryName:
  - Task1:
      Name: "Meeting"
      PM: 1
      TC: 0
      STC: 1
      Optional: false
  - Task2:
      Name: "Meeting v2"
      PM: 2
      TC: 0
      STC: 2
      Optional: false
`)

    var config Config
    err := yaml.Unmarshal(yamlData, &config)
    if err != nil {
        log.Fatalf("Erreur lors de la décomposition du YAML : %v", err)
    }

    fmt.Printf("Configuration : %+v\n", config)
}

以上就是《保持 Go 中读取 YAML 文件的顺序不变》的详细内容,更多关于的资料请关注golang学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>