原文地址
CLI或者“command line interface”是用户在命令行下交互的程序。由于通过将程序编译到一个静态文件中来减少依赖,一次Go特别适合开发CLI程序。如果你编写过安装时需要各种依赖的CLI程序你就知道这个是有多重要了。
?? 在这篇博客中我们将介绍使用Go开发CLI的基本知识。
Arguments
?? 大多数CLI程序都需要输入一些参数。Go 语言将这些参数以字符串slice处理。
查找当前应用的名字。
package main
import (
"fmt"
"os"
)
func main() {
// Program Name is always the first (implicit) argument
cmd := os.Args[0]
fmt.Printf("Program Name: %s\n", cmd)
}
这个应用再code/example1下,你可以用一下命令编译运行:
输出的结果是:
判断传入程序的参数数量
?? 为了确定有多少参数传入,可以计算所有参数的长度减1(记住,第一个参数总是程序的名字)。或者可以直接从os.Args[1:]来判断他的长度。
package main
import (
"fmt"
"os"
)
func main() {
argCount := len(os.Args[1:])
fmt.Printf("Total Arguments (excluding program name): %d\n", argCount)
}
运行./example2 得到的结果将是0。运行./example2 -foo=bar 得到的记过将是1。
遍历参数
?? 下面是一个很快速的遍历参数的例子。
package main
import (
"fmt"
"os"
)
func main() {
for i, a := range os.Args[1:] {
fmt.Printf("Argument %d is %s\n", i+1, a)
}
}
Running the program with ./example3 -local u=admin --help results in:
Argument 1 is -local
Argument 2 is u=admin
Argument 3 is --help
Flag 包
?? 目前为止我们已经知道如何在一个程序中查找参数的基本的方法。在这个级别查询他们并且将他们赋值给我们的程序是很麻烦的。所有就有了Flag包。
package main
import (
"flag"
"fmt"
)
func main() {
var port int
flag.IntVar(&port, "p", 8000, "specify port to use. defaults to 8000.")
flag.Parse()
fmt.Printf("port = %d", port)
}
我们首先做的是设置一个int类型的默认值是8000,并且有文字提示的标识。
?? 为了让flag包对设置的变量赋值,需要是用flag.Parse()方法。
?? 不加参数的运行这个程序得到的结果是port = 8000,因为我们明确的指定了如果没有参数传递给port,那么就采用默认的8000.
?? 运行./example4 -p=9000 结果是 port = 9000
?? 同事flag提供了 “program useage”的输出。如果我们运行 ./example4 -help 我们会得到:
Usage of ./example4:
-p=8000: specify port to use. defaults to 8000.
flag.Args()
?? 很多CLI程序同时包含有标识和没有标识的参数。flag.Args() 将会直接返回哪些没有标识的参数。
package main
import (
"flag"
"fmt"
)
func main() {
var port int
flag.IntVar(&port, "p", 8000, "specify port to use. defaults to 8000.")
flag.Parse()
fmt.Printf("port = %d\n", port)
fmt.Printf("other args: %+v\n", flag.Args())
}
运行./example5 -p=9000 foo=10 -bar 将会得到:
port = 9000
other args: [foo=10 -bar]
flag只要找到一个不包含的flag就会立即停止查询。
无效的flag参数
?? Go是一个强语言类型,所以如果我们传递一个string给一个int类型的flag,它将会提示我们:
package main
import (
"flag"
"fmt"
)
func main() {
var port int
flag.IntVar(&port, "p", 8000, "specify port to use. defaults to 8000")
flag.Parse()
fmt.Printf("port = %d", port)
}
运行程序./example6 -p=foo 得到的结果是:
invalid value “foo” for flag -p: strconv.ParseInt: parsing “foo”: invalid syntax
Usage of ./example6:
-p=8000: specify port to use. defaults to 8000
flag不仅会提示我们输入错误,同时还会输出默认的使用方法。
flag.Usage
?? flag包声明了一个Usage的方法。这样我们就可以输出我们想要输出的Usage了。
package main
import (
"flag"
"fmt"
"os"
)
func main() {
flag.Usage = func() {
fmt.Printf("Usage of %s:\n", os.Args[0])
fmt.Printf(" example7 file1 file2 ...\n")
flag.PrintDefaults()
}
flag.Parse()
}
运行