命令行参数
代码实例
// [_命令行参数_](http://en./wiki/Command-line_interface#Arguments)
// 是指定程序运行参数的一个常见方式。例如,`go run hello.go`,
// 程序 `go` 使用了 `run` 和 `hello.go` 两个参数。
package main
import "os"
import "fmt"
func main() {
// `os.Args` 提供原始命令行参数访问功能。注意,切片中
// 的第一个参数是该程序的路径,并且 `os.Args[1:]`保存
// 所有程序的的参数。
argsWithProg := os.Args
argsWithoutProg := os.Args[1:]
// 你可以使用标准的索引位置方式取得单个参数的值。
arg := os.Args[3]
fmt.Println(argsWithProg)
fmt.Println(argsWithoutProg)
fmt.Println(arg)
}
运行程序
# 要实验命令行参数,最好先使用 `go build` 编译一个可执行
# 二进制文件
$ go build command-line-arguments.go
$ ./command-line-arguments a b c d
[./command-line-arguments a b c d]
[a b c d]
c
# 下面我们要看看更高级的使用标记的命令行处理方法。
命令行标志
代码实例
// [_命令行标志_](http://en./wiki/Command-line_interface#Command-line_option)
// 是命令行程序指定选项的常用方式。例如,在 `wc -l` 中,
// 这个 `-l` 就是一个命令行标志。
package main
// Go 提供了一个 `flag` 包,支持基本的命令行标志解析。
// 我们将用这个包来实现我们的命令行程序示例。
import "flag"
import "fmt"
func main() {
// 基本的标记声明仅支持字符串、整数和布尔值选项。
// 这里我们声明一个默认值为 `"foo"` 的字符串标志 `word`
// 并带有一个简短的描述。这里的 `flag.String` 函数返回一个字
// 符串指针(不是一个字符串值),在下面我们会看到是如何
// 使用这个指针的。
wordPtr := flag.String("word", "foo", "a string")
// 使用和声明 `word` 标志相同的方法来声明 `numb` 和 `fork` 标志。
numbPtr := flag.Int("numb", 42, "an int")
boolPtr := flag.Bool("fork", false, "a bool")
// 用程序中已有的参数来声明一个标志也是可以的。注
// 意在标志声明函数中需要使用该参数的指针。
var svar string
flag.StringVar(&svar, "svar", "bar", "a string var")
// 所有标志都声明完成以后,调用 `flag.Parse()` 来执行
// 命令行解析。
flag.Parse()
// 这里我们将仅输出解析的选项以及后面的位置参数。注意,
// 我们需要使用类似 `*wordPtr` 这样的语法来对指针解引用,从而
// 得到选项的实际值。
fmt.Println("word:", *wordPtr)
fmt.Println("numb:", *numbPtr)
fmt.Println("fork:", *boolPtr)
fmt.Println("svar:", svar)
fmt.Println("tail:", flag.Args())
}
运行程序
# 测试这个程序前,最好将这个程序编译成二进制文件,然后再运
# 行这个程序。
$ go build command-line-flags.go
word: opt
numb: 7
fork: true
svar: flag
tail: []
# 注意到,如果你省略一个标志,那么这个标志的值自动的设
# 定为他的默认值。
$ ./command-line-flags -word=opt
word: opt
numb: 42
fork: false
svar: bar
tail: []
# 位置参数可以出现在任何标志后面。
$ ./command-line-flags -word=opt a1 a2 a3
word: opt
...
tail: [a1 a2 a3]
# 注意,`flag` 包需要所有的标志出现位置参数之前(
# 否则,这个标志将会被解析为位置参数)。
$ ./command-line-flags -word=opt a1 a2 a3 -numb=7
word: opt
numb: 42
fork: false
svar: bar
trailing: [a1 a2 a3 -numb=7]
# 使用 `-h` 或者 `--help` 标志来得到自动生成的这个命
# 令行程序的帮助文本。
$ ./command-line-flags -h
Usage of ./command-line-flags:
-fork=false: a bool
-numb=42: an int
-svar="bar": a string var
-word="foo": a string
# 如果你提供一个没有使用 `flag` 包指定的标志,程序会输出一
# 个错误信息,并再次显示帮助文本。
$ ./command-line-flags -wat
flag provided but not defined: -wat
Usage of ./command-line-flags:
...
# 后面,我们将会看一下环境变量,另一个用于参数化程序的基本方式。
环境变量
代码实例
// [_环境变量_](http://zh./wiki/%E7%8E%AF%E5%A2%83%E5%8F%98%E9%87%8F)
// 是一个在[为 Unix 程序传递配置信息](http://www./config)的普遍方式。
// 让我们来看看如何设置,获取并列举环境变量。
package main
import "os"
import "strings"
import "fmt"
func main() {
// 使用 `os.Setenv` 来设置一个键值队。使用 `os.Getenv`
// 获取一个键对应的值。如果键不存在,将会返回一个空字符
// 串。
os.Setenv("FOO", "1")
fmt.Println("FOO:", os.Getenv("FOO"))
fmt.Println("BAR:", os.Getenv("BAR"))
// 使用 `os.Environ` 来列出所有环境变量键值队。这个函数
// 会返回一个 `KEY=value` 形式的字符串切片。你可以使用
// `strings.Split` 来得到键和值。这里我们打印所有的键。
fmt.Println()
for _, e := range os.Environ() {
pair := strings.Split(e, "=")
fmt.Println(pair[0])
}
}
运行程序
# 运行这个程序,显示我们在程序中设置的 `FOO` 的值,然而
# 没有设置的 `BAR` 是空的。
$ go run environment-variables.go
FOO: 1
BAR:
# 键的列表是由你的电脑情况而定的。
TERM_PROGRAM
PATH
SHELL
...
# 如果我们在运行前设置了 `BAR` 的值,那么运行程序将会获
# 取到这个值。
$ BAR=2 go run environment-variables.go
FOO: 1
BAR: 2
...