使用 Cobra/Viper 时遇到问题
Trouble using Cobra/Viper
我在同时使用 Cobra 和 Viper 时遇到问题。这就是我正在做的:
var options util.Config = util.Config{}
var rootCmd = &cobra.Command{
Use: "test [command] [subcommands]",
Run: func(cmd *cobra.Command, args []string) {
if err := server.Run(); err != nil {
l.Fatal(err)
}
},
}
// initConfig helps initialise configuration with a stated path
func initConfig() {
if options.Path != "" {
viper.SetConfigFile(options.Path)
}
viper.AutomaticEnv()
if err := viper.ReadInConfig(); err != nil {
fmt.Println("Could not use config file: ", viper.ConfigFileUsed())
}
}
func init() {
cobra.OnInitialize(initConfig)
rootCmd.PersistentFlags().StringVarP(&options.Path, "config", "n", "", "Path of a configuration file")
rootCmd.PersistentFlags().StringVarP(&options.Password, "password", "d", "", "Password to access the server")
viper.BindPFlag("password", rootCmd.PersistentFlags().Lookup("password"))
rootCmd.AddCommand(log.Cmd(&options))
}
func main() {
rootCmd.Execute()
}
我正在尝试在我的子命令(log.Cmd(&options)
中添加的命令)中检索值 options.Password,但是该字段未被填充。我很确定我正确地遵循了 Cobra 文档:https://github.com/spf13/cobra#create-rootcmd
将 cobra 标志绑定到 viper 选项只会将 cobra 标志绑定到 viper 选项,反之亦然。所以你可以通过
访问密码
pass := viper.GetString("password")
如果密码是通过 viper 或 cobra 设置的,而不是通过标志定义中定义的变量设置的。
基本上,您在这里有两个选择:要么您使用 cobra 而不将您的标志指向变量,然后通过对 viper.Get*
的各种调用来设置您的全局变量(您甚至可以在处理它们时清理它们),或者您将 viper 用作某种“参数注册表”,并在需要时调用 viper.Get*
。我倾向于使用前一种解决方案。
我在同时使用 Cobra 和 Viper 时遇到问题。这就是我正在做的:
var options util.Config = util.Config{}
var rootCmd = &cobra.Command{
Use: "test [command] [subcommands]",
Run: func(cmd *cobra.Command, args []string) {
if err := server.Run(); err != nil {
l.Fatal(err)
}
},
}
// initConfig helps initialise configuration with a stated path
func initConfig() {
if options.Path != "" {
viper.SetConfigFile(options.Path)
}
viper.AutomaticEnv()
if err := viper.ReadInConfig(); err != nil {
fmt.Println("Could not use config file: ", viper.ConfigFileUsed())
}
}
func init() {
cobra.OnInitialize(initConfig)
rootCmd.PersistentFlags().StringVarP(&options.Path, "config", "n", "", "Path of a configuration file")
rootCmd.PersistentFlags().StringVarP(&options.Password, "password", "d", "", "Password to access the server")
viper.BindPFlag("password", rootCmd.PersistentFlags().Lookup("password"))
rootCmd.AddCommand(log.Cmd(&options))
}
func main() {
rootCmd.Execute()
}
我正在尝试在我的子命令(log.Cmd(&options)
中添加的命令)中检索值 options.Password,但是该字段未被填充。我很确定我正确地遵循了 Cobra 文档:https://github.com/spf13/cobra#create-rootcmd
将 cobra 标志绑定到 viper 选项只会将 cobra 标志绑定到 viper 选项,反之亦然。所以你可以通过
访问密码pass := viper.GetString("password")
如果密码是通过 viper 或 cobra 设置的,而不是通过标志定义中定义的变量设置的。
基本上,您在这里有两个选择:要么您使用 cobra 而不将您的标志指向变量,然后通过对 viper.Get*
的各种调用来设置您的全局变量(您甚至可以在处理它们时清理它们),或者您将 viper 用作某种“参数注册表”,并在需要时调用 viper.Get*
。我倾向于使用前一种解决方案。