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)
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 stringage intsmoking boolweight 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 mainimport ("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 = splitreturn 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=男