2

尝试将Unmarshal配置hcl文件转换为结构，使用viper，返回此错误：1 error(s) decoding:\n\n* 'NATS' expected a map, got 'slice'。什么不见​​了？

编码：

func lab() {
    var c conf

    // config file
    viper.SetConfigName("draft")
    viper.AddConfigPath(".")
    viper.SetConfigType("hcl")
    if err := viper.ReadInConfig(); err != nil {
        log.Error(err)
        return
    }

    log.Info(viper.Get("NATS")) // gives [map[port:10041 username:cl1 password:__Psw__4433__ http_port:10044]]

    if err := viper.Unmarshal(&c); err != nil {
        log.Error(err)
        return
    }

    log.Infow("got conf", "conf", c)
}

type conf struct {
    NATS struct {
        HTTPPort int
        Port     int
        Username string
        Password string
    }
}

和配置文件（draft.hcl在当前目录内）：

NATS {
    HTTPPort = 10044
    Port     = 10041
    Username = "cl1"
    Password = "__Psw__4433__"
}

编辑

已经用包检查了这个结构，hcl它被正确地编组/解组。这也适用于yamland viper。

这两个地方是有区别log.Info(viper.Get("NATS"))的。当hcl版本返回一个地图切片时，该yaml版本返回一个地图：map[password:__psw__4433__ httpport:10044 port:10041 username:cl1]。

4

2 回答 2

5

您的 conf 结构与 HCL 不匹配。转换为 json 时，HCL 如下所示

{
"NATS": [
    {
      "HTTPPort": 10044,
      "Password": "__Psw__4433__",
      "Port": 10041,
      "Username": "cl1"
    }
  ]
}

所以Conf结构应该是这样的

type Conf struct {
    NATS []struct{
        HTTPPort int
        Port     int
        Username string
        Password string
    }
}

修改代码

package main
import (
  "log"
  "github.com/spf13/viper"
  "fmt"
)

type Conf struct {
    NATS []struct{
        HTTPPort int
        Port     int
        Username string
        Password string
    }
}

func main() {
    var c Conf
    // config file
    viper.SetConfigName("draft")
    viper.AddConfigPath(".")
    viper.SetConfigType("hcl")
    if err := viper.ReadInConfig(); err != nil {
        log.Fatal(err)
    }
    fmt.Println(viper.Get("NATS")) // gives [map[port:10041 username:cl1 password:__Psw__4433__ http_port:10044]]

    if err := viper.Unmarshal(&c); err != nil {
        log.Fatal(err)
    }
    fmt.Println(c.NATS[0].Username)
}
于 2018-01-13T15:05:31.253 回答
2

我知道这个问题已经有两年多了，但我最近遇到了同样的问题。

我使用 viper 能够将不同的配置文件加载到 Go 结构中，允许在 JSON、YAML、TOML、HCL 中进行配置，只需选择你喜欢的 :)

HCL 文件格式确实将映射包装到切片中，因为它允许重新定义如下部分：

section = {
  key1 = "value"
}

section = {
  key2 = "value"
}

这是其他格式不支持的。

这是我修复它的方法：

  • 我的解决方案意味着每个新块将覆盖同一键的任何先前定义，并保留所有其他块。你可以做一些合并魔法，但我不需要。
  1. 您需要制作一个挂钩来将一片地图转换为地图：
// sliceOfMapsToMapHookFunc merges a slice of maps to a map
func sliceOfMapsToMapHookFunc() mapstructure.DecodeHookFunc {
    return func(from reflect.Type, to reflect.Type, data interface{}) (interface{}, error) {
        if from.Kind() == reflect.Slice && from.Elem().Kind() == reflect.Map && (to.Kind() == reflect.Struct || to.Kind() == reflect.Map) {
            source, ok := data.([]map[string]interface{})
            if !ok {
                return data, nil
            }
            if len(source) == 0 {
                return data, nil
            }
            if len(source) == 1 {
                return source[0], nil
            }
            // flatten the slice into one map
            convert := make(map[string]interface{})
            for _, mapItem := range source {
                for key, value := range mapItem {
                    convert[key] = value
                }
            }
            return convert, nil
        }
        return data, nil
    }
}
  1. 那么你需要创建一个 DecodeHook：
configOption := viper.DecodeHook(mapstructure.ComposeDecodeHookFunc(
        sliceOfMapsToMapHookFunc(),
        mapstructure.StringToTimeDurationHookFunc(),
        mapstructure.StringToSliceHookFunc(","),
    ))

另外两个钩子是默认钩子，所以你可能想保留它们

然后将选项传递给 Unmarshal 方法

viper.Unmarshal(&c, configOption)

使用这种方法，您不需要在结构或地图周围进行切片。这也使它与其他配置文件格式兼容

于 2020-06-26T21:29:41.673 回答