cobra 和 viper 的配置文件

Question

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

我有一个带有标志绑定的命令监听，我想在 yaml 文件中配置它。

代码：

listen 命令的初始化函数如下所示：

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"

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

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

Answer 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 来取回它们。

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

cobra 和 viper 的配置文件

Config file with cobra and viper

go

viper-go

go-cobra

问题

行为