Go語言庫系列之flag的具體使用

背景

終端（命令行）操作是程序員的必備技能，但是你知道怎么通過golang制作出如下命令嗎？

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

$ flag girl --height 170

恭喜你獲得了身高 170 的女朋友

極速上手

整個實現(xiàn)非常簡單，只需要5個步驟

第一步，引庫

import "flag"

第二步，定義變量

定義該變量的作用是存儲命令行參數(shù)傳來的值

var height int

第三步，配置命令信息

在main函數(shù)或init函數(shù)中加入

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

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

第四步，解析參數(shù)

在main函數(shù)中追加

flag.Parse()

第五步，打印變量

在main函數(shù)中追加

fmt.Println("恭喜你獲得了身高", height, "的女朋友")

大功告成，感興趣的同學可以自行測試。

完整代碼

package main

import (
 "flag"
 "fmt"
)

var height int

func main() {
 flag.IntVar(height, "height", 140, "身高")
 flag.Parse()
 fmt.Println("恭喜你獲得了身高", height, "的女朋友")
}

高手進階

支持多種執(zhí)行方式

在終端執(zhí)行時，以下幾種傳參方式都被支持

$ girl -height 170
$ girl --height 170
$ girl --height=170
$ girl -height=170

支持多種參數(shù)值類型

目前支持字符串、布爾、整型、浮點、時間等多種類型，完全該所有場景。

布爾類型支持多種寫法

雖然布爾只有true或false，但寫法卻可以各種變種，非常騷氣，支持的寫法有

1, 0, t, f, T, F, true, false, TRUE, FALSE, True, False

隱式處理接收的參數(shù)值

如果我要悄悄地把傳入身高減去5厘米怎么辦？

第一步，定義一個height類型

type height int

第二步，實現(xiàn)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方法用于處理變量設(shè)置時候的邏輯。

第三步，定義height類型變量

var heightFlag height

第四步，主函數(shù)配置及解析

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

代碼與極速上手中的主函數(shù)內(nèi)容大致相同，只是調(diào)用了flag.Var方法，該方法不需要傳入默認值。

到此這篇關(guān)于Go語言庫系列之flag的具體使用的文章就介紹到這了,更多相關(guān)Go flag內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！