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

ghj1976 · · 6464 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

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

本文来自:博客园

感谢作者:ghj1976

查看原文:类似 go get –u 的命令行参数实现

入群交流(该群和以上内容无关):Go中文网 QQ交流群:798786647 或 加微信入微信群:274768166 备注:入群; 公众号:Go语言中文网

6464 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传