作用: 处理命令行参数
flag包绑定解析参数有三种定义方式
方法一 flag.xxx() 例如:flag.Int, flag.String, 返回解析变量类型的指针
package mainimport ("flag""fmt")func main() {host := flag.String("host", "127.0.0.1", "请输入host地址")port := flag.Int("port", 80, "请输入端口号")flag.Parse() // 解析参数fmt.Printf("%s:%d\n", *host, *port)}
执行: go run main.go -host=127.0.0.1 -port=80
输出:127.0.0.1:80
如果直接执行go run main.go, 这时候就会使用你的默认值
方法二 flag.XxxVar() 例如:flag.IntVar, flag.StringVar
package main
import (
"flag"
"fmt"
)
func main() {
var host string
var port int
flag.StringVar(&host, "host", "127.0.0.1", "请输入host地址")
flag.IntVar(&port, "port", 80, "请输入端口号")
flag.Parse() // 解析参数
fmt.Printf("%s:%d\n", host, port)
}
方法三 自定义参数解析flag.Var()
package main
import (
"flag"
"fmt"
"strings"
)
// 自定义类型
type HandsomeBoys []string
// 实现String()方法
func (h *HandsomeBoys) String() string {
return fmt.Sprintf("%v", *h)
}
// 实现Set方法,Set接口决定了如何解析flag的值
func (h *HandsomeBoys) Set(s string) error {
for _, v := range strings.Split(s, ",") {
*h = append(*h, v)
}
return nil
}
// 定义一个HandsomeBoys类型的变量
var names HandsomeBoys
func init() {
// 绑定变量boys
flag.Var(&boys, "boys", "请输入一组人名:-names=丁三,王五")
}
func main() {
flag.Parse()
fmt.Println(boys)
}
以上推荐使用第二种
