Go 很适合开发命令行工具,获取命令行参数是关键的一步。


有两种方式可以获取命令行参数

os

引入

1
import "os"

使用

1
2
3
4
func main() {
fmt.Prinln(os.Args)
// [/var/folders/kz/8syfctw919zdt3shr9w5j8v00000gn/T/go-build512483375/b001/exe/flag -v -name wxnacy -age 2 arg1 arg2]
}

该方法返回的是参数列表,像 shell 语言一样,数组的第一个参数是运行文件的位置,该方法很简单,但是想要获取 - 标示的变量参数就不太够用了。

flag

flag 包时 Go 中专业的解析参数的包。

引入

1
import "flag"

语法

1
2
3
4
5
// xxx 代表参数类型,如:string bool int
// name 参数名
// value 默认值
// usage 使用说明
func Xxx(name string, value xxx, usage string) *xxx

该方法返回的是指定参数类型的指针

1
2
3
4
5
// xxx 代表参数类型,如:string bool int
// name 参数名
// value 默认值
// usage 使用说明
func XxxVar(p *xxx, name string, value xxx, usage string)

该方法传入已定义的参数指针,可以得到相应的技术。

flag 可以解析的参数类型有

1
2
3
4
-k v
-k=v
--k v
--k=v

使用

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package main

import (
"fmt"
"flag"
"os"
)

var version bool
var name string
var age *int

func InitArgs() {
flag.BoolVar(&version, "v", false, "Get Version")
flag.StringVar(&name, "name", "", "Get Name")
age = flag.Int("age", 0, "Get Age")
}

func main() {
InitArgs()

flag.Parse()

fmt.Printf("version %t name %s age %d", version, name, *age)
fmt.Println("")
fmt.Printf("args %s", flag.Args())
fmt.Println("")
fmt.Printf("args num %d, flag args num %d", flag.NArg(), flag.NFlag())
fmt.Println("")
fmt.Printf("os.Args %s", os.Args)
fmt.Println("")



}
1
2
3
4
5
6
> $ go run src/simple/flag.go -v -name wxnacy -age 2 arg1 arg2

version true name wxnacy age 2
args [arg1 arg2]
args num 2, flag args num 3
os.Args [/var/folders/kz/8syfctw919zdt3shr9w5j8v00000gn/T/go-build512483375/b001/exe/flag -v -name wxnacy -age 2 arg1 arg2]
03-16 21:44