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

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Go›Go教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

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

Flag

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

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

FlagSet

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

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

Int

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

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

IntVar

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

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

func Parse()

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

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

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()

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

NFlag

func NFlag() int

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

Lookup

func Lookup(name string) *Flag

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

NArg

func NArg() int

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

Args

func Args() []string

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

Arg

func Arg(i int) string

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

Var

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

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

Set

func Set(name, value string) error

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

Visit

func Visit(fn func(*Flag))

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

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