前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Go语言库系列之flag

Go语言库系列之flag

作者头像
平也
发布2020-04-03 15:19:26
4640
发布2020-04-03 15:19:26
举报
文章被收录于专栏:平也平也

背景

终端(命令行)操作是程序员的必备技能,但是你知道怎么通过golang制作出如下命令吗?

代码语言:javascript
复制
$ flag girl -h
Usage of girl:
  -height int
        身高 (default 140)
代码语言:javascript
复制
$ flag girl --height 170
恭喜你获得了身高 170 的女朋友

极速上手

整个实现非常简单,只需要5个步骤

第一步,引库
代码语言:javascript
复制
import "flag"
第二步,定义变量

定义该变量的作用是存储命令行参数传来的值

代码语言:javascript
复制
var height int
第三步,配置命令信息

main函数或init函数中加入

代码语言:javascript
复制
flag.IntVar(&height, "height", 140, "身高")
  • 第一个参数是定义的变量引用
  • 第二个参数:命令的名称
  • 第三个参数:命令参数的默认值
  • 第四个参数:命令的用法提示
第四步,解析参数

main函数中追加

代码语言:javascript
复制
flag.Parse()
第五步,打印变量

main函数中追加

代码语言:javascript
复制
fmt.Println("恭喜你获得了身高", height, "的女朋友")

大功告成,感兴趣的同学可以自行测试。

完整代码
代码语言:javascript
复制
package main

import (
	"flag"
	"fmt"
)

var height int

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

高手进阶

支持多种执行方式

在终端执行时,以下几种传参方式都被支持

代码语言:javascript
复制
$ girl -height 170
$ girl --height 170
$ girl --height=170
$ girl -height=170
支持多种参数值类型

目前支持字符串、布尔、整型、浮点、时间等多种类型,完全该所有场景。

布尔类型支持多种写法

虽然布尔只有truefalse,但写法却可以各种变种,非常骚气,支持的写法有

代码语言:javascript
复制
1, 0, t, f, T, F, true, false, TRUE, FALSE, True, False
隐式处理接收的参数值

如果我要悄悄地把传入身高减去5厘米怎么办?

第一步,定义一个height类型

代码语言:javascript
复制
type height int

第二步,实现flag.Value接口的两个方法

代码语言:javascript
复制
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类型变量

代码语言:javascript
复制
var heightFlag height

第四步,主函数配置及解析

代码语言:javascript
复制
func main() {
	flag.Var(&heightFlag, "height", "身高")
	flag.Parse()
	fmt.Println(heightFlag)
}

代码与极速上手中的主函数内容大致相同,只是调用了flag.Var方法,该方法不需要传入默认值。

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-04-02 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 背景
  • 极速上手
    • 第一步,引库
      • 第二步,定义变量
        • 第三步,配置命令信息
          • 第四步,解析参数
            • 第五步,打印变量
              • 完整代码
              • 高手进阶
                • 支持多种执行方式
                  • 支持多种参数值类型
                    • 布尔类型支持多种写法
                      • 隐式处理接收的参数值
                      领券
                      问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档