源码文件又分为三种,即:命令源码文件、库源码文件和测试源码文件,它们都有着不同的用途和编写规则。
命令源码文件是程序的运行入口,是每个可独立运行的程序必须拥有的。我们可以通过构建或安装生成与其对应的可执行文件,后者一般会与该命令源码文件的直接父目录同名。
如果一个源码文件声明属于main包,并且包含一个无参数声明且无结果声明的main函数,那么就是命令源码文件。就像这样:
package main
import "fmt"
func main() {
fmt.Println("Hello, world!")
}
我们使用 go run 运行该段代码,就可看到 Hello, world!
无论是 Linux 还是 Windows,如果你用过命令行(command line)的话,肯定就会知道几乎所有命令(command)都是可以接收参数(argument)的。通过构建或安装命令源码文件生成的可执行文件就可以被视为“命令”,既然是命令,那么就应该具备接收参数的能力。
package main
import (
"os"
"flag"
"fmt"
)
var name string
func init() {
flag.StringVar(&name, "name", "everyone", "The greeting object.")
}
func main() {
flag.Parse()
fmt.Printf("Hello, %s!\n", name)
}
运行结果
PS D:\GO\src\commad_src_file_test> go run .\command_src_file.go -name xx
Hello, xx!
flag是Go语言标准库中的代码包,用来接收和解析命令参数
flag.StringVar 有四个参数, 第一个参数是用于存储该命令参数值的地址, 第二个参数是该命令参数的名称, 第三个是该参数的默认值, 第四个是命令参数的说明
PS D:\GO\src\commad_src_file_test> go run .\command_src_file.go --help
Usage of C:\Users\84030\AppData\Local\Temp\go-build412649366\b001\exe\command_src_file.exe:
-name string
The greeting object. (default "everyone")
exit status 2
1、使用 flag.Usage重新赋值, flag.Usage的类型是func(), 即一种无参数声明且无结果声明的函数类型
package main
import (
"os"
"flag"
"fmt"
)
var name string
func init() {
flag.StringVar(&name, "name", "everyone", "The greeting object.")
}
func main() {
flag.Usage = func() {
fmt.Fprintf(os.Stderr, "Usage of %s:\n", "question")
flag.PrintDefaults()
}
flag.Parse()
fmt.Printf("Hello, %s!\n", name)
}
PS D:\GO\src\commad_src_file_test> go run .\command_src_file.go --help
Usage of question:
-name string
The greeting object. (default "everyone")
exit status 2
2、使用 flag.CommandLine
package main
import (
"os"
"flag"
"fmt"
)
var name string
func init() {
flag.StringVar(&name, "name", "everyone", "The greeting object.")
}
func main() {
flag.CommandLine = flag.NewFlagSet("", flag.ExitOnError)
flag.CommandLine.Usage = func() {
fmt.Fprintf(os.Stderr, "Usage of %s:\n", "question")
flag.PrintDefaults()
}
flag.Parse()
fmt.Printf("Hello, %s!\n", name)
}
运行结果
PS D:\GO\src\commad_src_file_test> go run .\command_src_file.go --help
Usage of question:
exit status 2
将 flag.CommandLine 代码改为
flag.CommandLine = flag.NewFlagSet("", flag.PanicOnError)
运行结果
PS D:\GO\src\commad_src_file_test> go run .\command_src_file.go --help
Usage of question:
panic: flag: help requested
goroutine 1 [running]:
flag.(*FlagSet).Parse(0xc0000440c0, 0xc000004430, 0x1, 0x1, 0xc0000401f0, 0xc0000440c0)
E:/Go/src/flag/flag.go:938 +0x107
flag.Parse()
E:/Go/src/flag/flag.go:953 +0x76
main.main()
D:/GO/src/commad_src_file_test/command_src_file.go:25 +0x74
exit status 2
flag.ExitOnError含义是,告诉命令参数容器,当命令后跟–help或者参数设置的不正确的时候,在打印命令参数使用说明后以状态码2结束当前程序。
状态码2代表用户错误地使用了命令,而flag.PanicOnError与之的区别是在最后抛出“运行时恐慌(panic)”。
还可以使用这种方式,更灵活
package main
import (
"os"
"flag"
"fmt"
)
var name string
var cmdLine = flag.NewFlagSet("question", flag.ExitOnError)
func init() {
cmdLine.StringVar(&name, "name", "everyone", "The greeting object.")
}
func main() {
cmdLine.Parse(os.Args[1:]) // 给定的命令参数
fmt.Printf("Hello, %s!\n", name)
}