1.介绍

在 Go中，如果要接收命令行参数，需要使用 flag包进行解析。不同的参数类型可以通过不同的方法接收。

2.参数接受

2.1 接受方式

使用flag接收参数，可以由以下三种方式接受：

  • 方式一: flag.Type(name,defaultVal,desc)
  • 方式二: flag.TypeVar(&flagVar,name,defaultVal,desc)
  • 方式三: flag.Var(&flagVar,name,desc)
  • 方式四: flag.Args()
  • 方式五:flag.Arg(i)
字段 说明
flag.Type 这里的Type是个抽象的概念,不是具体函数名
name 代表输入参数名
defaultVal 默认值
desc 描述信息

2.2 注意事项

  • 上面接收方式一、二、三，都是最后调用flag.Parse(),而方式四和方式五是提前调用flag.Parse()。
  • 方式一、二、三传参时，注意布尔类型，尽量设置成-flag=val，否则会影响后面的参数解析。
  • 方式四、五不能使用-flag=val这种格式传参。

3.flag.Type

3.1 代码

package main
import (
	"flag"
	"fmt"
)
var (
	// 接收字符串
	str =  flag.String("name","默认名","用户姓名")
	// 接收整型
	age = flag.Int("age",18,"用户年龄")
	// 接收布尔型
	smoking = flag.Bool("smoking",false,"是否吸烟")
)
func main()  {
	// 解析
	flag.Parse()
	fmt.Printf("str: %v\n",*str)
	fmt.Printf("age: %v\n",*age)
	fmt.Printf("smoking: %v\n",*smoking)
}

3.2 编译运行

# 编译
➜ go build main.go
# 运行(不传参数时)
➜ ./main                                  
str: 默认名
age: 18
smoking: false

# 运行(传参数时)
➜ /main -name 张三 -age 100 -smoking=true
str: 张三
age: 100
smoking: true

4.flag.TypeVar

4.1 代码

package main
import (
	"flag"
	"fmt"
)
// 先定义变量
var (
	name string
	age int
	smoking bool
	weight float64
)
func main()  {
	// 接收字符串
	flag.StringVar(&name,"name","默认名字","用户名")
	// 接收整型
	flag.IntVar(&age,"age",18,"用户年龄")
	// 接收布尔类型
	flag.BoolVar(&smoking,"smoking",false,"是否吸烟？")
	// 接收浮点型
	flag.Float64Var(&weight,"w",60.0,"体重")
	// 解析
	flag.Parse()
	fmt.Printf("姓名: %v\n",name)
	fmt.Printf("年龄: %v\n",age)
	fmt.Printf("是否吸烟？: %v\n",smoking)
	fmt.Printf("体重: %v\n",weight)
}

4.2 编译运行

# 编译
➜ go build main.go

# 运行(不传参数时)
➜  ./main                                  
姓名: 默认名字
年龄: 18
是否吸烟？: false
体重: 60
# 运行(传参数时)
➜ ./main -name 张三 -age 22 -smoking=true  -w=88.9
姓名: 张三
年龄: 22
是否吸烟？: true
体重: 88.9

4.3 细节

在传参数过程中，发现布尔型传参如果不设置=，会影响后面的参数接收,具体如下:

# 布尔型不设置 = (发现名字没有被接收)
➜  gin-use git:(dev) ✗ ./main -smoking true -name 哈哈                   
姓名: 默认名字
年龄: 18
是否吸烟？: true
体重: 60

# 布尔型设置 = 
➜  gin-use git:(dev) ✗ ./main -smoking=true -name 哈哈
姓名: 哈哈
年龄: 18
是否吸烟？: true
体重: 60

5.flag.Var

通过 flag.Var() 绑定自定义类型，自定义类型需要实现 Value 接口(Receives必须为指针)

5.1 代码

package main

import (
	"flag"
	"fmt"
	"strings"
)
// 自定义变量，并实现flag.Value接口
type Likes []string
func (l *Likes) String() string {
	return fmt.Sprintf("%v",*l)
}
func (l *Likes)Set(s string) error  {
	// 分割字符串
	split := strings.Split(s, ",")
	*l = split
	return nil
}
func main() {
	var likeList Likes
	// 接收自定义类型
	flag.Var(&likeList,"likes","接收自定义类型")
	// 解析
	flag.Parse()
	fmt.Println(likeList)
}

5.2 编译运行

# 编译
➜  go build main.go 
# 运行
➜  ./main -likes=篮球,足球,游戏
[篮球 足球 游戏]

6.flag.Args

一次打印出全部的入参，注意入参格式不能是-flag=val格式。

6.1 代码

package main
import (
	"flag"
	"fmt"
)
func main() {
	// 注意Parse是在Args之前调用
	flag.Parse()
	// 一次接收所有的参数
	args := flag.Args()
	fmt.Println(args)
}

6.2 编译运行

# 编译
➜  go build main.go 
# 错误入参示例
➜  ./main -name=张三 -age=18 -sex=男  
flag provided but not defined: -name
Usage of ./main:
# 正确入参示例
➜ ./main 张三 18 男            
[张三 18 男]

7.flag.Arg(i)

获取指定索引位置的参数，默认索引位置是0;注意入参格式不能是-flag=val格式。

7.1 代码

package main
import (
	"flag"
	"fmt"
)
func main() {
	// 注意Parse是在Arg之前调用
	flag.Parse()
	// 获取指定索引位置参数
	p0 := flag.Arg(0)
	p1 := flag.Arg(1)
	p2 := flag.Arg(2)
	fmt.Printf("索引=0，v=%v\n",p0)
	fmt.Printf("索引=1，v=%v\n",p1)
	fmt.Printf("索引=2，v=%v\n",p2)
}

7.2 编译运行

# 编译
➜  go build main.go 
# 传参
➜  ./main 张三 18 男  
索引=0，v=张三
索引=1，v=18
索引=2，v=男