一篇文章帶你搞定Go語言中flag包使用

作者：Go進(jìn)階者 2021-02-27 10:20:18

開發(fā) 前端

本次章節(jié)主要學(xué)習(xí)的是flag包的使用，flag包主要處理的命令行參數(shù)傳遞問題。

[[383958]]

前言

Hey，大家好，我是碼農(nóng)，星期八，這次來講一下的Go語言中，比較常用的一個(gè)包flag包。

這個(gè)包的用途是接收命令行參數(shù)的。

如果你玩過Mysql，肯定對(duì)這個(gè)界面很熟悉

對(duì)的，就是干這個(gè)用的，接收命令行參數(shù)。

注意

這一章所學(xué)的任何代碼，都要編譯成.exe才能跑，禁止右擊 run!!

os.Args

其實(shí)Go的os模塊，也帶有這個(gè)功能，但是這個(gè)功能有點(diǎn)欠缺。

不能接收像-u root -p 3306這樣指定key的值，只能接收像root 3306這樣的方式。簡(jiǎn)單了解一下吧。

代碼

func main() { 
    cmdArgs := os.Args 
    if len(cmdArgs) > 0 { 
        for index， value := range cmdArgs { 
            fmt.Printf("索引:%v，value值:%v\n"， index， value) 
        } 
} 
}

我們通過命令go build -o mysql.exe main.go編譯成一個(gè)mysql.exe。

mysql.exe圖

執(zhí)行結(jié)果

注:返回的cmdArgs是一個(gè)切片，第一個(gè)值是自己的文件名，如果不想要自己第一個(gè)值，直接切片就好了。

修改第2行代碼。

cmdArgs := os.Args[1:]

執(zhí)行結(jié)果

flag

上述的可能也發(fā)現(xiàn)問題了，只能接收簡(jiǎn)單的值，不能接收key value這種。

flag可以做到，flag有只能接收以下幾種類型。

bool
int系列(int，int64，uint，uint64)
float系列(float，float64)
string
duration

注意

使用flag包，最后要使用flag.Parse()轉(zhuǎn)換一下才能獲取命令行參數(shù)。

flag.Type()

話不多說，直接代碼

func main() { 
    // 第一個(gè)參數(shù)是命令行key，第二個(gè)參數(shù)是默認(rèn)是，第三個(gè)參數(shù)是 .exe -h 提示 
    var user = flag.String("user"， "root"， "用戶名") 
    var port = flag.Int("port"， 3306， "端口") 
    var ip = flag.String("ip"， "localhost"， "mysql ip") 
 
    //必須使用flag.Parse()解析一下命令行參數(shù) 
    flag.Parse() 
    //flag.Type返回的是一個(gè)指針，必須通過 *變量取值 
    fmt.Println(*user，*port，*ip) 
}

執(zhí)行結(jié)果

.exe -h結(jié)果圖

懂了叭，你今天，學(xué)廢了嗎!

flag.TypeVar()

上述你可能也發(fā)現(xiàn)了問題，需要用*變量才能取到值，是不是感覺不太方便，那就來看看flag.TypeVar()。

func main() { 
    //聲明變量用于接收命令行參數(shù) 
    var user string 
    var port int 
    var ip string 
 
    //從命令行掃描參數(shù)賦值到變量 
    flag.StringVar(&user， "user"， "root"， "用戶名") 
    flag.IntVar(&port， "port"， 3306， "端口") 
    flag.StringVar(&ip， "ip"， "localhost"， "mysql ip") 
 
    //必須使用flag.Parse()解析一下命令行參數(shù) 
    flag.Parse() 
    //flag.Type返回的是一個(gè)指針，必須通過 *變量取值 
    fmt.Println(user， port， ip) 
}

執(zhí)行結(jié)果

flag其他方法

flag還有一些其他無關(guān)緊要的方法，了解就好。

func main() { 
    //聲明變量用于接收命令行參數(shù) 
    var user string 
    var port int 
    var ip string 
 
    //從命令行掃描參數(shù)賦值到變量 
    flag.StringVar(&user， "user"， "root"， "用戶名") 
    flag.IntVar(&port， "port"， 3306， "端口") 
    flag.StringVar(&ip， "ip"， "localhost"， "mysql ip") 
 
    //必須使用flag.Parse()解析一下命令行參數(shù) 
    flag.Parse() 
    //flag.Type返回的是一個(gè)指針，必須通過 *變量取值 
    fmt.Println(user， port， ip) 
    ///////////////// 其他方法 
    //返回命令行參數(shù)后的其他參數(shù) 
    fmt.Println(flag.Args()) 
    //返回命令行參數(shù)后的其他參數(shù)個(gè)數(shù) 
    fmt.Println(flag.NArg()) 
    //返回使用的命令行參數(shù)個(gè)數(shù) 
    fmt.Println(flag.NFlag()) 
}