在 cobra 中确保持久标志只被检索一次

“纵有疾风来，人生不言弃”，这句话送给正在学习Golang的朋友们，也希望在阅读本文《在 cobra 中确保持久标志只被检索一次》后，能够真的帮助到大家。我也会在后续的文章中，陆续更新Golang相关的技术文章，有好的建议欢迎大家在评论留言，非常感谢！

我有以下 cobra 设置

var rootcmd = &cobra.command{
    use:   "basic",
    short: "this is the basic command",
    prerune: func(cmd *cobra.command, args []string) error {
        if err := viper.bindpflags(cmd.flags()); err != nil {
            return err
        }
        return nil
    },
    run: func(cmd *cobra.command, args []string) {
        fmt.println("hello")
        author := viper.getstring("author")
        fmt.println(author)
    },
}

var subcmd1 = &cobra.command{
    use:   "subcommand1",
    short: "this is test subcommand 1",
    prerune: func(cmd *cobra.command, args []string) error {
        if err := viper.bindpflags(cmd.flags()); err != nil {
            return err
        }
        return nil
    },
    run: func(cmd *cobra.command, args []string) {
        fmt.println("executing subcommand 1")
        author := viper.getstring("author")
        location := viper.getstring("location")
        fmt.println(author)
        fmt.println(location)
    },
}

var subcmd2 = &cobra.command{
    use:   "subcommand2",
    short: "this is test subcommand 2",
    prerune: func(cmd *cobra.command, args []string) error {
        if err := viper.bindpflags(cmd.flags()); err != nil {
            return err
        }
        return nil
    },
    run: func(cmd *cobra.command, args []string) {
        fmt.println("executing subcommand 2")
        author := viper.getstring("author")
        duration := viper.getint("duration")
        fmt.println(author)
        fmt.printf("%d\n", duration)
    },
}

func execute() {
    if err := rootcmd.execute(); err != nil {
        fmt.fprintln(os.stderr, err)
        os.exit(1)
    }
}

func init() {
    rootcmd.persistentflags().stringp("author", "a", "pantelis karamolegkos", "author name for copyright attribution")
    subcmd1.flags().stringp("location", "l", "athens", "location of the command execution")
    subcmd2.flags().intp("duration", "d", 10, "duration of the event in minutes")
    rootcmd.addcommand(subcmd1)
    rootcmd.addcommand(subcmd2)
}

我使用 author 作为 persistentflag，以便它在所有子命令中都可用。

我正在尝试找到一种方法来跳过重复

author := viper.GetString("author")

在每个子命令中（我当然希望它在所有子命令中都可用）

虽然这段代码并不复杂，但在我的实际代码中，我有一些 perstitentflags 和几个需要它们的子命令，因此代码库很快就会变得非常重复。

有什么办法可以解决这个问题吗？是否可以在父命令中仅检索持久标志一次并将其传递给子命令？ 此处找不到建议。

谢谢。

正确答案

解决这个问题的一种方法是使用“标志结构”：

type Cmd1Flags struct {
  Author string
  ... // Other persistent flags for cmd1
}

func GetCmd1Flags() Cmd1Flags {
   ret:=Cmd1Flags{}
   ret.Author=viper.GetString("author")
   ... // Initialize other flags
   return ret
}

然后，在所有子命令中使用 getcmd1flags()。

今天关于《在 cobra 中确保持久标志只被检索一次》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！