Golang flag包的具體使用
在 Golang 程序中有很多種方法來處理命令行參數(shù)。簡(jiǎn)單的情況下可以不使用任何庫，直接處理 os.Args；其實(shí) Golang 的標(biāo)準(zhǔn)庫提供了 flag 包來處理命令行參數(shù)；還有第三方提供的處理命令行參數(shù)的庫，比如 Pflag 等。本文將介紹 Golang 標(biāo)準(zhǔn)庫中 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ù)對(duì)應(yīng)的變量，這三個(gè)變量都是指針類型
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")
// 定義一個(gè)值類型的命令行參數(shù)變量，在 Init() 函數(shù)中對(duì)其初始化
// 因此，命令行參數(shù)對(duì)應(yīng)變量的定義和初始化是可以分開的
var cliFlag int
func Init() {
flag.IntVar(&cliFlag, "flagname", 1234, "Just for demo")
}
func main() {
// 初始化變量 cliFlag
Init()
// 把用戶傳遞的命令行參數(shù)解析為對(duì)應(yīng)變量的值
flag.Parse()
// flag.Args() 函數(shù)返回沒有被解析的命令行參數(shù)
// func NArg() 函數(shù)返回沒有被解析的命令行參數(shù)的個(gè)數(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 命令導(dǎo)入該包：
import "flag"
定義一個(gè)整型的參數(shù) age，返回指針類型的變量：
var cliAge = flag.Int("age", 28, "Input Your Age")創(chuàng)建值類型的參數(shù)變量，并在 Init() 函數(shù)中對(duì)其初始化(注意這里調(diào)用的是 flag.IntVar 方法)：
var cliFlag int
func Init() {
flag.IntVar(&cliFlag, "flagname", 1234, "Just for demo")
}通過 flag.Parse() 函數(shù)接下命令行參數(shù)，解析函數(shù)將會(huì)在碰到第一個(gè)非 flag 命令行參數(shù)時(shí)停止：
flag.Parse()
命令行傳參的格式：
-isbool (一個(gè) - 符號(hào)，布爾類型該寫法等同于 -isbool=true) -age=x (一個(gè) - 符號(hào)，使用等號(hào)) -age x (一個(gè) - 符號(hào)，使用空格) --age=x (兩個(gè) - 符號(hào)，使用等號(hào)) --age x (兩個(gè) - 符號(hào)，使用空格)
運(yùn)行 demo
在 flagdemo 目錄下執(zhí)行 go build 命令編譯 demo 生成可執(zhí)行文件 flagdemo。
不傳遞命令行參數(shù)
此時(shí)輸出的命令行參數(shù)都是定義的默認(rèn)值。
傳遞命令行參數(shù)
傳遞的命令行參數(shù)會(huì)覆蓋默認(rèn)值。
傳遞多余的命令行參數(shù)
可以通過 flag.Args() 和 flag.NArg() 函數(shù)獲取未能解析的命令行參數(shù)。
傳遞錯(cuò)誤的命令行參
如果通過 -xx 傳入未定義的命令行參數(shù)，則會(huì)直接報(bào)錯(cuò)退出，并輸出幫助信息。
查看幫助信息
通過命令行參數(shù) -h 或 --help 可以查看幫助信息：
解讀 flag 包源碼
flag 包支持的類型有 Bool、Duration、Float64、Int、Int64、String、Uint、Uint64。這些類型的參數(shù)被封裝到其對(duì)應(yīng)的后端類型中，比如 Int 類型的參數(shù)被封裝為 intValue，String 類型的參數(shù)被封裝為 stringValue。這些后端的類型都實(shí)現(xiàn)了 Value 接口，因此可以把一個(gè)命令行參數(shù)抽象為一個(gè) Flag 類型的實(shí)例。下面是 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 是個(gè) interface，因此可以是不同類型的實(shí)例。
DefValue string // default value (as text); for usage message
}intValue 等類型實(shí)現(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 類型的實(shí)例中，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ù)實(shí)例
formal map[string]*Flag // 中保存定義的命令行參數(shù)實(shí)例(實(shí)例中包含了默認(rèn)值)
args []string // arguments after flags
errorHandling ErrorHandling
output io.Writer // nil means stderr; use out() accessor
}Flag 包被導(dǎo)入時(shí)創(chuàng)建了 FlagSet 類型的對(duì)象 CommandLine：
var CommandLine = NewFlagSet(os.Args[0], ExitOnError)
在程序中定義的所有命令行參數(shù)變量都會(huì)被加入到 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ù)對(duì)應(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 標(biāo)準(zhǔn)庫中 flag 包的基本用法，并進(jìn)一步分析了其主要的代碼邏輯。其實(shí) flag 包還支持用戶自定義類型的命令行參數(shù)，本文不再贅述，有興趣的朋友請(qǐng)參考官方 demo。
參考：
package flag
Go by Example: Command-Line Flags
USING COMMAND LINE FLAGS IN GO
Golang之使用Flag和Pflag
Go語言學(xué)習(xí)之flag包(The way to go)
Golang flag demo
作者：sparkdev
出處：http://www.cnblogs.com/sparkdev/
到此這篇關(guān)于Golang flag包的具體使用的文章就介紹到這了,更多相關(guān)Golang flag包內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Go Web 編程中的模板庫應(yīng)用指南(超詳細(xì))
這篇文章主要介紹了Go Web 編程中的模板庫應(yīng)用指南，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2020-03-03
Go常用標(biāo)準(zhǔn)庫之fmt的簡(jiǎn)介與使用詳解
fmt 是 Go 語言中的一個(gè)常用標(biāo)準(zhǔn)庫,它用于格式化輸入和輸出數(shù)據(jù),這篇文章主要為大家介紹了fmt的基本使用,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2023-10-10
Goland項(xiàng)目使用gomod配置的詳細(xì)步驟
Goland是一個(gè)用于Go語言開發(fā)的IDE,Goland的項(xiàng)目結(jié)構(gòu)與Go語言的項(xiàng)目結(jié)構(gòu)相似,下面這篇文章主要給大家介紹了關(guān)于Goland項(xiàng)目使用gomod配置的詳細(xì)步驟,文中通過圖文介紹的非常詳細(xì),需要的朋友可以參考下2023-04-04
這篇文章主要介紹了Golang中泛型的使用，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-06-06
分布式鎖是控制分布式系統(tǒng)之間同步訪問共享資源的一種方式。如果不同的系統(tǒng)或是同一個(gè)系統(tǒng)的不同主機(jī)之間共享了一個(gè)或一組資源，那么訪問這些資源時(shí)，需要通過一些互斥手段來防止彼此之間的干擾以保證一致性，在這種情況下，就需要使用分布式鎖了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)，具有一定的參考價(jià)值，感興趣的可以了解一下2022-01-01
這篇文章我們主要介紹垃圾收集器的設(shè)計(jì)原理以及Golang垃圾收集器的實(shí)現(xiàn)原理，文中有詳細(xì)的代碼示例及圖文介紹，感興趣的小伙伴跟著小編一起來學(xué)習(xí)吧2023-06-06
淺析Golang中如何實(shí)現(xiàn)一個(gè)強(qiáng)大的重試機(jī)制
在 Go 語言中,實(shí)現(xiàn)一個(gè)強(qiáng)大的重試機(jī)制可以通過多種方式來完成,這篇文章為大家介紹了一個(gè)常見的實(shí)現(xiàn)方法,需要的小伙伴可以參考一下2025-03-03
Go語言中g(shù)oroutine和WaitGroup的使用示例詳解
goroutine 是Go中一個(gè)輕量級(jí)的線程， 只需要一個(gè)go關(guān)鍵字就可以創(chuàng)建一個(gè)goroutine，這篇文章主要介紹了Go語言中g(shù)oroutine和WaitGroup的使用,需要的朋友可以參考下2023-03-03
Golang 語言極簡(jiǎn)類型轉(zhuǎn)換庫cast的使用詳解
本文我們通過 cast.ToString() 函數(shù)的使用，簡(jiǎn)單介紹了cast 的使用方法，除此之外，它還支持很多其他類型，在這沒有多多介紹，對(duì)Golang 類型轉(zhuǎn)換庫 cast相關(guān)知識(shí)感興趣的朋友一起看看吧2021-11-11