Go语言库系列之flag

背景

终端（命令行）操做是程序员的必备技能，可是你知道怎么经过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方法，该方法不须要传入默认值。