Go 示例: 命令行子命令

一些命令行工具，例如 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)
    }
}
 
$ go build command-line-subcommands.go

首先调用 foo 子命令。

 
$ ./command-line-subcommands foo -enable -name=joe a1 a2
subcommand 'foo'
  enable: true
  name: joe
  tail: [a1 a2]

现在试试 bar。

 
$ ./command-line-subcommands bar -level 8 a1
subcommand 'bar'
  level: 8
  tail: [a1]

但 bar 不会接受 foo 的标志。

 
$ ./command-line-subcommands bar -enable a1
flag provided but not defined: -enable
Usage of bar:
  -level int
        level

接下来我们将看看环境变量，这是参数化程序的另一种常见方法。

下一个示例：环境变量.

作者：Mark McGranaghan 和 Eli Bendersky | 源代码 | 许可证