首页 > 后端开发 > Golang > 如何从 golang 中的 yaml 中解组结构内的类型别名?

如何从 golang 中的 yaml 中解组结构内的类型别名?

王林
发布: 2024-02-06 08:57:08
转载
622 人浏览过

如何从 golang 中的 yaml 中解组结构内的类型别名?

问题内容

我想要以下 yaml

kind: bar
name: baryaml
登录后复制

在结构体 resource 中解组

type kind int

const (
    kind_foo kind = iota
    kind_bar
)

type resource struct {
    kind kind
    name string
}
登录后复制

有人可以解释为什么下面的代码无法存储正确的类型,即使它被正确解组了吗?

# output:
unmarshaled kind: 1
yamlbar: {0 baryaml}
登录后复制
# expected output:
unmarshaled kind: 1
yamlbar: {1 baryaml}
登录后复制
package main

import (
    "fmt"

    "gopkg.in/yaml.v3"
)

type Kind int

const (
    KIND_FOO Kind = iota
    KIND_BAR
)

func (k *Kind) UnmarshalYAML(value *yaml.Node) error {
    var kind string
    err := value.Decode(&kind)

    if err != nil {
        return err
    }

    var x Kind

    switch kind {
    case "foo":
        x = KIND_FOO
    case "bar":
        x = KIND_BAR
    default:
        return fmt.Errorf("unknown kind: %s", kind)
    }

    k = &x
    fmt.Println("Unmarshaled kind:", *k)
    return nil
}

type Resource struct {
    Kind Kind
    Name string
}

func main() {

    var yamlBar = `
kind: bar
name: baryaml
`
    r := Resource{}
    err := yaml.Unmarshal([]byte(yamlBar), &r)

    if err != nil {
        panic(err)
    }

    fmt.Println("yamlBar:", r)
}

登录后复制


正确答案


感谢@jimb 建议取消引用 k 指针:

func (k *Kind) UnmarshalYAML(value *yaml.Node) error {
    var kind string
    err := value.Decode(&kind)

    if err != nil {
        return err
    }

    switch kind {
    case "foo":
        *k = KIND_FOO
    case "bar":
        *k = KIND_BAR
    default:
        return fmt.Errorf("unknown kind: %s", kind)
    }

    fmt.Println("Unmarshaled kind:", *k)
    return nil
}
登录后复制

以上是如何从 golang 中的 yaml 中解组结构内的类型别名?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:stackoverflow.com
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板