首页 > golang > go命令行参数os.Args和flag包
2021
01-25

go命令行参数os.Args和flag包

最近在学习golang,结果遇到了golang如何获取命令行参数?的疑问,下面顺便分享给大家

os库

os可以通过变量Args来获取命令参数,os.Args返回一个字符串数组,其中第一个参数就是执行文件本身。

package main
import (
   "fmt"
   "os"
)
func main() {
   fmt.Println("命令行的参数有", len(os.Args))
   // 遍历 os.Args 切片,就可以得到所有的命令行输入参数值
   for i, v := range os.Args {
       fmt.Printf("args[%v]=%v\n", i, v)
   }
}

执行结果:

> go run  ./main.go   laofan  123  456  789
命令行的参数有 5
args[0]=Temp\go-build\b001\exe\main.exe
args[1]=laofan
args[2]=123
args[3]=456
args[4]=789
>

flag包

如像 mysql 那样:mysql -u root -p 123456,可以指定输入内容,输入顺序可变。 flag包实现了命令行参数的解析。

使用flag来操作命令行参数,支持的格式如下:像flag.Int、flag.Bool、flag.String这样的函数格式都是一样的,第一个参数表示参数名称,第二个参数表示默认值,第三个参数表示使用说明和描述。flag.StringVar这样的函数第一个参数换成了变量地址,后面的参数和flag.String是一样的。

package main

import (
   "flag"
   "fmt"
)

func main() {
   // 定义几个变量,用于接收命令行的参数值
   var user        string
   var password    string
   var host        string
   var port        int
   // &user 就是接收命令行中输入 -u 后面的参数值,其他同理
   flag.StringVar(&user, "u", "root", "账号,默认为root")
   flag.StringVar(&password, "p", "", "密码,默认为空")
   flag.StringVar(&host, "h", "localhost", "主机名,默认为localhost")
   flag.IntVar(&port, "P", 3306, "端口号,默认为3306")
   // 解析命令行参数写入注册的flag里
   flag.Parse()
   // 输出结果
   fmt.Printf("user:%v\npassword:%v\nhost:%v\nport:%v\n",
       user, password, host, port)
}

执行结果:

> go run  ./main.go   -p 123 -u laofan -P 3306 -h 127.0.0.1
user:laofan
password:123
host:127.0.0.1
port:3306
>

第三种方法:flag包另一种写法

package main
import (
   "flag"
   "fmt"
)

func main () {
  //获取命令参数
 Act := flag.String("act", "", "is ok")
 date := flag.String("date", "", "id")
 // 解析命令行参数写入注册的flag里
 flag.Parse()
 fmt.Println("act :", *Act)
 fmt.Println("date :", *date)
}

执行结果:

> go run  ./main.go   -act laofan  -date today
act : laofan
date : today
>


本文》有 0 条评论

留下一个回复