GO學習筆記 - 命令列解析
阿新 • • 發佈：2019-07-22
本文主題：基於os.Args與flag實現Golang命令列解析。
小慢哥的原創文章，歡迎轉載
目錄
▪ 一. os.Args
▪ 二. flag
▪ 三. 結合os.Args與flag實現子命令
▪ 附. 參考文件
一. os.Args
os.Args用於獲取通過命令列傳入的引數
▷ os.Args[0]：程式執行路徑
▷ os.Args[1]：第1個引數
▷ os.Args[2]：第2個引數
▷ len(os.Args)：引數數量
示例如下
package main import ( "fmt" "os" ) func main() { fmt.Println("Program:", os.Args[0]) for i, v := range os.Args[1:] { fmt.Printf("Arg[%d]: %v\n", i, v) } }
測試效果
# 編譯執行
go build arg.go
./arg foo "hello world" bar
# 輸出
Program: ./arg
Arg[0]: foo
Arg[1]: hello world
Arg[2]: bar二. flag
golang內建的flag模組可以實現命令列flag解析。
什麼是flag，舉例說明（加粗部分就是flag）：
▷ wc -l
▷ ./configure --prefix=/usr/local/nginx
示例如下（摘取自gobyexample）
package main import ( "flag" "fmt" ) func main() { // flag.String返回的是指標 // word表示引數名，foo表示預設值，a string表示引數釋義（在-h或解析異常時候會看到） // flag.Int、flag.Bool與flag.String同理，不再贅述 wordPtr := flag.String("word", "foo", "a string") numbPtr := flag.Int("numb", 42, "an int") boolPtr := flag.Bool("fork", false, "a bool") // flag.StringVar返回的是非指標 var svar string flag.StringVar(&svar, "svar", "bar", "a string var") // 進行flag解析 flag.Parse() fmt.Println("word:", *wordPtr) fmt.Println("numb:", *numbPtr) fmt.Println("fork:", *boolPtr) fmt.Println("svar:", svar) fmt.Println("tail:", flag.Args()) }
測試效果
# 編譯 go build flag.go ./flag -word=opt -numb=7 -fork -svar=flag # 輸出 word: opt numb: 7 fork: true svar: flag tail: [] ./flag -word=opt # 輸出 word: opt numb: 42 fork: false svar: bar tail: [] ./flag -word=opt a1 a2 a3 # 輸出 word: opt numb: 42 fork: false svar: bar tail: [a1 a2 a3] ./flag -word=opt a1 a2 a3 -numb=7 # 輸出 word: opt numb: 42 fork: false svar: bar tail: [a1 a2 a3 -numb=7] ./flag -h # 輸出 Usage of ./flag: -fork a bool -numb int an int (default 42) -svar string a string var (default "bar") -word string a string (default "foo") ./flag -wat # 輸出 flag provided but not defined: -wat Usage of ./flag: ...
該例子，在測試時候只使用了“單橫線+引數名”，實際有4種方式，效果是相同的
-word opt
-word=opt
--word opt
--word=opt三. 結合os.Args與flag實現子命令
如果想實現子命令效果，例如：
▷ docker ps --last 10
▷ docker images --help
這需要將os.Args和flag結合使用。基本原理：建立多個flag集合（即*flag.FlagSet），每個flag集合對應不同的os.Args
示例如下（摘取自gobyexample）
package main
import (
"flag"
"fmt"
"os"
)
func main() {
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 subflag.go
./subflag foo -enable -name=joe a1 a2
# 輸出
subcommand 'foo'
enable: true
name: joe
tail: [a1 a2]
./subflag bar -level 8 a1
# 輸出
subcommand 'bar'
level: 8
tail: [a1]
./subflag bar -enable a1
# 輸出
flag provided but not defined: -enable
Usage of bar:
-level int
level附. 參考文件
# os.Args
https://gobyexample.com/command-line-arguments
# flag
https://gobyexample.com/command-line-flags
# 子命令
https://gobyexample.com/command-line-subcommands