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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章