我正在尝试使用 GO 读取 YAML 文件并将其映射到我定义的结构。 The YAML is below :

--- # go_time_tracker.yml
owner: "Phillip Dudley"
initialized: "2012-10-31 15:50:13.793654 +0000 UTC"
time_data:
  - action: "start"
    time: "2012-10-31 15:50:13.793654 +0000 UTC"
  - action: "stop"
time: "2012-10-31 16:00:00.000000 +0000 UTC"

我用了the following code读入文件，解码数据，然后打印一些数据。

package main

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

type Date_File struct {
    Owner    string      `yaml:"owner"`
    Init     time.Time   `yaml:"initialized"`
    TimeData []Time_Data `yaml:"time_data"`
}

type Time_Data struct {
    //
    Action string    `yaml:"action"`
    Time   time.Time `yaml:"time"`
}

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

func read() (td *Date_File) {
    //td := &Date_File{}
    gtt_config, err := ioutil.ReadFile("go_time_tracker.yml")
    checkerr(err)
    err = yaml.Unmarshal(gtt_config, &td)
    return td
}

func main() {
    //
    time_data := read()
    fmt.Println(time_data)
    fmt.Println(time_data.TimeData[0])
    fmt.Println(time_data.Owner)
}

当我运行它时，第一个 fmt.Println(time_data) 起作用，显示引用及其数据。下一行虽然没有说明索引超出范围。 This is the error :

$ go run yaml_practice_2.go 
&{Phillip Dudley 0001-01-01 00:00:00 +0000 UTC []}
panic: runtime error: index out of range

goroutine 1 [running]:
panic(0x559840, 0xc82000a0e0)
    /usr/lib/go-1.6/src/runtime/panic.go:481 +0x3e6
main.main()
    /home/predatorian/Documents/go/src/predatorian/yaml/yaml_practice_2.go:41 +0x2aa
exit status 2

然后我想也许我的 YAML 格式不正确，所以我将 YAML 文件加载到 Ruby 的 IRB 中，并且 this is what I got .

irb(main):004:0> data2 = YAML.load(File.read("go_time_tracker.yml"))
=> {"owner"=>"Phillip Dudley", "initialized"=>"2012-10-31 15:50:13.793654 +0000 UTC", "time_data"=>[{"action"=>"start", "time"=>"2012-10-31 15:50:13.793654 +0000 UTC"}, {"action"=>"stop", "time"=>"2012-10-31 16:00:00.000000 +0000 UTC"}]}

IRB 输出显示我的 YAML 格式正确，但是，我认为我没有正确解码它。但是，我不确定我需要做什么才能让它发挥作用。我确定我没有考虑如何正确地执行此操作，因为 Ruby 隐藏了很多。

最佳答案

首先，在

err = yaml.Unmarshal(gtt_config, &td)

您将得到相应的错误，即时间解析错误。 yaml 解码器期望时间在 RFC3339 中格式。有几种方法可以解决此问题:

1. 将 YAML 文件中的时间数据更改为 RFC3339 格式，例如“2012-10-31T15:50:13.793654Z”
2. 如果您无法修改 YAML 文件，则需要实现自定义解码器。

对于(2)，我想到了两种解决方案:

1. 实现yaml.Unmarshaler接口(interface)，或者
2. 将time.Time包装成自定义类型，然后实现encoding.TextUnmarshaler接口(interface)

解决方案 (1):您需要为 Date_File 和 Time_Data 类型实现自定义 Unmarshaler。将以下内容添加到您的源代码中。

func (df *Date_File) UnmarshalYAML(unmarshal func(interface{}) error) error {
    //Unmarshal time to string then convert to time.Time manually
    var tmp struct {
        Owner    string      `yaml:"owner"`
        Init     string      `yaml:"initialized"`
        TimeData []Time_Data `yaml:"time_data"`
    }
    if err := unmarshal(&tmp); err != nil {
        return err;
    }

    const layout = "2006-01-02 15:04:05.999999999 -0700 MST"
    tm, err := time.Parse(layout, tmp.Init)
    if err != nil {
        return err
    }

    df.Owner    = tmp.Owner
    df.Init     = tm
    df.TimeData = tmp.TimeData

    return nil
}

func (td *Time_Data) UnmarshalYAML(unmarshal func(interface{}) error) error {
    //Unmarshal time to string then convert to time.Time manually
    var tmp struct {
        Action string `yaml:"action"`
        Time   string `yaml:"time"`
    }
    if err := unmarshal(&tmp); err != nil {
        return err;
    }

    const layout = "2006-01-02 15:04:05.999999999 -0700 MST"
    tm, err := time.Parse(layout, tmp.Time)
    if err != nil {
        return err
    }

    td.Action = tmp.Action
    td.Time   = tm

    return nil
}

如果您有许多具有 time.Time 字段的 type，解决方案 (1) 可能不切实际。

解决方案(2):如果您查看 yaml decoder 的源代码，它依赖于 TextUnmarshaler 将字符串转换为相应的类型。在这里你需要:

1. 定义自定义时间类型(例如 CustomTime)
2. 将结构中的每个 time.Time 字段替换为 CustomTime
3. 实现UnmarshalText

(3) 的片段:

type CustomTime struct {
    time.Time
}
func (tm *CustomTime) UnmarshalText(text []byte) error {
    const layout = "2006-01-02 15:04:05.999999999 -0700 MST"
    tmValue, err := time.Parse(layout, string(text))
    if err != nil {
        return err
    }

    tm.Time = tmValue
    return nil    
}
//Not directly related, for print function etc.
func (tm CustomTime) String() string {
    return tm.Time.String()
}

关于GO 读取 YAML 文件并映射到结构 slice ，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/39680445/