带有 cobra 和 viper 的配置文件答案

【问题标题】：Config file with cobra and viper带有 cobra 和 viper 的配置文件
【发布时间】：2017-05-25 11:59:14
【问题描述】：

基本信息：我已经创建了一个 go 应用程序并使用 Cobra 来实现它。 Cobra 使用 Viper 作为命令行参数和标志。

我有一个带有标志绑定的命令侦听，我想在 yaml 文件中对其进行配置。

代码：

listen命令的init函数如下所示：

func init() {
    RootCmd.AddCommand(listenCmd)
    listenCmd.Flags().StringP("bind", "b", ":50051", "Provide bind definition")
    viper.BindPFlag("bind", listenCmd.Flags().Lookup("bind"))
}

我的申请代码在https://github.com/sascha-andres/go-logsink

问题：

当我使用 listen --bind "bla" 调用应用程序时，标志被正确设置为 bla，但我还没有找到使用位于我的主目录中的 YAML 文件来实现此目的的方法。

已尝试配置文件：

---

connect:
  bind: "bla"

和

---

bind: "bla"

在这两种情况下都找到了配置文件，但标志不是预期值而是默认值。

如何编写配置文件才能正确填充标志？

【问题讨论】：

抱歉，我无法重现您的问题：我尝试在 /tmp 和您的代码中使用我的 yaml 配置文件，并且正确找到了绑定参数并将其设置为它的配置值。但是，我注意到绑定参数一定不能嵌套才能找到，你的配置是这样吗？
@T.Claverie 我目前有第二个版本（未嵌套）。我已将其粘贴到pastebin.com/7TcqrdgV。更新问题并附有完整来源的链接
why is cobra not reading my config file的可能重复
@WGH供您参考：这个1年5个月，另一个1年2个月，这里不用投票关闭
@Sascha 我知道，但另一个问题更容易理解，因为它包括有问题的部分 - 使用绑定到 pflag 的变量而不是 viper.GetString() - 这个问题只有一个到存储库的链接不再有相关代码。

标签： go viper-go go-cobra

【解决方案1】：

好的，感谢您提供的额外信息，这对您有很大帮助！

问题

问题出在您检索标志值的方式上。这是您目前拥有的：

bind := cmd.Flag("bind").Value.String()
fmt.Printf("Binding definition provided: %s\n", bind)
server.Listen(bind)

当用 viper 绑定一个标志时，根据这个优先级，实际上是 viper 将持有最终值：

1. If present, use the flag value
2. Else, use the value from the config file
3. Else, use the default flag value

您的问题是您从命令的标志集中检索标志值，而不是从 viper。

行为

这是我测试的代码：

bind := cmd.Flag("bind").Value.String()
fmt.Printf("Binding definition provided: %s\n", bind)
fmt.Printf("Binding definition provided from viper: %s\n", viper.GetString("bind"))

没有绑定配置参数：

$ go-logsink listen
Using config file: /xxx/.go-logsink.yaml
Binding definition provided: :50051
Binding definition provided from viper: :50051

将绑定配置参数设置为“bla”（非嵌套，第二个配置文件）：

$ go-logsink listen
Using config file: /xxx/.go-logsink.yaml
Binding definition provided: :50051
Binding definition provided from viper: bla

将绑定配置参数设置为“bla”（不是嵌套的，第二个配置文件）和一个显式标志：

$ go-logsink listen --bind ":3333"
Using config file: /xxx/.go-logsink.yaml
Binding definition provided: :3333
Binding definition provided from viper: :3333

底线：当你用 viper 绑定你的标志时，使用 viper 来检索它们。

附加说明：在您的 README 中，生成 grpc 兼容代码的正确方法是将 grpc 插件添加到 protobuf 生成中：protoc --go_out=plugins=grpc:. *.proto

【讨论】：