一些命令行工具(如 go
工具或 git
)有许多子命令,每个子命令都有自己的一组标志。例如,go build
和 go get
是 go
工具的两个不同子命令。flag
包让我们可以轻松定义具有自己标志的简单子命令。
package main
import (
"flag"
"fmt"
"os"
)
func main() {
// 我们使用 NewFlagSet 函数声明一个子命令,并继续定义特定于该子命令的新标志。
fooCmd := flag.NewFlagSet("foo", flag.ExitOnError)
fooEnable := fooCmd.Bool("enable", false, "enable")
fooName := fooCmd.String("name", "", "name")
// 对于不同的子命令,我们可以定义不同的支持标志。
barCmd := flag.NewFlagSet("bar", flag.ExitOnError)
barLevel := barCmd.Int("level", 0, "level")
// 子命令应作为程序的第一个参数传入。
if len(os.Args) < 2 {
fmt.Println("expected 'foo' or 'bar' subcommands")
os.Exit(1)
}
// 检查调用了哪个子命令。
switch os.Args[1] {
// 对于每个子命令,我们解析其自己的标志,并可以访问尾随的位置参数。
case "foo":
fooCmd.Parse(os.Args[2:])
fmt.Println("subcommand 'foo'")
fmt.Println(" enable:", *fooEnable)
fmt.Println(" name:", *fooName)
fmt.Println(" tail:", fooCmd.Args())
case "bar":
barCmd.Parse(os.Args[2:])
fmt.Println("subcommand 'bar'")
fmt.Println(" level:", *barLevel)
fmt.Println(" tail:", barCmd.Args())
default:
fmt.Println("expected 'foo' or 'bar' subcommands")
os.Exit(1)
}
}
运行结果:
i/go-example/command-line-subcommands via 🐹 v1.22.1
➜ go run command-line-subcommands.go foo -enable -name=joe a1 a2
subcommand 'foo'
enable: true
name: joe
tail: [a1 a2]
i/go-example/command-line-subcommands via 🐹 v1.22.1
➜ go run command-line-subcommands.go bar -level 8 a1
subcommand 'bar'
level: 8
tail: [a1]
i/go-example/command-line-subcommands via 🐹 v1.22.1
➜ go run command-line-subcommands.go bar -enable a1
flag provided but not defined: -enable
Usage of bar:
-level int
level
exit status 2
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。