Go 编程实例【命令行子命令 Subcommands】

本是古典何须时尚 2023-10-08 23:09 67阅读 0赞

#### 阅读目录 ####

go 和 git 这种命令行工具，都有很多的 子命令。

并且每个工具都有一套自己的 flag，比如： go build 和 go get 是 go 里面的两个不同的子命令。

flag 包让我们可以轻松的为工具定义简单的子命令。

// command-line-subcommands.go
    package main
    
    import (
    	"flag"
    	"fmt"
    	"os"
    )
    
    func main() {
        
    
    	// 我们使用 `NewFlagSet` 函数声明一个子命令，
    	// 然后为这个子命令定义一个专用的 flag。
    	fooCmd := flag.NewFlagSet("foo", flag.ExitOnError)
    	fooEnable := fooCmd.Bool("enable", false, "enable")
    	fooName := fooCmd.String("name", "", "name")
    
    	// 对于不同的子命令，我们可以定义不同的 flag。
    	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] {
        
    
    	// 每个子命令，都会解析自己的 flag 并允许它访问后续的参数。
    	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)
    	}
    }

[root@bogon test]# go run main.go foo -enable -name=joe a1 a2
    subcommand 'foo'
      enable: true
      name: joe
      tail: [a1 a2]
    [root@bogon test]#

然后试一下 bar 子命令。

[root@bogon test]# go run main.go bar -level 8 a1
    subcommand 'bar'
      level: 8
      tail: [a1]
    [root@bogon test]#

但是 bar 不接受 foo 的 flag（enable）。

[root@bogon test]# go run main.go bar -enable a1
    flag provided but not defined: -enable
    Usage of bar:
      -level int
            level
    exit status 2
    [root@bogon test]#