Golang flag包的具體使用
在 Golang 程序中有很多種方法來處理命令行參數(shù)。簡單的情況下可以不使用任何庫，直接處理 os.Args；其實 Golang 的標準庫提供了 flag 包來處理命令行參數(shù)；還有第三方提供的處理命令行參數(shù)的庫，比如 Pflag 等。本文將介紹 Golang 標準庫中 flag 包的用法。本文的演示環(huán)境為 ubuntu 18.04。
入門 demo
在 Go workspace 的 src 目錄下創(chuàng)建 flagdemo 目錄，并在目錄下創(chuàng)建 main.go 文件，編輯其內(nèi)容如下：
package main
import "flag"
import "fmt"
// 定義命令行參數(shù)對應(yīng)的變量，這三個變量都是指針類型
var cliName = flag.String("name", "nick", "Input Your Name")
var cliAge = flag.Int("age", 28, "Input Your Age")
var cliGender = flag.String("gender", "male", "Input Your Gender")
// 定義一個值類型的命令行參數(shù)變量，在 Init() 函數(shù)中對其初始化
// 因此，命令行參數(shù)對應(yīng)變量的定義和初始化是可以分開的
var cliFlag int
func Init() {
flag.IntVar(&cliFlag, "flagname", 1234, "Just for demo")
}
func main() {
// 初始化變量 cliFlag
Init()
// 把用戶傳遞的命令行參數(shù)解析為對應(yīng)變量的值
flag.Parse()
// flag.Args() 函數(shù)返回沒有被解析的命令行參數(shù)
// func NArg() 函數(shù)返回沒有被解析的命令行參數(shù)的個數(shù)
fmt.Printf("args=%s, num=%d\n", flag.Args(), flag.NArg())
for i := 0; i != flag.NArg(); i++ {
fmt.Printf("arg[%d]=%s\n", i, flag.Arg(i))
}
// 輸出命令行參數(shù)
fmt.Println("name=", *cliName)
fmt.Println("age=", *cliAge)
fmt.Println("gender=", *cliGender)
fmt.Println("flagname=", cliFlag)
}使用 flag 包前要通過 import 命令導入該包：
import "flag"
定義一個整型的參數(shù) age，返回指針類型的變量：
var cliAge = flag.Int("age", 28, "Input Your Age")創(chuàng)建值類型的參數(shù)變量，并在 Init() 函數(shù)中對其初始化(注意這里調(diào)用的是 flag.IntVar 方法)：
var cliFlag int
func Init() {
flag.IntVar(&cliFlag, "flagname", 1234, "Just for demo")
}通過 flag.Parse() 函數(shù)接下命令行參數(shù)，解析函數(shù)將會在碰到第一個非 flag 命令行參數(shù)時停止：
flag.Parse()
命令行傳參的格式：
-isbool (一個 - 符號，布爾類型該寫法等同于 -isbool=true) -age=x (一個 - 符號，使用等號) -age x (一個 - 符號，使用空格) --age=x (兩個 - 符號，使用等號) --age x (兩個 - 符號，使用空格)
運行 demo
在 flagdemo 目錄下執(zhí)行 go build 命令編譯 demo 生成可執(zhí)行文件 flagdemo。
不傳遞命令行參數(shù)
此時輸出的命令行參數(shù)都是定義的默認值。
傳遞命令行參數(shù)
傳遞的命令行參數(shù)會覆蓋默認值。
傳遞多余的命令行參數(shù)
可以通過 flag.Args() 和 flag.NArg() 函數(shù)獲取未能解析的命令行參數(shù)。
傳遞錯誤的命令行參
如果通過 -xx 傳入未定義的命令行參數(shù)，則會直接報錯退出，并輸出幫助信息。
查看幫助信息
通過命令行參數(shù) -h 或 --help 可以查看幫助信息：
解讀 flag 包源碼
flag 包支持的類型有 Bool、Duration、Float64、Int、Int64、String、Uint、Uint64。這些類型的參數(shù)被封裝到其對應(yīng)的后端類型中，比如 Int 類型的參數(shù)被封裝為 intValue，String 類型的參數(shù)被封裝為 stringValue。這些后端的類型都實現(xiàn)了 Value 接口，因此可以把一個命令行參數(shù)抽象為一個 Flag 類型的實例。下面是 Value 接口和 Flag 類型的代碼：
// Value 接口
type Value interface {
String() string
Set(string) error
}
// Flag 類型
type Flag struct {
Name string // name as it appears on command line
Usage string // help message
Value Value // value as set 是個 interface，因此可以是不同類型的實例。
DefValue string // default value (as text); for usage message
}intValue 等類型實現(xiàn)了 Value 接口，因此可以賦值給 Flag 類型中的 Value 字段，下面是 intValue 類型的定義：
// -- int Value
type intValue int
func newIntValue(val int, p *int) *intValue {
*p = val
return (*intValue)(p)
}
func (i *intValue) Set(s string) error {
v, err := strconv.ParseInt(s, 0, strconv.IntSize)
*i = intValue(v)
return err
}
func (i *intValue) Get() interface{} { return int(*i) }
func (i *intValue) String() string { return strconv.Itoa(int(*i)) }所有的參數(shù)被保存在 FlagSet 類型的實例中，F(xiàn)lagSet 類型的定義如下：
// A FlagSet represents a set of defined flags.
type FlagSet struct {
Usage func()
name string
parsed bool
actual map[string]*Flag // 中保存從命令行參數(shù)中解析到的參數(shù)實例
formal map[string]*Flag // 中保存定義的命令行參數(shù)實例(實例中包含了默認值)
args []string // arguments after flags
errorHandling ErrorHandling
output io.Writer // nil means stderr; use out() accessor
}Flag 包被導入時創(chuàng)建了 FlagSet 類型的對象 CommandLine：
var CommandLine = NewFlagSet(os.Args[0], ExitOnError)
在程序中定義的所有命令行參數(shù)變量都會被加入到 CommandLine 的 formal 屬性中，其具體的調(diào)用過程如下：
var cliAge = flag.Int("age", 28, "Input Your Age")
func Int(name string, value int, usage string) *int {
return CommandLine.Int(name, value, usage)
}
func (f *FlagSet) Int(name string, value int, usage string) *int {
p := new(int)
f.IntVar(p, name, value, usage)
return p
}
func (f *FlagSet) IntVar(p *int, name string, value int, usage string) {
f.Var(newIntValue(value, p), name, usage)
}
func (f *FlagSet) Var(value Value, name string, usage string) {
// Remember the default value as a string; it won't change.
flag := &Flag{name, usage, value, value.String()}
_, alreadythere := f.formal[name]
if alreadythere {
var msg string
if f.name == "" {
msg = fmt.Sprintf("flag redefined: %s", name)
} else {
msg = fmt.Sprintf("%s flag redefined: %s", f.name, name)
}
fmt.Fprintln(f.Output(), msg)
panic(msg) // Happens only if flags are declared with identical names
}
if f.formal == nil {
f.formal = make(map[string]*Flag)
}
// 把命令行參數(shù)對應(yīng)的變量添加到 formal 中
f.formal[name] = flag
}命令行參數(shù)的解析過程則由 flag.Parse() 函數(shù)完成，其調(diào)用過程大致如下：
func Parse() {
CommandLine.Parse(os.Args[1:])
}
func (f *FlagSet) Parse(arguments []string) error {
f.parsed = true
f.args = arguments
for {
seen, err := f.parseOne()
if seen {
continue
}
if err == nil {
break
}
switch f.errorHandling {
case ContinueOnError:
return err
case ExitOnError:
os.Exit(2)
case PanicOnError:
panic(err)
}
}
return nil
}最終由 FlagSet 的 parseOne() 方法執(zhí)行解析任務(wù)：
func (f *FlagSet) parseOne() (bool, error) {
…
flag.Value.Set(value)
…
f.actual[name] = flag
…
}總結(jié)
本文介紹了 Golang 標準庫中 flag 包的基本用法，并進一步分析了其主要的代碼邏輯。其實 flag 包還支持用戶自定義類型的命令行參數(shù)，本文不再贅述，有興趣的朋友請參考官方 demo。
參考：
package flag
Go by Example: Command-Line Flags
USING COMMAND LINE FLAGS IN GO
Golang之使用Flag和Pflag
Go語言學習之flag包(The way to go)
Golang flag demo
作者：sparkdev
出處：http://www.cnblogs.com/sparkdev/
到此這篇關(guān)于Golang flag包的具體使用的文章就介紹到這了,更多相關(guān)Golang flag包內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了Go Web 編程中的模板庫應(yīng)用指南，本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-03-03
fmt 是 Go 語言中的一個常用標準庫,它用于格式化輸入和輸出數(shù)據(jù),這篇文章主要為大家介紹了fmt的基本使用,感興趣的小伙伴可以跟隨小編一起學習一下2023-10-10
Goland是一個用于Go語言開發(fā)的IDE,Goland的項目結(jié)構(gòu)與Go語言的項目結(jié)構(gòu)相似,下面這篇文章主要給大家介紹了關(guān)于Goland項目使用gomod配置的詳細步驟,文中通過圖文介紹的非常詳細,需要的朋友可以參考下2023-04-04
這篇文章主要介紹了Golang中泛型的使用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-06-06
分布式鎖是控制分布式系統(tǒng)之間同步訪問共享資源的一種方式。如果不同的系統(tǒng)或是同一個系統(tǒng)的不同主機之間共享了一個或一組資源，那么訪問這些資源時，需要通過一些互斥手段來防止彼此之間的干擾以保證一致性，在這種情況下，就需要使用分布式鎖了2023-01-01
go語言開發(fā)環(huán)境配置(sublime text3+gosublime)
網(wǎng)上google了下go的開發(fā)工具，大都推薦sublime text3+gosublime，本文就介紹了go語言開發(fā)環(huán)境配置(sublime text3+gosublime)，具有一定的參考價值，感興趣的可以了解一下2022-01-01
這篇文章我們主要介紹垃圾收集器的設(shè)計原理以及Golang垃圾收集器的實現(xiàn)原理，文中有詳細的代碼示例及圖文介紹，感興趣的小伙伴跟著小編一起來學習吧2023-06-06
在 Go 語言中,實現(xiàn)一個強大的重試機制可以通過多種方式來完成,這篇文章為大家介紹了一個常見的實現(xiàn)方法,需要的小伙伴可以參考一下2025-03-03
Go語言中g(shù)oroutine和WaitGroup的使用示例詳解
goroutine 是Go中一個輕量級的線程， 只需要一個go關(guān)鍵字就可以創(chuàng)建一個goroutine，這篇文章主要介紹了Go語言中g(shù)oroutine和WaitGroup的使用,需要的朋友可以參考下2023-03-03
Golang 語言極簡類型轉(zhuǎn)換庫cast的使用詳解
本文我們通過 cast.ToString() 函數(shù)的使用，簡單介紹了cast 的使用方法，除此之外，它還支持很多其他類型，在這沒有多多介紹，對Golang 類型轉(zhuǎn)換庫 cast相關(guān)知識感興趣的朋友一起看看吧2021-11-11