go語言 從命令行獲取參數(shù)解析

go語言內(nèi)置的flag包實(shí)現(xiàn)了命令行參數(shù)的解析，flag包使得開發(fā)命令行工具更為簡單。

os.Args

如果你只是簡單的想要獲取命令行參數(shù)，可以像下面的示例代碼一樣使用os.Args來獲取命令行參數(shù)

package main
import "fmt"
import "os"
func main(){
    if len(os.Args)>0 {
        for index , value := range os.Args {
            fmt.Println(index, value)
        }
    }
}

將上面的代碼執(zhí)行g(shù)o build -o "args_demo"編譯之后，執(zhí)行：

0 ./args_demo
1 a
2 b
3 c
4 d

os.Args是一個存儲命令行參數(shù)的字符串切片，它的第一個元素是執(zhí)行文件的名稱。

flag包的基本使用

本文介紹了flag包的常用函數(shù)以及基本用法，更詳細(xì)的內(nèi)容請參考官方文檔。

導(dǎo)入flag包

import flag

flag參數(shù)類型

flag包支持的命令行參數(shù)類型有bool、int、int64、uint、uint64、float、float64、string、duration.

flag.Type()

基本格式如下：
flag.Type(flag名，默認(rèn)值，幫助信息)*Type 例如我們要定義姓名、年齡、婚否三個命令行參數(shù)，我們可以按照如下定義：

    name := flag.String("name","ali","姓名")
    age := flag.Int("age",18,"年齡")
    married := flag.Bool("married",false,"婚否")
    delay := flag.Duration("d",0,"時間間隔")

需要注意的是，此時的 name,age,married,delay均為對應(yīng)類型的指針。

flag.TypeVar()

基本格式如下：flag.TypeVar(Type指針，flag名，默認(rèn)值，幫助信息)例如我們要定義姓名、年齡、婚否三個命令行參數(shù)，我們可以按照如下方式定義：

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參數(shù)后，需要通過調(diào)用flag.Parse()來對命令行參數(shù)進(jìn)行解析。
支持的命令行參數(shù)格式有一下幾種：

• -flag xxx (使用空格，一個 - 符號）
• --flag xxx (使用空格，兩個 - 符號)
• -flag=xxx （使用等號， 一個 - 符號）
• --flag = xxx (使用等號， 兩個- 符號)

其中，布爾類型的參數(shù)必須用等號的方式指定。
flag在解析第一個非flag參數(shù)之前停止，或者在終止符"-"之后停止。

flag其他函數(shù)

flag.Args()  //返回命令行參數(shù)后的其他參數(shù)，以[]string類型
flag.NArg() //返回命令行參數(shù)后的其他參數(shù)個數(shù)
flag.NFlag() //返回使用命令行參數(shù)個數(shù)

完整示例

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參數(shù):

 ./args_demo --name 霍帥兵 --age 18 --married=false -d 1h40m
霍帥兵 18 false 1h40m0s
[]
0
4

使用非flag命令行參數(shù)：

./args_demo a b c
張三 18 false 0s
[a b c]
3
0