使用cobra和viper配置文件

Question

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

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

代码：

侦听命令的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"

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

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

Answer 1

好的，感谢您提供的额外信息，它帮了我们很多忙！

问题

问题产生于检索标志值的方式。以下是您目前拥有的：

bind := cmd.Flag("bind").Value.String()
fmt.Printf("Binding definition provided: %sn", 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: %sn", bind)
fmt.Printf("Binding definition provided from viper: %sn", 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

将bind-config-param设置为"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