部署golang项目时难免要通过命令行来设置一些参数,那么在golang中如何操作命令行参数呢?可以使用os库和flag库。

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() {
ok := flag.Bool("ok", false, "is ok")
id := flag.Int("id", , "id")
port := flag.String("port", ":8080", "http listen port")
var name string
flag.StringVar(&name, "name", "", "name") flag.Parse() fmt.Println("ok:", *ok)
fmt.Println("id:", *id)
fmt.Println("port:", *port)
fmt.Println("name:", name)
}

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

使用flag来操作命令行参数,支持的格式如下:

-id=
--id=
-id
--id

执行一下:

$ go run flag.go -id= -name="golang"
ok: false
id:
port: :
name: golang

使用-h参数可以查看使用帮助:

$ go run flag.go -h
-id=: id
-name="": name
-ok=false: is ok
-port=":8080": http listen port