使用 Viper 语言处理界面

Question

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

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

而且我已经使用 Viper 读取了配置文件。

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

([]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 中的接口完全陌生，所以这不是很清楚，而且这方面的文献非常混乱:(

感谢任何帮助。

Answer 1

通过定义配置文件类型并使用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

使用 Viper 语言处理界面

Working with interfaces using Viper language

go

viper-go