go語言 從命令列獲取引數解析
go語言內建的flag包實現了命令列引數的解析,flag包使得開發命令列工具更為簡單。
os.Args
如果你只是簡單的想要獲取命令列引數,可以像下面的示例程式碼一樣使用os.Args來獲取命令列引數
package main import "fmt" import "os" func main(){ if len(os.Args)>0 { for index , value := range os.Args { fmt.Println(index, value) } } }
將上面的程式碼執行go build -o "args_demo"編譯之後,執行:
0 ./args_demo 1 a 2 b 3 c 4 d
os.Args是一個儲存命令列引數的字串切片,它的第一個元素是執行檔案的名稱。
flag包的基本使用
本文介紹了flag包的常用函式以及基本用法,更詳細的內容請參考官方文件。
匯入flag包
import flag
flag引數型別
flag包支援的命令列引數型別有bool、int、int64、uint、uint64、float、float64、string、duration.
flag.Type()
基本格式如下:
flag.Type(flag名,預設值,幫助資訊)*Type 例如我們要定義姓名、年齡、婚否三個命令列引數,我們可以按照如下定義:
name := flag.String("name","ali","姓名") age := flag.Int("age",18,"年齡") married := flag.Bool("married",false,"婚否") delay := flag.Duration("d",0,"時間間隔")
需要注意的是,此時的 name,age,married,delay均為對應型別的指標。
flag.TypeVar()
基本格式如下:flag.TypeVar(Type指標,flag名,預設值,幫助資訊)例如我們要定義姓名、年齡、婚否三個命令列引數,我們可以按照如下方式定義:
var name string var age int var married bool var delay time.Duration flag.StringVar(&name,"name","張三","姓名") flag.IntVar(&age,"age",18,"年齡") flag.BoolVar(&married,"married",false,"婚否") flag.Duration(&delay,"d",0,"時間間隔")
flag.Parse()
通過以上兩種方法定義命令列flag引數後,需要通過呼叫flag.Parse()來對命令列引數進行解析。
支援的命令列引數格式有一下幾種:
- -flag xxx(使用空格,一個 - 符號)
- --flag xxx (使用空格,兩個 - 符號)
- -flag=xxx (使用等號, 一個 - 符號)
- --flag = xxx (使用等號, 兩個- 符號)
其中,布林型別的引數必須用等號的方式指定。
flag在解析第一個非flag引數之前停止,或者在終止符"-"之後停止。
flag其他函式
flag.Args()//返回命令列引數後的其他引數,以[]string型別 flag.NArg() //返回命令列引數後的其他引數個數 flag.NFlag() //返回使用命令列引數個數
完整示例
package main import ( "fmt" "flag" "time" ) func main(){ var name string var age int var married bool var delay time.Duration flag.StringVar(&name,"name","張三","姓名") flag.IntVar(&age,"age",18,"年齡") flag.BoolVar(&married,"married",false,"婚否") flag.DurationVar(&delay, "d", 0, "延遲的時間間隔") flag.Parse() fmt.Println(name,age,married,delay) fmt.Println(flag.Args()) fmt.Println(flag.NArg()) fmt.Println(flag.NFlag()) }
正常使用命令列flag引數:
./args_demo --name 霍帥兵 --age 18 --married=false -d 1h30m 霍帥兵 18 false 1h30m0s [] 0 4
使用非flag命令列引數:
./args_demo a b c 張三 18 false 0s [a b c] 3 0