在 Go 语言的 flag 包中，如何确定用户实际输入了哪些标志？

flag 包中根据实际输入确定已设置标志的方法

在 golang 中，flag 包提供了命令行解析功能。在解析子命令时，如何确定用户实际输入了哪些标志呢？

一种方法是使用 visit() 方法遍历所有定义的标志，并检查每个标志的实际值是否与默认值不同。如果不同，则表示该标志已被用户设置。

// 遍历所有定义的 flag
cmd.visitall(func(f *flag.flag) {
    if f.value.string() != f.defvalue {
        fmt.printf("%s = %s [default = %s]\n", f.name, f.value.string(), f.defvalue)
    } else {
        fmt.printf("%s = %s\n", f.name, f.value.string())
    }
})

除了检查已定义的标志外，还可以使用 visit() 方法遍历用户实际输入的标志。它只会遍历那些实际被设置过的标志，忽略默认值。

Img.Upscaler

免费的AI图片放大工具

下载

// 遍历实际输入的 flag
cmd.visit(func(f *flag.flag) {
    if f.value.string() != f.defvalue {
        fmt.printf("%s = %s [default = %s]\n", f.name, f.value.string(), f.defvalue)
    } else {
        fmt.printf("%s = %s\n", f.name, f.value.string())
    }
})

这样，就可以根据实际输入的标志来决定下一步的操作，例如执行不同的函数或提供不同类型的输出。

例如，在下面这个程序中，根据实际输入的 -a、-b 和 -c 标志的值，不同的函数将被调用。

import (
    "fmt"
    "os"

    "github.com/spf13/cobra"
)

var fooCmd = &cobra.Command{
    Use:   "foo",
    Short: "Foo command",
    Args:  cobra.MinimumNArgs(0),
    Run: func(cmd *cobra.Command, args []string) {
        a := cmd.Flag("a").Value.String()
        b := cmd.Flag("b").Value.String()

        if a != "" && b != "" {
            AB()
        } else if a != "" {
            A()
        } else if b != "" {
            B()
        } else {
            X()
        }
    },
}

func init() {
    fooCmd.Flags().StringP("a", "a", "", "A flag")
    fooCmd.Flags().StringP("b", "b", "", "B flag")
}

func A() {
    fmt.Println("Function A")
}

func B() {
    fmt.Println("Function B")
}

func AB() {
    fmt.Println("Function AB")
}

func X() {
    fmt.Println("Function X")
}

func main() {
    rootCmd := &cobra.Command{
        Use:   "test",
        Short: "Test command",
    }

    rootCmd.AddCommand(fooCmd)

    rootCmd.Execute()
}