Question

我正在使用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
 }
}

我想在这里做的是遍历每个数组元素并使用name，port和key的值执行操作。

我对Golang中的接口完全不熟悉，所以这不是很清楚，关于这方面的文献非常令人困惑：（

感谢任何帮助。

Answer 1

通过定义配置文件类型并使用/** filter */ function firstLetter() { return function(input) { return input.substring(0,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)
    }
}

使用Viper语言

1 个答案: