背景
終端(命令行)操作是程序員的必備技能,但是你知道怎么通過golang制作出如下命令嗎?
$ flag girl -h
Usage of girl:
-height int
身高 (default 140)
$ flag girl --height 170
恭喜你獲得了身高 170 的女朋友
極速上手
整個實現非常簡單,只需要5個步驟
第一步,引庫
第二步,定義變量
定義該變量的作用是存儲命令行參數傳來的值
第三步,配置命令信息
在main函數或init函數中加入
flag.IntVar(height, "height", 140, "身高")
- 第一個參數是定義的變量引用
- 第二個參數:命令的名稱
- 第三個參數:命令參數的默認值
- 第四個參數:命令的用法提示
第四步,解析參數
在main函數中追加
第五步,打印變量
在main函數中追加
fmt.Println("恭喜你獲得了身高", height, "的女朋友")
大功告成,感興趣的同學可以自行測試。
完整代碼
package main
import (
"flag"
"fmt"
)
var height int
func main() {
flag.IntVar(height, "height", 140, "身高")
flag.Parse()
fmt.Println("恭喜你獲得了身高", height, "的女朋友")
}
高手進階
支持多種執行方式
在終端執行時,以下幾種傳參方式都被支持
$ girl -height 170
$ girl --height 170
$ girl --height=170
$ girl -height=170
支持多種參數值類型
目前支持字符串、布爾、整型、浮點、時間等多種類型,完全該所有場景。
布爾類型支持多種寫法
雖然布爾只有true或false,但寫法卻可以各種變種,非常騷氣,支持的寫法有
1, 0, t, f, T, F, true, false, TRUE, FALSE, True, False
隱式處理接收的參數值
如果我要悄悄地把傳入身高減去5厘米怎么辦?
第一步,定義一個height類型
第二步,實現flag.Value接口的兩個方法
func (h *height) String() string {
return fmt.Sprint(*h)
}
func (h *height) Set(value string) error {
valueInt, _ := strconv.Atoi(value)
*h = height(valueInt - 5)
return nil
}
String方法用于格式化值,Set方法用于處理變量設置時候的邏輯。
第三步,定義height類型變量
第四步,主函數配置及解析
func main() {
flag.Var(heightFlag, "height", "身高")
flag.Parse()
fmt.Println(heightFlag)
}
代碼與極速上手中的主函數內容大致相同,只是調用了flag.Var方法,該方法不需要傳入默認值。
到此這篇關于Go語言庫系列之flag的具體使用的文章就介紹到這了,更多相關Go flag內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持腳本之家!