0

这是一个片段my-tool/cmd/root.go

func Execute() {
    if err := rootCmd.Execute(); err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
}

func init() {
    cobra.OnInitialize(initConfig)

    // Here you will define your flags and configuration settings.
    // Cobra supports persistent flags, which, if defined here,
    // will be global for your application.

    rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.my-tool.yaml)")

    // Cobra also supports local flags, which will only run
    // when this action is called directly.
    rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}

// initConfig reads in config file and ENV variables if set.
func initConfig() {
    if cfgFile != "" {
        fmt.Println("Config file set")
        // Use config file from the flag.
        viper.SetConfigFile(cfgFile)
    } else {
        fmt.Println("Config file NOT set")
        // Find home directory.
        home, err := homedir.Dir()
        if err != nil {
            fmt.Println(err)
            os.Exit(1)
        }

代码已经从cobracli的脚手架过程中生成，即通过~/go/cobra/init my-tool --pkg-name github.com/something/my-tool

我正在尝试暂时传递config标志以检查程序是否正在处理它：

▶ go run main.go  --config "test"

但是，尽管我希望该init()函数能够调用cobra.OnInitialize(initConfig)并解析标志，如以下行所示：

    rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.my-tool.yaml)")

最后在if声明中看到这两条消息之一：

func initConfig() {
    if cfgFile != "" {
        fmt.Println("Config file set")
        // Use config file from the flag.
        viper.SetConfigFile(cfgFile)
    } else {
        fmt.Println("Config file NOT set")
        // Find home directory.
        home, err := homedir.Dir()
        if err != nil {
            fmt.Println(err)
            os.Exit(1)
        }

相反，我得到的只是根命令的help消息；这是为什么？

编辑：从我通过添加一些打印语句看到的initConfig()情况来看，从未调用过（出于某种原因），即好像cobra.OnInitialize(initConfig)什么都不做。

4

1 回答 1

2

您需要先指定您的命令

▶ 运行 main.go "yourcommand" --config "test"

看：

» go run main.go --config "blah"
A longer description that spans multiple lines and likely contains
examples and usage of using your application. For example:
...

» go run main.go preview --config "blah"
Error:  open : no such file or directory
exit status 1
于 2019-12-05T15:46:51.843 回答