我们可能需要类似 go get –u …. 这样的方式来实现我们的应用,这时候我们无法简单地使用 flag.Parse 了,而是要用 FlagSet 了, 使用例子如下:
package main
import (
"flag"
"log"
"os"
)
var ()
func main() {
// 读取命令参数,注意,这里根据执行任务的不同,参数也不一样
// 技术参考:
// http://stackoverflow.com/questions/24504024/defining-independent-flagsets-in-golang
// http://play.golang.org/p/eaEEx_EReX
if len(os.Args) <= 1 {
log.Println("请输入命令,支持的命令: esb、web、host!")
os.Exit(-1)
}
log.Println(os.Args[1])
cmd := os.Args[1]
// esb 客户端工具
fesb := flag.NewFlagSet("fesb", flag.ContinueOnError)
consumer := fesb.Int("c", 0, "ESB 关系中过滤消费者的编号,不输代表全部。")
provider := fesb.Int("p", 0, "ESB 关系中过滤生产者的编号,不输代表全部。")
dependType := fesb.String("t", "cache", "ESB关系依赖的类型,默认来自线上的缓存,可选项:config (数据库的配置),cache(线上调用的缓存)")
// web 站点
fweb := flag.NewFlagSet("fweb", flag.ContinueOnError)
webPort := fweb.Int("p", 81, "WEB 站点使用的端口号。")
// 修改host文件方式
fhost := flag.NewFlagSet("fhost", flag.ContinueOnError)
aaa := fhost.Bool("b", false, "test 参数!")
switch cmd {
case "esb":
err := fesb.Parse(os.Args[2:])
if err != nil {
log.Println("esb 参数类型错误,", err)
os.Exit(-1)
}
log.Println("dependType:", *dependType)
log.Println("consumer:", *consumer)
log.Println("provider:", *provider)
case "web":
err := fweb.Parse(os.Args[2:])
if err != nil {
log.Println("web 参数类型错误,", err)
os.Exit(-1)
}
log.Println("webPort:", *webPort)
case "host":
err := fhost.Parse(os.Args[2:])
if err != nil {
log.Println("host 参数类型错误,", err)
os.Exit(-1)
}
log.Println("aaa:", *aaa)
}
}
参考资料:
http://play.golang.org/p/eaEEx_EReX
http://stackoverflow.com/questions/24504024/defining-independent-flagsets-in-golang
https://golang.org/pkg/flag/#NewFlagSet
有疑问加站长微信联系(非本文作者)