7

我正在使用Viper和Cobra构建一个小应用程序。目前，我有一个这样的 yaml 文件：

hosts:
  - name: host1
    port: 90
    key: my_key
  - name: host2
    port: 90
    key: prompt

我已经阅读了使用 Viper 的配置文件。

当我运行viper.Get("hosts")它时，它会返回一个接口（或一部分接口？）。这是我最终得到的数据结构：

([]interface {}) (len=2 cap=2) {
 (map[interface {}]interface {}) (len=3) {
  (string) (len=4) "name": (string) (len=20) "host1",
  (string) (len=4) "port": (int) 90,
  (string) (len=3) "key": (string) (len=6) "my_key"
 },
 (map[interface {}]interface {}) (len=3) {
  (string) (len=3) "key": (string) (len=6) "prompt",
  (string) (len=4) "name": (string) (len=20) "host2",
  (string) (len=4) "port": (int) 90
 }
}

我想在这里做的是遍历每个数组元素并使用名称、端口和键的值执行操作。

我对 Golang 中的接口完全陌生，所以这不是很清楚，关于这方面的文献非常混乱:(

任何帮助表示赞赏。

4

1 回答 1

10

通过定义配置文件类型并使用viper.Unmarshal，您可以将接口转换为您需要的特定类型。这是一个例子：

main.go

package main

import (
    "fmt"

    "github.com/spf13/viper"
)

type Host struct {
    Name string
    Port int
    Key  string
}

type Config struct {
    Hosts []Host
}

func main() {
    viper.AddConfigPath("./")
    viper.SetConfigName("test")
    viper.ReadInConfig()
    var config Config
    err := viper.Unmarshal(&config)
    if err != nil {
        panic("Unable to unmarshal config")
    }
    for _, h := range config.Hosts {
        fmt.Printf("Name: %s, Port: %d, Key: %s\n", h.Name, h.Port, h.Key)
    }
}

test.yml

hosts:
  - name: host1
    port: 90
    key: my_key
  - name: host2
    port: 90
    key: prompt

跑：

$ go run main.go
Name: host1, Port: 90, Key: my_key
Name: host2, Port: 90, Key: prompt

如果您只想解码一些密钥，而不是整个配置文件，请使用viper.UnmarshalKey.

main.go

package main

import (
    "fmt"

    "github.com/spf13/viper"
)

type Host struct {
    Name string
    Port int
    Key  string
}

func main() {
    viper.AddConfigPath("./")
    viper.SetConfigName("test")
    viper.ReadInConfig()
    var hosts []Host
    err := viper.UnmarshalKey("hosts", &hosts)
    if err != nil {
        panic("Unable to unmarshal hosts")
    }
    for _, h := range hosts {
        fmt.Printf("Name: %s, Port: %d, Key: %s\n", h.Name, h.Port, h.Key)
    }
}

跑：

$ go run main.go
Name: host1, Port: 90, Key: my_key
Name: host2, Port: 90, Key: prompt
于 2017-03-12T14:34:55.983 回答