Go语言flag包：如何根据子命令参数组合执行不同操作？

go 语言 flag 包子命令参数判断

在使用 go 语言的 flag 包编写命令行工具时，经常需要判断用户实际输入的子命令参数。本文将详细阐述如何确定 foocmd 子命令中 -a 和 -b 参数的输入情况，并根据不同的输入组合执行相应的操作。

假设我们有一个名为 foocmd 的子命令，它包含 -a 和 -b 两个 flag：

foocmd := flag.newflagset("foo", flag.exitonerror)
fooa := foocmd.string("a", "0", "a")
foob := foocmd.string("b", "0", "b")
// ... flag 可以有多个

foocmd.parse(os.args[2:])

// 如何判断？
if 仅有 a flag {
    a()
} else if 仅有 b flag {
    b()
} else if 同时有 a 和 b {
    ab()
} else {
    x()
}

我们需要根据用户输入的参数组合，例如 go run test.go foo、go run test.go foo -a 10、go run test.go foo -a 10 -b 20 等，来判断应该执行 a()、b()、ab() 还是 x() 函数。

一种有效的解决方法是利用 flag 包提供的 visit() 和 visitall() 方法遍历标志。visitall() 方法会遍历所有已定义的标志，而 visit() 方法只遍历已设置的标志。通过比较标志的当前值和默认值，我们可以判断用户是否设置了该标志。

下面是一个示例代码，演示如何使用 visit() 方法判断 -a 和 -b 参数的输入情况：

fooCmd := flag.NewFlagSet("foo", flag.ExitOnError)
fooA := fooCmd.String("a", "-1", "a")
fooB := fooCmd.String("b", "-1", "b")
fooC := fooCmd.Bool("c", false, "c") //增加一个bool类型的flag用于演示

fooCmd.Parse(os.Args[2:])

// ... (省略打印fooA, fooB, fooC的代码)...

// 遍历实际传入的 flag
aSet := false
bSet := false
fooCmd.Visit(func(f *flag.Flag) {
    if f.Name == "a" && f.Value.String() != f.DefValue {
        aSet = true
    }
    if f.Name == "b" && f.Value.String() != f.DefValue {
        bSet = true
    }
})

if aSet && !bSet {
    A()
} else if !aSet && bSet {
    B()
} else if aSet && bSet {
    AB()
} else {
    X()
}

这段代码首先定义了两个布尔变量 aset 和 bset，初始值为 false。然后使用 visit() 方法遍历实际传入的 flag，如果 -a 或 -b 的值与默认值不同，则将对应的布尔变量设置为 true。最后根据 aset 和 bset 的值执行不同的操作。 通过这种方式，我们可以准确地判断用户输入的 flag 组合，并执行相应的操作。 需要注意的是，此方法需要预先知道所有可能存在的flag名称，才能进行精确判断。 如果flag数量较多且名称不固定，可能需要采用其他的方法，例如解析os.args数组。