golang获取命令行参数三种方法

作者: adm 分类: go 发布时间: 2022-05-07

最近在学习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
>

如果觉得我的文章对您有用,请随意赞赏。您的支持将鼓励我继续创作!