Go語言庫系列之flag

• 2020 年 4 月 3 日
• 筆記

背景

終端（命令行）操作是程式設計師的必備技能，但是你知道怎麼通過golang製作出如下命令嗎？

$ flag girl -h  Usage of girl:    -height int          身高 (default 140)

$ flag girl --height 170  恭喜你獲得了身高 170 的女朋友

極速上手

整個實現非常簡單，只需要5個步驟

第一步，引庫

import "flag"

第二步，定義變數

定義該變數的作用是存儲命令行參數傳來的值

var height int

第三步，配置命令資訊

在main函數或init函數中加入

flag.IntVar(&height, "height", 140, "身高")

• 第一個參數是定義的變數引用
• 第二個參數：命令的名稱
• 第三個參數：命令參數的默認值
• 第四個參數：命令的用法提示

第四步，解析參數

在main函數中追加

flag.Parse()

第五步，列印變數

在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類型

type height int

第二步，實現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類型變數

var heightFlag height

第四步，主函數配置及解析

func main() {  	flag.Var(&heightFlag, "height", "身高")  	flag.Parse()  	fmt.Println(heightFlag)  }

程式碼與極速上手中的主函數內容大致相同，只是調用了flag.Var方法，該方法不需要傳入默認值。