Go 语言内置的flag
包实现了命令行参数的解析,flag
包使得开发命令行工具更为简单。
如果你只是简单的想要获取命令行参数,可以像下面的代码示例一样使用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 参数类型
flag 包支持的命令行参数类型有bool
、int
、int64
、uint
、uint64
、float``float64
、string
、duration
。
flag 参数 | 有效值 |
---|---|
字符串 flag | 合法字符串 |
整数 flag | 1234、0664、0x1234 等类型,也可以是负数。 |
浮点数 flag | 合法浮点数 |
bool 类型 flag | 1, 0, t, f, T, F, true, false, TRUE, FALSE, True, False。 |
时间段 flag | 任何合法的时间段字符串。如”300ms”、”-1.5h”、”2h45m”。 |
合法的单位有”ns”、”us” /“µs”、”ms”、”s”、”m”、”h”。 |
定义命令行 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, "时间间隔")
需要注意的是,此时name
、age
、married
、delay
均为对应类型的指针。
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 18 -married=false -d=1h30m
小明 18 false 1h30m0s
[]
0
4
使用非 flag 命令行参数:
$ ./flag_demo a b c
张三 18 false 0s
[a b c]
3
0