1

我正在尝试编写一个只有一个命令的应用程序，因此，我想我可能会跳过 Cobra。

该应用程序应该能够支持所有类型的配置：

  • 命令行参数
  • 环境变量
  • 配置文件

我正在使用 viper，但我无法让它读取我的 cli 参数。

  v := viper.New()
  viper.SetConfigName("config") 
  viper.AddConfigPath(".")      
  v.SetDefault(pathKey, defaultPath)

  fs := pflag.NewFlagSet("app", pflag.ExitOnError)
  fs.String(pathKey, defaultPath, "Default path")
  fs.StringSlice(wordsKey, []string{""}, "Words")
  fs.String(loglevelKey, "info", "Log level")

  if err := v.BindPFlags(fs); err != nil {
    fmt.Println(err)
    os.Exit(1)
  }
  if err := fs.Parse(os.Args[1:]); err != nil {
    fmt.Println(err)
    os.Exit(1)
  }

  v.AutomaticEnv()

  if err := v.ReadInConfig(); err != nil {
    fmt.Println("no conf file") //ignore, it can be either cli params, or conf file
  }

  var c conf
  if err := v.Unmarshal(&c); err != nil {
    fmt.Println(err)
    os.Exit(1)
  }

但是我从来没有将 cli 参数放入 conf 结构中。v之前的打印Unmarshal不显示我提供的任何 cli 参数。

我错过了什么？我需要为此使用眼镜蛇吗？还是我必须手动分配每个标志集标志，例如fs.String(pathKey, defaultPath, "Default path")，手动分配给配置结构？

4

2 回答 2

1

对于后代，我想我发现了这个问题：

我的conf结构没有标志对应的键名。设置json:"logLevel"example 虽然字段称为 DisplayLogLevel 是不够的，它必须是：

const (
  pathKey = "path"
  wordsKey = "words"
  logLevelKey = "logLevel"
)

type conf struct {
   Path string `json:"path"`
   Words []string `json:"words"`
   LogLevel string `json:"logLevel"`
}
于 2022-01-11T20:50:54.360 回答
0

也许你必须设置配置类型。来自https://github.com/spf13/viper：

viper.SetConfigType("yaml") // REQUIRED if the config file does not have the extension in the name

于 2022-01-11T21:57:58.887 回答