类似 go get –u 的命令行参数实现

我们可能需要类似 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