flag包

标识是特殊格式的字符串,可以传入程序以控制其行为。如果希望支持多个标识,那么自己单独处理标识可能会很困难。因此,如果你正在开发Unix系统命令行实用程序,你会发现flag包非常有趣和有用。在它的其他特性中,flag包对命令行参数和选项的顺序没有任何假设,并且在命令行实用程序执行过程中出现错误时,它会打印有用的信息。

flag包的最大好处是,它是Go标准库的一部分,意味着它经过了大量的测试和调试。

我将介绍两个使用flag包的Go程序:一个简单的和一个复杂的。

第一个程序,称为simpleFlag.go,分为四部分。simpleFlag.go程序识别两个命令行选项:第一个是Boolean选项,第二个是整数值。

simpleFlag.go的第一部分如下:

  1. import (
  2. "flag"
  3. "fmt"
  4. )

simpleFlag.go的第二部分如下:

  1. func main() {
  2. minusK := flag.Bool("k", true, "k")
  3. minusO := flag.Int("O", 1, "O")
  4. flag.Parse()

flag.Bool("k", true, "k")语句定义了名为kBoolean型的命令行选项,默认值为true。语句的最后一个参数是将与程序的用法信息一起显示的字符串。类似,flag.Int()函数的作用是增加对整数命令行选项的支持。

通常,在定义了命令行选项后,需要调用flag.Parse()

simpleFlag.go的第三部分程序如下:

  1. valueK := *minusK
  2. valueO := *minusO
  3. valueO++

在前面的Go代码中,你看到了如何获取选项的值。好处是flag自动将flag.Int()标识关联的输入转换为整数值。即你不用再做转换操作。另外,flag包确保它得到一个可接受的整数值。

simpleFlag.go的剩余代码如下:

  1. fmt.Println("-k:", valueK)
  2. fmt.Println("-O:", valueO)
  3. }

获取到需要的参数后,你就可以使用它们了。

simpleFlag.go交互将创建如下输出:

  1. $ go run simpleFlag.go -O 100
  2. -k: true
  3. -O: 101
  4. $ go run simpleFlag.go -O=100
  5. -k: true
  6. -O: 101
  7. $ go run simpleFlag.go -O=100 -k
  8. -k: true
  9. -O: 101
  10. $ go run simpleFlag.go -O=100 -k false
  11. -k: true
  12. -O: 101
  13. $ go run simpleFlag.go -O=100 -k=false
  14. -k: false
  15. -O: 101

如果在执行simpleFlag.go中遇到错误,则会得到如下错误信息:

  1. $ go run simpleFlag.go -O=notAnInteger
  2. invalid value "notAnInteger" for flag -O: strconv.ParseInt: parsing "notAnInteger": invalid syntax
  3. Usage of /var/folders/sk/ltk8cnw501zdtr2hxcj5sv2m0000gn/T/go-build020625525/command-line arguments/_obj/exe/simpleFlag:
  4. -O int
  5. O (default 1)
  6. -k k (default true)
  7. exit status 2

注意:当程序的命令行选项中出现错误时,将自动打印常规用法信息。

现在该是展示一个使用flag包更实际、更高级的程序了。程序funWithFlag.go分为五部分。funWithFlag.go识别各种选项,包括一个由逗号分隔的多个值的选项。此外,它还将说明如何访问位于可执行文件末尾且不属于任何选项的命令行参数。

funWithFlag.go中使用的flag.Var()函数创建一个满足flag.Value接口的任意类型的标识,flag.Value接口定义如下:

  1. type Value interface {
  2. String() string
  3. Set(string) error
  4. }

funWithFlag.go第一部分代码如下:

  1. package main
  2. import(
  3. "flag"
  4. "fmt"
  5. "strings"
  6. )
  7. type NamesFlag struct {
  8. Names []string
  9. }

NamesFlag 数据结构用于 flag.Value 接口。

funWithFlag.go 的第二部分如下:

  1. func (s *NamesFlag) GetNames() []string{
  2. return s.Names
  3. }
  4. func (s *NamesFlag) String() string {
  5. return fmt.Sprint(s.Names)
  6. }

funWithFlag.go 的第三部分代码如下:

  1. func (s *NamesFlag) Set(v string) error {
  2. if len(s.Names) > 0 {
  3. return fmt.Errorf("Cannot use names flag more than once!")
  4. }
  5. names := strings.Split(v, ",")
  6. for _, item := range names {
  7. s.Names = append(s.Names, item)
  8. }
  9. return nil
  10. }

首先,Set() 方法确保相关命令行选项没有被设置。之后,获取输入并使用 strings.Split() 函数来分隔参数。最后,参数被保存在 NamesFlag 结构的 Names 字段。

funWithFlag.go 的第四部分代码如下:

  1. func main() {
  2. var manyNames NamesFlag
  3. minusK := flag.Int("k:", 0, "An int")
  4. minusO := flag.String("o", "Mihalis", "The name")
  5. flag.Var(&manyNames, "names", "Comma-separated list")
  6. flag.Parse()
  7. fmt.Println("-k:", *minusK)
  8. fmt.Println("-o:", *minusO)

funWithFlag.go 的最后部分如下:

  1. for i, item := range manyNames.GetNames() {
  2. fmt.Println(i, item)
  3. }
  4. fmt.Println("Remaing command-line arugments:")
  5. for index, val := range flag.Args() {
  6. fmt.Println(index, ":", val)
  7. }
  8. }

flag.Args() 切片保留命令行参数,而 manyNames 变量保留来自 flag.Var() 命令行选项的值。

执行 funWithFlag.go 产生如下输出:

""

除非你开发一个不需要选项的命令行小工具,否则你极需要使用 flag 包来处理您的程序的命令行选项。