golang flag 包的使用指北

Golang
211
0
0
2024-02-03

说起 golang 的 flag 个包,我们第一反应的是什么呢?至少我曾经第一次看到 flag 包的时候,第一反应是想起写 C 语言的时候咱们用于定义一个表示的,我们一般会命名为 flag 变量

实际上 golang 的 flag 包是用于处理命令行参数的工具包,我们可以基于这个包来开发自定义的命令行工具

对于命令行,相信大家都不陌生吧,学过 C 的就知道,如下代码是否很熟悉

int main(int argc,char **argv)
{
    // 打印 agrc  和 argv
        return 0;
}

其中 argc 表示命令行的参数个数, argv 二级指针表示实际的命令参数

那么,对于咱们的 golang 是如何玩的,咱们先写个 demo 看看效果

main.go

func main() {
   if len(os.Args) > 0 {
      fmt.Println("len(os.Args) == ", len(os.Args))
      for _, v := range os.Args {
         fmt.Println(v)
      }
   }
}

效果如下:

咱们 go build 输出程序名为 testFlag 的可执行程序,如下运行即可

当然需要看非常详细和全面的关于 flag 包的内容,可以直接查看官方文档 ,此处咱们是对 flag 包的介绍,平常都会使用到的一些功能,以及注意事项,分别从如下 3 个小点介绍一下常用的功能

  • 注册 flag 和 Flag 解析支持的数据类型
  • 绑定 flag 变量
  • flag.Parse() 的使用方式

注册 flag 和 Flag 解析支持的数据类型

我们可以看到上述简单 demo 演示的是咱们运行可执行程序的时候追加我们期望的参数

实际上我们使用的比较多的是在程序内部去设置 flag,例如一般写 web 服务器的时候,就会去使用 flag 包设置端口

咱们在程序中使用 flag 包,定义一个端口

通过效果,我们可看到,咱们使用 flag 包来设置命令行参数时,实际上返回的是一个指针,这个地方可别弄错了

在我们的代码中,我们可以看到 flag 包他支持这么多数据类型的注册,如下:

  • 基本的 int,int64,uint,uint64 ,整数的 flag 支持这些进制的:1234、0664、0x1234
  • 布尔值,bool , 布尔值可以这样来写:1, 0, t, f, T, F, true, false, TRUE, FALSE, True, False
  • 浮点数,float,float64
  • 字符串,string
  • 时间段 duration 也是支持的很好的,例如:我们可以设置为 100ms,1h,1us,1h25m 等等都是可以的

那么对应到咱们去设置各种数据类型的命令行参数的时候,我们就可以简单的使用 flag 包对应的函数即可,如:

port := flag.Int("port", 8866, "端口")
name := flag.String("name", "小猪", "姓名")
money := flag.Float64("money",10.36,"钱")

其他的照葫芦画瓢就可以了,这些函数的参数都是一样的效果

  • 参数1 flag 名称
  • 参数2 默认值
  • 参数3 帮助信息

这里再来敲一下黑板,上述的 port,name,money 都是指针类型的哦

绑定 flag 变量

如果用不习惯上述方式来注册 flag 的话,我们也是可以先定义一个变量,然后将命令行参数绑定到变量上,如:

var city string
var food string
var age int

flag.StringVar(&city,"city","beijing","城市")
flag.StringVar(&food,"food","roast duck","美食")
flag.IntVar(&age,"age",18,"年龄")

在 flag 包中,我们可以看到也有一堆关于各种数据类型绑定变量的函数

如:flag.StringVar(&city,"city","beijing","城市")

参数类型和上述说到的类似,多了一个参数放在最前面

  • 参数1 具体的变量地址
  • 参数2 flag 名称
  • 参数3 默认值
  • 参数4 帮助信息

flag.Parse() 的使用方式

前面说了那么多的铺垫,那么我应该如何来使用呢?不可能我们都只去使用默认值吧

自然是自己开发自定义的命名行工具提供出去使用咯,这就需要咱们使用 flag.Parse() 了

flag.Parse() 来解析命令行参数写入注册的 flag 里,解析出来的是值,还是指针,这个取决于上面咱们注册 flag 的方式,再次敲黑板,此处可别弄错了

例如咱们写一个简单的 demo

此处,我们可以看到 port 是注册的自身的flag 的,port 是一个指针,所以咱们通过 flag.Parse() 之后,使用 port 时需要取地址

咱们可以参看效果如下:

编译执行可执行程序之后,我们可以看到,输出的是默认值,另外,咱们机上 --help 参数,就可以看到我们编写的帮助信息

站在使用者的角度,去使用这个工具,按照提示输入相关信息之后,我们可以看到输出的结果是我们所期望的

当然此处需要提几个注意事项:

  • 上述我们可以看到,咱们给对应的 flag 传值的时候是, -flag xxx,我们也是可以 --flag xxx,我们还可以 -flag=xxx ,同理 --flag=xxx 也是 ok 的

当然,如果对 golang 自身对于 flag 包的实现方式感兴趣的兄弟,也可以好好看看 flag 包中对于命名行参数的字符串数据解析

func (f *FlagSet) ``parseOne``() (bool, error) {