如何使用 Viper 从嵌套的 YAML 结构中获取值？

Question

如何编写下面的代码以从我的嵌套 yaml 结构中获取字符串？

这是我的 yaml：

element:

  - one:

      url: http://test

      nested: 123

  - two:

      url: http://test

      nested: 123

weather:

  - test:

      zipcode: 12345

  - ca:

      zipcode: 90210

这是示例代码

viper.SetConfigName("main_config")

  viper.AddConfigPath(".")

  err := viper.ReadInConfig()

  if err != nil {

    panic(err)

  }

testvar := viper.GetString("element.one.url")

我的问题：

当我打印这个时，我得到一个空白字符串。根据文档，这就是获得嵌套元素的方式。我怀疑它不起作用，因为元素是列表。我需要做一个结构吗？我不确定如何制作，特别是如果它需要嵌套。

Answer 1

您可以解组嵌套的配置文件。主程序package mainimport (    "fmt"    "github.com/spf13/viper")type NestedURL struct {    URL string `mapstructure:"url"`    Nested int `mapstructure:"nested"`}type ZipCode struct {    Zipcode string `mapstructure:"zipcode"`}type Config struct {    Element [] map[string]NestedURL `mapstructure:"element"`    Weather [] map[string]ZipCode `mapstructure:"weather"`}func main() {    viper.SetConfigName("config")    viper.AddConfigPath(".")    if err := viper.ReadInConfig();  err != nil {        return    }    var config Config    if err := viper.Unmarshal(&config); err != nil {        fmt.Println(err)        return    }    fmt.Println(config)}配置文件element:  - one:      url: http://test      nested: 123  - two:      url: http://test      nested: 123weather:  - test:      zipcode: 12345  - ca:      zipcode: 90210

Answer 2

viper 库中有不同的 Get 方法可用，并且您的 YML 结构是类型[]map[string]string，因此要解析您的 YML 配置文件，您必须使用viper.Get方法。所以你必须像这样解析你的文件..注意：您也可以使用 struct 来取消编组数据。package mainimport (    "fmt"    "github.com/spf13/viper")func main() {    viper.SetConfigName("config")    viper.AddConfigPath(".")    err := viper.ReadInConfig()    if err != nil {        panic(err)    }    testvar := viper.Get("element")    fmt.Println(testvar)    elementsMap := testvar.([]interface{})    for k, vmap := range elementsMap {        fmt.Print("Key: ", k)         fmt.Println(" Value: ", vmap)        eachElementsMap := vmap.(map[interface{}]interface{})        for k, vEachValMap := range eachElementsMap {            fmt.Printf("%v: %v \n", k, vEachValMap)            vEachValDataMap := vEachValMap.(map[interface{}]interface{})            for k, v := range vEachValDataMap {                fmt.Printf("%v: %v \n", k, v)            }        }    }}// Output:/*Key: 0 Value:  map[one:map[url:http://test nested:123]]one: map[url:http://test nested:123]url: http://testnested: 123Key: 1 Value:  map[two:map[url:http://test nested:123]]two: map[url:http://test nested:123]url: http://testnested: 123*/

Answer 3

您可以使用Unmarshal或UnmarshalKey解析全部或部分数据并填充结构。它与解组 json 非常相似。在你的情况下，代码将是这样的：package mainimport (    "fmt"    "github.com/spf13/viper")// here we define schema of data, just like what we might do when we parse jsontype Element struct {    Url    string `mapstructure:"url"`    Nested int    `mapstructure:"nested"`}func main() {    viper.SetConfigName("config")    viper.AddConfigPath(".")    err := viper.ReadInConfig()    if err != nil {        panic(err)    }        // data in `element` key is a map of string to Element. We define a variable to store data into it.    elementParsed := make(map[string]*Element)    // read the key `element` in the yaml file, and parse it's data and put it in `elementParsed` variable    err = viper.UnmarshalKey("element", &elementParsed)    if err != nil {        panic(err)    }    fmt.Println(elementParsed["one"].Url) // will print: http://test     fmt.Println(elementParsed["one"].Nested) // will print: 123}