配有眼镜蛇和毒蛇的配置文件

时间:2017-01-10 20:56:51

标签: go viper-go go-cobra

基本信息:我创建了一个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"

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

如何编写配置文件以正确填充标记?

1 个答案:

答案 0 :(得分:5)

好的,感谢您提供更多信息,它帮助了很多!

问题

问题来自于您检索标志值的方式。这是你现在拥有的:

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

当用v蛇绑定旗帜时,根据这个优先级,它实际上是v蛇,它将保留最终值:

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"))

没有bind config参数:

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

将bind config参数设置为“bla”(不嵌套,第二个配置文件):

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

将bind config参数设置为“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