zoukankan      html  css  js  c++  java
  • 类似 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

  • 相关阅读:
    【2021-08-09】问题还需一点一点去改正
    【2021-08-08】连岳摘抄
    【2021-08-07】请教帖
    21春助教总结
    实践总结+技术博客评分
    来吧 ,来吧 自己搭建一个erp 系统
    博客索引
    「CCNU21暑期第六次周赛」
    「CCNU21暑期第五次周赛」
    「图论」连通性问题
  • 原文地址:https://www.cnblogs.com/ghj1976/p/5085874.html
Copyright © 2011-2022 走看看