go语言标准库之flag

亦凉 2023-02-20 02:10 13阅读 0赞

Go语言内置的`flag`包实现了命令行参数的解析，`flag`包使得开发命令行工具更为简单。

# os.Args #

如果你只是简单的想要获取命令行参数，可以像下面的代码示例一样使用`os.Args`来获取命令行参数。

package main
    
    import (
    	"fmt"
    	"os"
    )
    
    //os.Args demo
    func main() { 
    	//os.Args是一个[]string
    	if len(os.Args) > 0 { 
    		for index, arg := range os.Args { 
    			fmt.Printf("args[%d]=%v\n", index, arg)
    		}
    	}
    }

将上面的代码执行`go build -o "args_demo"`编译之后，执行：

$ ./args_demo a b c d
    args[0]=./args_demo
    args[1]=a
    args[2]=b
    args[3]=c
    args[4]=d

`os.Args`是一个存储命令行参数的字符串切片，它的第一个元素是执行文件的名称。

# flag包基本使用 #

本文介绍了flag包的常用函数和基本用法，更详细的内容请查看[官方文档][Link 1]。

## 导入flag包 ##

import flag

## flag参数类型 ##

flag包支持的命令行参数类型有`bool`、`int`、`int64`、`uint`、`uint64`、`float` `float64`、`string`、`duration`。

<table> 
 <thead> 
  <tr> 
   <th align="center">flag参数</th> 
   <th align="center">有效值</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td align="center">字符串flag</td> 
   <td align="center">合法字符串</td> 
  </tr> 
  <tr> 
   <td align="center">整数flag</td> 
   <td align="center">1234、0664、0x1234等类型，也可以是负数。</td> 
  </tr> 
  <tr> 
   <td align="center">浮点数flag</td> 
   <td align="center">合法浮点数</td> 
  </tr> 
  <tr> 
   <td align="center">bool类型flag</td> 
   <td align="center">1, 0, t, f, T, F, true, false, TRUE, FALSE, True, False。</td> 
  </tr> 
  <tr> 
   <td align="center">时间段flag</td> 
   <td align="center">任何合法的时间段字符串。如”300ms”、”-1.5h”、”2h45m”。 合法的单位有”ns”、”us” /“µs”、”ms”、”s”、”m”、”h”。</td> 
  </tr> 
 </tbody> 
</table>

## 定义命令行flag参数 ##

有以下两种常用的定义命令行`flag`参数的方法。

### flag.Type() ###

基本格式如下：

`flag.Type(flag名, 默认值, 帮助信息)*Type` 例如我们要定义姓名、年龄、婚否三个命令行参数，我们可以按如下方式定义：

name := flag.String("name", "张三", "姓名")
    age := flag.Int("age", 18, "年龄")
    married := flag.Bool("married", false, "婚否")
    delay := flag.Duration("d", 0, "时间间隔")

需要注意的是，此时`name`、`age`、`married`、`delay`均为对应类型的指针。

### flag.TypeVar() ###

基本格式如下： `flag.TypeVar(Type指针, flag名, 默认值, 帮助信息)` 例如我们要定义姓名、年龄、婚否三个命令行参数，我们可以按如下方式定义：

var name string
    var age int
    var married bool
    var delay time.Duration
    flag.StringVar(&name, "name", "张三", "姓名")
    flag.IntVar(&age, "age", 18, "年龄")
    flag.BoolVar(&married, "married", false, "婚否")
    flag.DurationVar(&delay, "d", 0, "时间间隔")

## flag.Parse() ##

通过以上两种方法定义好命令行flag参数后，需要通过调用`flag.Parse()`来对命令行参数进行解析。

支持的命令行参数格式有以下几种：

*  `-flag xxx` （使用空格，一个`-`符号）
 *  `--flag xxx` （使用空格，两个`-`符号）
 *  `-flag=xxx` （使用等号，一个`-`符号）
 *  `--flag=xxx` （使用等号，两个`-`符号）

其中，布尔类型的参数必须使用等号的方式指定。

Flag解析在第一个非flag参数（单个”-“不是flag参数）之前停止，或者在终止符”–“之后停止。

## flag其他函数 ##

flag.Args()  返回命令行参数后的其他参数，以[]string类型
    flag.NArg()  //返回命令行参数后的其他参数个数
    flag.NFlag() //返回使用的命令行参数个数

## 完整示例 ##

### 定义 ###

func main() { 
    	//定义命令行参数方式1
    	var name string
    	var age int
    	var married bool
    	var delay time.Duration
    	flag.StringVar(&name, "name", "张三", "姓名")
    	flag.IntVar(&age, "age", 18, "年龄")
    	flag.BoolVar(&married, "married", false, "婚否")
    	flag.DurationVar(&delay, "d", 0, "延迟的时间间隔")
    
    	//解析命令行参数
    	flag.Parse()
    	fmt.Println(name, age, married, delay)
    	//返回命令行参数后的其他参数
    	fmt.Println(flag.Args())
    	//返回命令行参数后的其他参数个数
    	fmt.Println(flag.NArg())
    	//返回使用的命令行参数个数
    	fmt.Println(flag.NFlag())
    }

### 使用 ###

命令行参数使用提示：

$ ./flag_demo -help
    Usage of ./flag_demo:
      -age int
            年龄 (default 18)
      -d duration
            时间间隔
      -married
            婚否
      -name string
            姓名 (default "张三")

正常使用命令行flag参数：

$ ./flag_demo -name 沙河娜扎 --age 28 -married=false -d=1h30m
    沙河娜扎 28 false 1h30m0s
    []
    0
    4

使用非flag命令行参数：

$ ./flag_demo a b c
    张三 18 false 0s
    [a b c]
    3
    0

[Link 1]: https://studygolang.com/pkgdoc