我用cobra和viper编写了一个简单的CLI工具,最近我一直在重构它以避免包全局变量,主要是因为使用cobra init
建议的布局进行测试很困难。
而不是...
var rootCmd = &cobra.Command{
...
}
func main() {
rootCmd.Execute()
}
我有更像是:
func NewCmdRoot() *cobra.Command {
cmd := &cobra.Command{
...
}
return cmd
}
func main() {
rootCmd := NewCmdRoot()
rootCmd.Execute()
}
这实际上效果很好,并且使测试更容易从一组干净的cli选项开始。我在将Viper集成到新方案中时遇到了一些困难。如果我只关心root命令,我可以在PersistentPreRun
命令中设置这些东西,如下所示:
func initConfig(cmd *cobra.Command) {
config := viper.New()
rootCmd := cmd.Root()
config.BindPFlag("my-nifty-option", rootCmd.Flag("my-nifty-option")); err != nil {
// ...stuff happens here...
config.ReadInConfig()
// What happens next?
}
func NewCmdRoot() *cobra.Command {
cmd := &cobra.Command{
PersistentPreRun: func(cmd *cobra.Command, args []string) {
initConfig(cmd)
},
}
这是一种工作方式:只要我只对与Cobra命令行选项对应的配置选项感兴趣,事情就会按预期工作,但如果我想访问config
变量本身呢?
我不确定如何在initConfig
方法之外公开config
变量而不将其转换为包全局变量。我希望能够示例化多个命令树,每个命令树都有自己独立的Viper配置对象,但我不清楚将其放在哪里。
1条答案
按热度按时间bq8i3lrv1#
cobra团队最近已经完成了这项工作,请参见https://github.com/spf13/cobra/pull/1551