登录
首页 >  Golang >  Go问答

golang flag 包子命令如何确定实际输入的 flag?

来源:SegmentFault

时间:2023-01-08 18:40:07 151浏览 收藏

亲爱的编程学习爱好者,如果你点开了这篇文章,说明你对《golang flag 包子命令如何确定实际输入的 flag?》很感兴趣。本篇文章就来给大家详细解析一下,主要介绍一下go,希望所有认真读完的童鞋们,都有实质性的提高。

问题内容

例如 fooCmd 子命令有多个 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
X()
> go run test.go foo -a 10
A()
> go run test.go foo -a 10 -b 20
AB()

正确答案

使用 visit() 方法遍历 flag

    fooCmd := flag.NewFlagSet("foo", flag.ExitOnError)
    fooA := fooCmd.String("a", "-1", "a")
    fooB := fooCmd.String("b", "-1", "b")
    fooC := fooCmd.Bool("c", false, "c")
    // ... flag 可以有多个

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

    fmt.Println("a:", *fooA)
    fmt.Println("b:", *fooB)
    fmt.Println("c:", *fooC)

    // 遍历定义好的 flag
    fooCmd.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())
        }
    })

    fmt.Println()

    // 遍历实际传入的 flag
    fooCmd.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())
        }
    })

    // 额外的参数
    fmt.Println(fooCmd.Args())

>go run test.go foo -a 10 -c xxx

a: 10
b: -1
c: true

a = 10 [default = -1]
b = -1
c = true [default = false]

a = 10 [default = -1]
c = true [default = false]

[xxx]

不知道还有没有更好的方法了

以上就是本文的全部内容了,是否有顺利帮助你解决问题?若是能给你带来学习上的帮助,请大家多多支持golang学习网!更多关于Golang的相关知识,也可关注golang学习网公众号。

声明:本文转载于:SegmentFault 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>
评论列表