将pflag . flag . value转换为任意类型

Question

我使用github.com/spf13/cobra来解析命令行标志，我所做的一切都工作得很好，但是我很难弄清楚如何在运行时从*cobra.Command中获得特定类型的值。

我的命令定义在cmd/generate.go中看起来像这样:

var (
generateCommand = &cobra.Command{
// ...
}
generateTokenCommand = &cobra.Command{
// ...
Run: func(cmd *cobra.Command, args []string) {
// command body here
}
}
)
func init() {
generateCommand.AddCommand(generateTokenCommand)
// setup generate token flags
flags := generateTokenCommand.Flags()
flags.Uint64P("user-id", "i", 1, "User ID")
flags.BoolP("is-admin", "a", "Is admin.")
// other flags.StringP, but those are easy
}

注意:我知道有api可以将这些绑定到静态变量，如果可能的话，我尽量避免这样做，以保持Cobra中包含的内容。

在generateTokenCommand.Run内部，例如我的入口点，我在从cobra.Command中获取非字符串值时遇到麻烦。

当我需要一个字符串值时，我可以简单地:

var email string = cmd.Flag("email").Value.String()

pflags.Flag有一个名为Value的字段，其类型为pflags.Value，如下所示:

// Value is the interface to the dynamic value stored in a flag.
// (The default value is represented as a string.)
type Value interface {
String() string
Set(string) error
Type() string
}

如果我进入调试器并像这样抓取值:

userIdValue := cmd.Flag("user-id").Value

我看到这是*github.com/spf13/pflag.uint64Value的运行时类型，但由于这是一个私有类型，我似乎无法转换值:

// does not compile
cmd.Flag("user-id").Value.(uint64)

如何转换或使用其他方法将此Value解析为uint64?

Answer 1

我可能在深入研究后找到了答案，pflags.FlagSet上有一个叫做GetUint64的方法:

// GetUint64 return the uint64 value of a flag with the given name
func (f *FlagSet) GetUint64(name string) (uint64, error) {
val, err := f.getFlagType(name, "uint64", uint64Conv)
if err != nil {
return 0, err
}
return val.(uint64), nil
}

我认为我的问题是习惯了文档和Golang源代码的组织方式，但这确实有效:

uid, err := cmd.Flags().GetUint64("user-id")