前言
go 自带的flag包可以很容易的实现一个命令行程序的参数解析，但是flag包默认只支持几个基本类型的参数解析，如果需要传递slice或者map类型时就要自定义了，这里记录一下。
原理
通过flag.Var()方法传递一个Value接口，即可自定义命令行参数的解析，flag.Value接口：
1
2
3
4
| type Value interface {
String() string
Set(string) error
}
|
接下来通过自定义类型来实现这个接口即可满足需求。
slice 传递
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| type sliceFlag []string
func (f *sliceFlag) String() string {
return fmt.Sprintf("%v", []string(*f))
}
func (f *sliceFlag) Set(value string) error {
*f = append(*f, value)
return nil
}
func main() {
var hostsFlag sliceFlag
flag.Var(&hostsFlag, "host", "Application hosts,for example: -host=a.com -host=b.com")
}
|
这里需要注意的是数组的扩容需要使用*f = append(*f, value)，来修改原本的数组
map 传递
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
func (f mapFlag) String() string {
return fmt.Sprintf("%v", map[string]string(f))
}
func (f mapFlag) Set(value string) error {
split := strings.SplitN(value, "=", 2)
f[split[0]] = split[1]
return nil
}
func main() {
var hostsFlag sliceFlag
flag.Var(&hostsFlag, "env", "env list,for example: -env key1=value1 -env key2=value2")
}
|
