您现在的位置是:首页 > Golang>日常简单笔记>Golang获取命令行参数(通过命令行指定参数控制程序运行) 网站首页 日常简单笔记
Golang获取命令行参数(通过命令行指定参数控制程序运行)
两种方式,本人使用第二种.
1、golang os库获取命令行参数
os可以通过变量Args来获取命令参数,os.Args返回一个字符串数组,其中第一个参数就是执行文件本身。
package main import ( "fmt" "os" ) func main() { fmt.Println(os.Args) }
编译执行后执行
$ ./cmd -user="root" [./cmd -user=root]
这种方式操作起来要自己封装,比较费时费劲。golang提供了flag库,可以很方便的操作命名行参数,下面介绍下flag的用法。
2、golang flag获取命令行参数
package main import ( "flag" "fmt" ) func main() { targetUrl := flag.String("url","","目标站点地址") path := flag.String("path","","目标站点地址") Parallelnum := flag.Int("parallel",500,"并发数") Delay := flag.Int("delay",0,"延迟(毫秒)") MaxLinks := flag.Int("maxlink",50000,"最大链接数") //[必须调用]:从 arguments 中解析注册的 flag,不然参数获取都没值 flag.Parse() fmt.Println("url:", *targetUrl ) ... }
像flag.Int、flag.Bool、flag.String这样的函数格式都是一样的,第一个参数表示参数名称,第二个参数表示默认值,第三个参数表示使用说明和描述。flag.StringVar这样的函数第一个参数换成了变量地址,后面的参数和flag.String是一样的。
使用flag来操作命令行参数,支持的格式如下四种:
-url=https://www.uppdd.com/ --url=https://www.uppdd.com/ -url https://www.uppdd.com/ --url https://www.uppdd.com/
执行一下:
$ go run flag.go -url=https://www.uppdd.com/ -path="/public/" url: https://www.uppdd.com/
使用-h参数可以查看使用帮助:
$ go run flag.go -h
-delay int 延迟(毫秒) -maxlink int 最大链接数 (default 50000) -parallel int 并发数 (default 500) -path string 目标站点地址 -url string 目标站点地址
随机推荐
文章评论
未开放