文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

命令行参数的解析:Flag 库详解

2024-12-03 13:48

关注

在 Golang 程序中有很多种方法来处理命令行参数。

简单的情况下可以不使用任何库,直接使用 os.Args

  1. package main 
  2.  
  3. import ( 
  4.     "fmt" 
  5.     "os" 
  6.  
  7. func main() { 
  8.     if len(os.Args) > 0 { 
  9.         for index, arg := range os.Args { 
  10.             fmt.Printf("args[%d]=%v\n"index, arg) 
  11.         } 
  12.     } 

试着运行一下,第一个参数是执行文件的路径。

  1. $ go run demo.go hello world hello golang 
  2. args[0]=/var/folders/72/lkr7ltfd27lcf36d75jdyjr40000gp/T/go-build187785213/b001/exe/demo 
  3. args[1]=hello 
  4. args[2]=world 
  5. args[3]=hello 
  6. args[4]=golang 

从上面你可以看到,os.Args 只能处理简单的参数,而且对于参数的位置有严格的要求。对于一些比较复杂的场景,就需要你自己定义解析规则,非常麻烦。

如果真的遇上了所谓的复杂场景,那么还可以使用 Golang 的标准库 flag 包来处理命令行参数。

本文将介绍 Golang 标准库中 flag 包的用法。

1. 参数种类

根据参数是否为布尔型,可以分为两种:

根据参数名的长短,还可以分为:

2. 入门示例

我先用一个字符串类型的参数的示例,抛砖引玉

  1. package main 
  2.  
  3. import ( 
  4.     "flag" 
  5.     "fmt" 
  6.  
  7. func main(){ 
  8.     var name string 
  9.     flag.StringVar(&name"name""jack""your name"
  10.  
  11. flag.Parse()  // 解析参数 
  12.     fmt.Println(name

flag.StringVar 定义了一个字符串参数,它接收几个参数

运行以上程序,输出如下

  1. $ go run demo.go 
  2. jack 
  3.  
  4. $ go run demo.go --name wangbm 
  5. wangbm 

3. 改进一下

如果你的程序只接收很少的几个参数时,上面那样写也没有什么问题。

但一旦参数数量多了以后,一大堆参数解析的代码堆积在 main 函数里,影响代码的可读性、美观性。

建议将参数解析的代码放入 init 函数中,init 函数会先于 main 函数执行。

  1. package main 
  2.  
  3. import ( 
  4.     "flag" 
  5.     "fmt" 
  6.  
  7. var name string 
  8.  
  9. func init()  { 
  10.     flag.StringVar(&name"name""jack""your name"
  11.  
  12. func main(){ 
  13.     flag.Parse() 
  14.     fmt.Println(name

4. 参数类型

当你在命令行中指定了参数,Go 如何解析这个参数,转化成何种类型,是需要你事先定义的。

不同的参数,对应着 flag 中不同的方法。

下面分别讲讲不同的参数类型,都该如何定义。

布尔型

实现效果:当不指定 --debug 时,debug 的默认值为 false,你一指定 --debug,debug 为赋值为 true。

  1. var debug bool 
  2.  
  3. func init()  { 
  4.     flag.BoolVar(&debug, "debug"false"是否开启 DEBUG 模式"
  5.  
  6. func main(){ 
  7.     flag.Parse() 
  8.     fmt.Println(debug) 

运行后,执行结果如下

  1. $ go run main.go  
  2. false 
  3. $ go run main.go --debug 
  4. true 

数值型

定义一个 age 参数,不指定默认为 18

  1. var age int 
  2.  
  3. func init()  { 
  4.     flag.IntVar(&age, "age", 18, "你的年龄"
  5.  
  6. func main(){ 
  7.     flag.Parse() 
  8.     fmt.Println(age) 

运行后,执行结果如下

  1. $ go run main.go  
  2. 18 
  3.  
  4. $ go run main.go --age 20 
  5. 20 

int64、 uint 和 float64 类型分别对应 Int64Var 、 UintVar、Float64Var 方法,也是同理,不再赘述。

字符串

定义一个 name参数,不指定默认为 jack

  1. var name string 
  2.  
  3. func init()  { 
  4.     flag.StringVar(&name"name""jack""你的名字"
  5.  
  6. func main(){ 
  7.     flag.Parse() 
  8.     fmt.Println(name

运行后,执行结果如下

  1. $ go run main.go  
  2. jack 
  3.  
  4. $ go run main.go --name wangbm 
  5. wangbm 

时间类型

定义一个 interval 参数,不指定默认为 1s

  1. var interval time.Duration 
  2.  
  3. func init()  { 
  4.     flag.DurationVar(&interval, "interval", 1 * time.Second"循环间隔"
  5.  
  6. func main(){ 
  7.     flag.Parse() 
  8.     fmt.Println(interval) 

验证效果如下

  1. $ go run main.go  
  2. 1s 
  3.  
  4. $ go run main.go --interval 2s 
  5. 2s 

5. 自定义类型

flag 包支持的类型有 Bool、Duration、Float64、Int、Int64、String、Uint、Uint64。

这些类型的参数被封装到其对应的后端类型中,比如 Int 类型的参数被封装为 intValue,String 类型的参数被封装为 stringValue。

这些后端的类型都实现了 flag.Value 接口,因此可以把一个命令行参数抽象为一个 Flag 类型的实例。下面是 Value 接口和 Flag 类型的代码:

  1. type Value interface { 
  2.     String() string 
  3.     Set(string) error 
  4.  
  5. // Flag 类型 
  6. type Flag struct { 
  7.     Name     string // name as it appears on command line 
  8.     Usage    string // help message 
  9.     Value    Value  // value as set 是个 interface,因此可以是不同类型的实例。 
  10.     DefValue string // default value (as text); for usage message 
  11.  
  12. func Var(value Value, name string, usage string) { 
  13.     CommandLine.Var(value, name, usage) 

想要实现自定义类型的参数,其实只要 Var 函数的第一个参数对象实现 flag.Value接口即可

  1. type sliceValue []string 
  2.  
  3.  
  4. func newSliceValue(vals []string, p *[]string) *sliceValue { 
  5.     *p = vals 
  6.     return (*sliceValue)(p) 
  7.  
  8. func (s *sliceValue) Set(val string) error { 
  9.          // 如何解析参数值 
  10.     *s = sliceValue(strings.Split(val, ",")) 
  11.     return nil 
  12.  
  13. func (s *sliceValue) String() string { 
  14.     return strings.Join([]string(*s), ","

比如我想实现如下效果,传入的参数是一个字符串,以逗号分隔,flag 的解析时将其转成 slice。

  1. $ go run demo.go -members "Jack,Tom" 
  2. [Jack Tom] 

那我可以这样子编写代码

  1. var members []string 
  2. type sliceValue []string 
  3.  
  4.  
  5. func newSliceValue(vals []string, p *[]string) *sliceValue { 
  6.     *p = vals 
  7.     return (*sliceValue)(p) 
  8.  
  9. func (s *sliceValue) Set(val string) error { 
  10.          // 如何解析参数值 
  11.     *s = sliceValue(strings.Split(val, ",")) 
  12.     return nil 
  13.  
  14.  
  15. func (s *sliceValue) String() string { 
  16.     return strings.Join([]string(*s), ","
  17.  
  18. func init()  { 
  19.     flag.Var(newSliceValue([]string{}, &members), "members""会员列表"
  20.  
  21. func main(){ 
  22.     flag.Parse() 
  23.     fmt.Println(members) 

有的朋友 可能会对 (*sliceValue)(p) 这行代码有所疑问,这是什么意思呢?

关于这个,其实之前在 【2.9 详细图解:静态类型与动态类型】有讲过,忘记了可以前往复习。

6. 长短选项

flag 包,在使用上,其实并没有没有长短选项之别,你可以看下面这个例子

  1. package main 
  2.  
  3. import ( 
  4.     "flag" 
  5.     "fmt" 
  6.  
  7. var name string 
  8.  
  9. func init()  { 
  10.     flag.StringVar(&name"name""明哥""你的名字"
  11.  
  12. func main(){ 
  13.     flag.Parse() 
  14.     fmt.Println(name

通过指定如下几种参数形式

  1. $ go run main.go  
  2. 明哥 
  3. $ go run main.go --name jack 
  4. jack 
  5. $ go run main.go -name jack 
  6. jack 

一个 - 和两个 - 执行结果是相同的。

那么再加一个呢?

终于报错了。说明最多只能指定两个 -

  1. $ go run main.go ---name jack 
  2. bad flag syntax: ---name 
  3. Usage of /tmp/go-build245956022/b001/exe/main: 
  4.   -name string 
  5.         你的名字 (default "明哥"
  6. exit status 2 

7. 总结一下

flag 在绝大多数场景下,它是够用的,但如果要支持更多的命令传入格式,flag 可能并不是最好的选择。

那些在标准库不能解决的场景,往往会有相应的Go爱好者提供第三方解决方案。我所了解到的 cobra 就是一个非常不错的库。

 本文转载自微信公众号「Go编程时光」,可以通过以下二维码关注。转载本文请联系Go编程时光公众号。

 

来源: Go编程时光内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯