侧边栏壁纸
博主头像
张种恩的技术小栈博主等级

行动起来,活在当下

  • 累计撰写 748 篇文章
  • 累计创建 65 个标签
  • 累计收到 39 条评论

目 录CONTENT

文章目录
Go

[转载]Go标准库-flag

zze
zze
2020-09-25 / 0 评论 / 0 点赞 / 457 阅读 / 5064 字

不定期更新相关视频,抖音点击左上角加号后扫一扫右方侧边栏二维码关注我~正在更新《Shell其实很简单》系列

Go 语言内置的 flag 包实现了命令行参数的解析,flag 包使得开发命令行工具更为简单。

os.Args

如果你只是简单的想要获取命令行参数,可以像下面的代码示例一样使用 os.Args 来获取命令行参数。

package main

import (
	"fmt"
	"os"
)

//os.Args demo
func main() {
	//os.Args是一个[]string
	if len(os.Args) > 0 {
		for index, arg := range os.Args {
			fmt.Printf("args[%d]=%v\n", index, arg)
		}
	}
}

将上面的代码执行 go build -o "args_demo" 编译之后,执行:

$ ./args_demo a b c d
args[0]=./args_demo
args[1]=a
args[2]=b
args[3]=c
args[4]=d

os.Args 是一个存储命令行参数的字符串切片,它的第一个元素是执行文件的名称。

flag 包基本使用

本文介绍了 flag 包的常用函数和基本用法,更详细的内容请查看官方文档

导入 flag 包

import flag

flag 参数类型

flag 包支持的命令行参数类型有 boolintint64uintuint64floatfloat64stringduration

flag 参数有效值
字符串 flag合法字符串
整数 flag123406640x1234 等类型,也可以是负数。
浮点数 flag合法浮点数
bool 类型 flag10tfTFtruefalseTRUEFALSETrueFalse
时间段 flag任何合法的时间段字符串。如 300ms-1.5h2h45m。合法的单位有 nsus/µsmssmh

定义命令行 flag 参数

有以下两种常用的定义命令行 flag 参数的方法。

flag.Type()

基本格式如下:
flag.Type(flag名, 默认值, 帮助信息)*Type,例如我们要定义姓名、年龄、婚否三个命令行参数,我们可以按如下方式定义:

name := flag.String("name", "张三", "姓名")
age := flag.Int("age", 18, "年龄")
married := flag.Bool("married", false, "婚否")
delay := flag.Duration("d", 0, "时间间隔")

需要注意的是,此时 nameagemarrieddelay 均为对应类型的指针。

flag.TypeVar()

基本格式如下: flag.TypeVar(Type指针, flag名, 默认值, 帮助信息) 例如我们要定义姓名、年龄、婚否三个命令行参数,我们可以按如下方式定义:

var name string
var age int
var married bool
var delay time.Duration
flag.StringVar(&name, "name", "张三", "姓名")
flag.IntVar(&age, "age", 18, "年龄")
flag.BoolVar(&married, "married", false, "婚否")
flag.DurationVar(&delay, "d", 0, "时间间隔")

flag.Parse()

通过以上两种方法定义好命令行 flag 参数后,需要通过调用 flag.Parse() 来对命令行参数进行解析。

支持的命令行参数格式有以下几种:

  • -flag xxx (使用空格,一个 - 符号);
  • --flag xxx (使用空格,两个 - 符号);
  • -flag=xxx (使用等号,一个 - 符号);
  • --flag=xxx (使用等号,两个 - 符号);
    其中,布尔类型的参数必须使用等号的方式指定。

flag 解析在第一个非 flag 参数(单个 - 不是 flag 参数)之前停止,或者在终止符 之后停止。

flag 其它函数

flag.Args()  ////返回命令行参数后的其他参数,以[]string类型
flag.NArg()  //返回命令行参数后的其他参数个数
flag.NFlag() //返回使用的命令行参数个数

完整示例

定义

func main() {
	//定义命令行参数方式1
	var name string
	var age int
	var married bool
	var delay time.Duration
	flag.StringVar(&name, "name", "张三", "姓名")
	flag.IntVar(&age, "age", 18, "年龄")
	flag.BoolVar(&married, "married", false, "婚否")
	flag.DurationVar(&delay, "d", 0, "延迟的时间间隔")

	//解析命令行参数
	flag.Parse()
	fmt.Println(name, age, married, delay)
	//返回命令行参数后的其他参数
	fmt.Println(flag.Args())
	//返回命令行参数后的其他参数个数
	fmt.Println(flag.NArg())
	//返回使用的命令行参数个数
	fmt.Println(flag.NFlag())
}

使用

命令行参数使用提示:

$ ./flag_demo -help
Usage of ./flag_demo:
  -age int
        年龄 (default 18)
  -d duration
        时间间隔
  -married
        婚否
  -name string
        姓名 (default "张三")

正常使用命令行 flag 参数:

$ ./flag_demo -name 沙河娜扎 --age 28 -married=false -d=1h30m
沙河娜扎 28 false 1h30m0s
[]
0
4

使用非 flag 命令行参数:

$ ./flag_demo a b c
张三 18 false 0s
[a b c]
3
0

转自:

0

评论区