
使用 golang flag 包判断子命令中的实际输入 flag
当一个子命令有多个 flag 时,我们需要区分只输入特定 flag 的场景。本文介绍如何使用 flag 包的 visit() 方法确定实际输入的 flag。
首先,创建子命令的 flag 集合:
foocmd := flag.newflagset("foo", flag.exitonerror)
fooa := foocmd.string("a", "0", "a")
foob := foocmd.string("b", "0", "b")解析命令行参数:
立即学习“go语言免费学习笔记(深入)”;
foocmd.parse(os.args[2:])
遍历定义好的 flag:
foocmd.visitall(func(f *flag.flag) {
// ... 判断逻辑
})遍历实际输入的 flag:
foocmd.visit(func(f *flag.flag) {
// ... 判断逻辑
})通过遍历,我们可以判断是否只输入了特定 flag,从而实现不同的功能。
例如,以下代码演示了判断只输入了 a flag 的场景:
aFlagVisited := false
fooCmd.Visit(func(f *flag.Flag) {
if f.Name == "a" {
aFlagVisited = true
}
})
if aFlagVisited && !bFlagVisited {
A()
}










