Go基础篇【第6篇】: 内置库模块 flag

import "flag"

flag包实现了命令行参数的解析。每个参数认为一条记录，根据实际进行定义，到一个set集合。每条都有各自的状态参数。

在使用flag时正常流程：

1.

通过flag.String(), flag.Bool(), flag.Int()等函数来定义命令行中需要使用的参数。

2.

在定义完flag后，通过调用flag.Parse()来进行对命令行参数的解析。

3.

获取flag.String(), flag.Bool(), flag.Int()等方法的返回值,即对应用户输入的参数.

注意的是flag.Xxx()返回的值是变量的内存地址,要获取值时要通过在变量前加*(星号)获取.

说明:
像flag.Int、flag.Bool、flag.String这样的函数格式都是一样的，调用的时候需要传入3个参数
参数的说明如下:
第一个arg表示参数名称，在控制台的时候,提供给用户使用.
第二个arg表示默认值，如果用户在控制台没有给该参数赋值的话,就会使用该默认值.
第三个arg表示使用说明和描述,在控制台中输入-arg的时候会显示该说明,类似-help

可注册flag类型有：Bool / Int / Int64 / Uint / Uint64 / Float / Float64 / String / Duration / Var

type Flag

type Flag struct {
Name     string // flag在命令行中的名字
Usage    string // 帮助信息
Value    Value  // 要设置的值
DefValue string // 默认值（文本格式），用于使用信息
}

Flag类型代表一条flag的状态。

type FlagSet

type FlagSet struct {
// Usage函数在解析flag出现错误时会被调用
// 该字段为一个函数（而非采用方法），以便修改为自定义的错误处理函数
Usage func()
// 内含隐藏或非导出字段
}

FlagSet代表一个已注册的flag的集合。FlagSet零值没有名字，采用ContinueOnError错误处理策略。

func Int

func Int(name string, value int, usage string) *int

Int用指定的名称、默认值、提示信息注册一个int类型flag。返回一个保存了该flag的值的指针。

func IntVar

func IntVar(p *int, name string, value int, usage string)

IntVar用指定的名称、默认值、使用信息注册一个int类型flag，并将flag的值保存到p指向的变量。

func Parse

func Parse()

从os.Args[1:]中解析注册的flag。必须在所有flag都注册好而未访问其值时执行。未注册却使用flag -help时，会返回ErrHelp。

即：将输入的flag和注册的flag进行匹配解析

func Parsed

func Parsed() bool

返回是否Parse已经被调用过。

package main

import (
"flag"
"fmt"
)

func main() {
married := flag.Bool("married", false, "Are you married?")
age := flag.Int("age", 22, "How old are you?")
name := flag.String("name", "", "What your name?")

var address string
//flag.StringVar这样的函数第一个参数换成了变量地址，后面的参数和flag.String是一样的。
flag.StringVar(&address, "address", "GuangZhou", "Where is your address?")

flag.Parse() //解析输入的参数

fmt.Println("输出的参数married的值是:", *married)//不加*号的话,输出的是内存地址
fmt.Println("输出的参数age的值是:", *age)
fmt.Println("输出的参数name的值是:", *name)
fmt.Println("输出的参数address的值是:", address)
}

从运行结果中可以看出,address参数我们并没有指定值,则输出的就是默认值.
另外,-arg后面的=号也不是必须的,可以用空格代替.
如果要查看该程序的所有参数的使用,可以使用-help来查看,如下:

func PrintDefaults

func PrintDefaults()

PrintDefault会向标准错误输出写入所有注册好的flag的默认值。

func NFlag

func NFlag() int

NFlag返回已被设置的flag的数量。

func Lookup

func Lookup(name string) *Flag

返回已经已注册flag的Flag结构体指针；如果flag不存在的话，返回nil。

func NArg

func NArg() int

NArg返回解析flag之后剩余参数的个数。

func Args

func Args() []string

返回解析之后剩下的非flag参数。（不包括命令名）.就是无法进行flag匹配的有哪些，

func Arg

func Arg(i int) string

返回解析之后剩下的第i个参数，从0开始索引。

func Var

func Var(value Value, name string, usage string)

Var方法使用指定的名字、使用信息注册一个flag。该flag的类型和值由第一个参数表示，该参数应实现了Value接口。例如，用户可以创建一个flag，可以用Value接口的Set方法将逗号分隔的字符串转化为字符串切片。

func Set

func Set(name, value string) error

设置已注册的flag的值，修改或者设置

func Visit

func Visit(fn func(*Flag))

按照字典顺序遍历标签，并且对每个标签调用fn。 这个函数只遍历解析时进行了设置的标签。[就是执行时提供非默认的值]

func VisitAll

func VisitAll(fn func(*Flag))

按照字典顺序遍历标签，并且对每个标签调用fn。 这个函数会遍历所有标签，不管解析时有无进行设置。

package main

import (
"flag"
"fmt"
)

func main() {
// golang的flag包的一些基本使用方法

// 待使用的变量
var id int
var name string
var male bool

// 是否已经解析
fmt.Println("parsed? = ", flag.Parsed())

// 设置flag参数 (变量指针，参数名，默认值，帮助信息)
// 也可以用以下带返回值的方法代替，不过他们返回的是指针，比较麻烦点
// Int(name string, value int, usage string) *int
// String(name string, value string, usage string) *string
// Bool(name string, value bool, usage string) *bool
flag.IntVar(&id, "id", 123, "help msg for id")
flag.StringVar(&name, "name", "default name", "help msg for name")
flag.BoolVar(&male, "male", false, "help msg for male")

// 解析
flag.Parse()

// 是否已经解析
fmt.Println("parsed? = ", flag.Parsed())

//

// 获取非flag参数
fmt.Println(flag.NArg())
fmt.Println("------ Args start ------")
fmt.Println(flag.Args())
for i, v := range flag.Args() {
fmt.Printf("arg[%d] = (%s).\n", i, v)
}
fmt.Println("------ Args end ------")
//
// visit只包含已经设置了的flag
fmt.Println("------ visit flag start ------")
flag.Visit(func(f *flag.Flag) {
fmt.Println(f.Name, f.Value, f.Usage, f.DefValue)
})
fmt.Println("------ visit flag end ------")
//
// visitAll只包含所有的flag(包括未设置的)
fmt.Println("------ visitAll flag start ------")
flag.VisitAll(func(f *flag.Flag) {
fmt.Println(f.Name, f.Value, f.Usage, f.DefValue)

})
fmt.Println("------ visitAll flag end ------")
//
// flag参数
fmt.Printf("id = %d\n", id)
fmt.Printf("name = %s\n", name)
fmt.Printf("male = %t\n", male)

// flag参数默认值
fmt.Println("------ PrintDefaults start ------")
flag.PrintDefaults()
fmt.Println("------ PrintDefaults end ------")

//非flag参数个数
fmt.Printf("NArg = %d\n", flag.NArg())
// 已设置的flag参数个数
fmt.Printf("NFlag = %d\n", flag.NFlag())
}

# go run demo2_flag.go -id 234 -name "xx" -male=true p1 p2 p3
parsed? =  false
parsed? =  true
3
------ Args start ------
[p1 p2 p3]
arg[0] = (p1).
arg[1] = (p2).
arg[2] = (p3).
------ Args end ------
------ visit flag start ------
id 234 help msg for id 123
male true help msg for male false
name xx help msg for name default name
------ visit flag end ------
------ visitAll flag start ------
id 234 help msg for id 123
male true help msg for male false
name xx help msg for name default name
------ visitAll flag end ------
id = 234
name = xx
male = true
------ PrintDefaults start ------
-id int
help msg for id (default 123)
-male
help msg for male
-name string
help msg for name (default "default name")
------ PrintDefaults end ------
NArg = 3
NFlag = 3

在flag包中，进行了进一步封装：将FlagSet的方法都重新定义了一遍，也就是提供了一序列函数，而函数中只是简单的调用已经实例化好了的FlagSet