您现在的位置是：首页 > 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
        目标站点地址