文档章节

Go笔记-flag参数解析

漂泊尘埃
 漂泊尘埃
发布于 2017/02/27 17:22
字数 759
阅读 181
收藏 0

先查看godoc文档

first demo

习自 https://github.com/on99/gocyclo/blob/master/gocyclo.go

package flag_demo

import (
	"flag"
	"fmt"
	"os"
)

const usageDoc = `Calculate cyclomatic complexities of Go functions.
Usage:
		gocyclo [flags] <Go file or directory> ...
		
Flags:
		-over N balabala..
		-top N balabala...
		-avg balabala...

...
`

func usage() {
	fmt.Fprintf(os.Stderr, usageDoc)
	os.Exit(2)
}

var (
	over = flag.Int("over", 0, "show functions with complexity > N only")
	top  = flag.Int("top", -1, "show the top N most complex functions only")
	avg  = flag.Bool("avg", false, "show the average complexity")
)

func MyMain() {
	flag.Usage = usage
	flag.Parse()

	args := flag.Args()
	if len(args) == 0 {
		usage()
	}
	fmt.Println("args: ", args)

	fmt.Println("over: ", *over)
	fmt.Println("top: ", *top)
	fmt.Println("avg: ", *avg)
}

使用1:

./src

输出:

Calculate cyclomatic complexities of Go functions.
Usage:
		gocyclo [flags] <Go file or directory> ...
		
Flags:
		-over N balabala..
		-top N balabala...
		-avg balabala...

...

使用2:

./src -over=5 -top=9 -avg=true some thing

输出:

args:  [some thing]
over:  5
top:  9
avg:  true

利用VisitAll自动生成flag信息

package main

import (
	"flag"
	"fmt"
	"os"
)

func main() {
	usage := GenerateUsageFunc()
	ParseFlag(usage)
	fmt.Println(myvar)
}

var myvar string

func init() {
	flag.StringVar(&myvar, "myarg", "def", "whatever...")
}

func GenerateUsageFunc() func() {
	name := os.Args[0] // 程序路径
	return func() {
		fmt.Fprintf(os.Stderr, "Usage of %s:\n", name)
		fmt.Fprintf(os.Stderr, "\t%s [flags] \n", name)
		fmt.Fprintf(os.Stderr, "Flags:\n")
		// 这里就是自动生成 flag 的信息
		flag.VisitAll(func(fg *flag.Flag) {
			fmt.Fprintf(os.Stderr, "  -%s = %s (%s)\n", fg.Name, fg.DefValue, fg.Usage)
		})
	}
}

func ParseFlag(usageFunc func()) {
	// CommandLine 是 flag 包中某些全局函数的默认实现,其实就是一个 FlagSet
	flag.CommandLine.Init(os.Args[0], flag.ExitOnError)
	flag.CommandLine.SetOutput(os.Stderr)
	flag.CommandLine.Usage = usageFunc
	flag.Parse()
	flag.VisitAll(func(fg *flag.Flag) {
		fmt.Printf("your flag: [%10s]=%10s \n", fg.Name, fg.Value.String())
	})
}

输出帮助信息 test.exe -h:

Usage of D:\go\code\test\bin\test.exe:
        D:\go\code\test\bin\test.exe [flags]
Flags:
  -myarg = def (whatever...)

设置正确的参数 test.exe -myarg=a

your flag: [     myarg]=         a
a

设置错误的参数 test.exe -myerr=a

flag provided but not defined: -myerr
Usage of D:\go\code\test\bin\test.exe:
        D:\go\code\test\bin\test.exe [flags]
Flags:
  -myarg = def (whatever...)

命令行参数

http://golang-china.github.io/gopl-zh/ch1/ch1-02.html

os.Args

os.Args这个变量是一个字符串(string)的slice

os.Args的第一个元素,即os.Args[0]是命令行执行时的命令本身;其它的元素则是执行该命令时传给这个程序的参数

os.Args vs flag.Args

func main() {
	flag.Parse()
	fmt.Println(os.Args)
	fmt.Println(flag.NArg(), flag.Args())
}

执行命令:

./test cmd1 cmd2 -arg1=v1 -arg2=v2

执行结果:

[./test cmd1 cmd2 -arg1=v1 -arg2=v2]
4 [cmd1 cmd2 -arg1=v1 -arg2=v2]

os.Argsflag.Args 多了一个运行路径,但 flag 需要先 parse

如果添加了下面的代码:

func init() {
	flag.String("arg1", "", "")
	flag.String("arg2", "", "")
}

再次执行 ./test cmd1 cmd2 -arg1=v1 -arg2=v2 时输出的就是:

[./test -arg1=v1 -arg2=v2 cmd1 cmd2]
2 [cmd1 cmd2]

flag.Args() 没有返回已经被解析的 -arg1-arg2

flag set

var (
	flagset = flag.NewFlagSet("test", flag.ExitOnError)
	arg1    = flagset.String("arg1", "arg1value", "usage of arg1")
)

func main() {
	flagset.Parse(os.Args[1:])
	fmt.Println(*arg1)
}

lookup获取的值的类型

var flagset = flag.NewFlagSet("hello", flag.ExitOnError)

func init() {
	flagset.Int("a", 1, "usage")
}

func main() {
	flagset.Parse(os.Args[1:])
	vv := flagset.Lookup("a").Value.(flag.Getter).Get().(int)
	fmt.Println(vv)
}

多个flag set

package main

import (
	"flag"
	"fmt"
	"os"
)

var flagset = flag.NewFlagSet("test", flag.ExitOnError)
var flagset2 = flag.NewFlagSet("test2", flag.ExitOnError)

func init() {
	flagset.String("arg1", "value1", "haha")
	flagset2.String("arg1", "value2", "hehe")
}

func main() {
	flagset.Parse(os.Args[1:])
	flagset2.Parse(os.Args[1:])

	f := flagset.Lookup("arg1")
	fmt.Println(f.Value)

	f = flagset2.Lookup("arg1")
	fmt.Println(f.Value)
}

usage 只显示第一个flagset的:

Usage of test:
  -arg1 string
        haha (default "value1")

并且设置了arg1后两个flagset都是可以获取到新值的。

-arg1=abc

如果两个flagset的变量一个是arg1,一个是arg2,那么设置了其中一个,另一个就会报错,没有定义这个变量,所以,flagset只能用一个

© 著作权归作者所有

共有 人打赏支持
上一篇: Go笔记-函数
下一篇: Go笔记-文件
漂泊尘埃

漂泊尘埃

粉丝 5
博文 36
码字总数 71385
作品 0
朝阳
私信 提问
golang flag包使用笔记

本文原文地址:flag - 命令行参数解析 在写命令行程序(工具、server)时,对命令参数进行解析是常见的需求。各种语言一般都会提供解析命令行参数的方法或库,以方便程序员使用。如果命令行参...

我的小碗汤
05/13
0
0
Raphael学习笔记(5)--绘图(路径【椭圆曲线】)

1、椭圆曲线简介 前面的学习笔记:Raphael学习笔记(3)--绘图(路径【直线】)简单的描述了绘制椭圆曲线的参数,但没有详细的介绍各个参数的含义,这次,我们就详细的学习一下椭圆曲线的绘制...

i33
2013/01/16
0
0
android笔记4-xml解析

SAXParserFactory factory = SAXParserFactory.newInstance(); XMLReader reader = factory.newSAXParser().getXMLReader(); reader.setContentHandler(new InfoXMLHander()); reader.parse(......

迷路的阿修罗
2013/01/28
0
2
Go语言学习笔记 package

加 Golang学习 QQ群共同学习进步成家立业工作 ^-^ 群号:96933959 strings strings包实现了用于操作字符的简单函数。 strings.Index("suoning", "n") //3(子串第一次出现的位置,不存在则返回...

xumaojun
03/12
0
0
golang flag 解析入参

首先一个例子: package mainimport ( "fmt" "flag") func main(){ data_path := flag.String("D","/home/manu/sample/","DB data path") log_file := flag.String("l","/home/manu/sample.l......

冷絮
2015/08/25
0
0

没有更多内容

加载失败,请刷新页面

加载更多

Java程序员可知为何公司宁花25K重新招人,也不花20K留住老员工?

身在职场,经常会暗自打听同事工资,尤其是得知身边新入职同事的工资居然比自己高,还高出一大截时,心里自然很不平衡,一心想要离职。 那么,为什么公司宁愿花高价招聘新员工也不愿意给老员...

Java填坑路
5分钟前
1
0
阿里云文件存储(NAS)助力业务系统承载双十一尖峰流量

2018天猫双11全球狂欢节,全天成交额再次刷新纪录达到2135亿元,其中总成交额在开场后仅仅用了2分05秒即突破100亿元,峰值的交易量达到惊人的高度,背后离不开阿里云大数据计算和存储能力的支...

阿里云云栖社区
9分钟前
0
0
【windows 找不到“\\192.168.X.X”,请检查拼写是否正确】错误&139端口和445端口区别

今天像往常一样,通过共享盘访问其他主机时,却出现如题的错误,百思不得其解,毕竟没有改动什么配置啊,地址也没输错啊。 然后就在网上看到了一篇博客,才想起来上周末因为考虑到安全问题,...

fang_faye
9分钟前
0
0
示例Express中路由规则及获取请求参数

本次给大家分享一篇基于express中路由规则及获取请求参数的方法,写的十分的全面细致,具有一定的参考价值,对此有需要的朋友可以参考学习下。如有不足之处,欢迎批评指正。 express中常见的...

前端攻城小牛
13分钟前
0
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部