Golang之命令行操作(第三章)

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

参考

简书说明系列
https://www.jianshu.com/c/bef4e958ecd6

视频教学系列
https://www.imooc.com/learn/1021

正文

  1. 进入到项目代码 main包的路径下执行 go install命令,会在 GOPATH/bin路径下面生成一个可执行文件,比如 coin.exe
  2. 然后通过 coin 命令 参数的形式执行 go代码,每次输入一次,都表示执行一次main方法(支持带参数的执行main方法),而不是启动一个进程持续监听

代码实现:依赖 flag和 os 类库,读取执行main方法的参数的个数和参数值
flag指向一个命令,os用来获取参数的值和个数

  • 默认命令算一个参数 比如 coin addCoin 命令就是2个参数
func (cli *CLI) validateArgs() {
    if len(os.Args) < 2 {
        cli.printUsage()
        os.Exit(1)
    }
}

完整调用

// 入口,会完成NewBlockchain的初始化(新链的话,就创建一个创世币,老链的话,就读取数据库,加载到内存中)
func main() {
    bc := NewBlockchain()
    defer bc.db.Close()
    cli := CLI{bc}
    cli.Run()
}

根据命令执行相应的响应

func (cli *CLI) Run() {
    cli.validateArgs()
    addBlockCmd := flag.NewFlagSet("addblock", flag.ExitOnError)
    printChainCmd := flag.NewFlagSet("printchain", flag.ExitOnError)

    addBlockData := addBlockCmd.String("data", "", "Block data")

    switch os.Args[1] {
    case "addblock":
        err := addBlockCmd.Parse(os.Args[2:])
        if err != nil {
            log.Panic(err)
        }
    case "printchain":
        err := printChainCmd.Parse(os.Args[2:])
        if err != nil {
            log.Panic(err)
        }
    default:
        cli.printUsage()
        os.Exit(1)
    }

    if addBlockCmd.Parsed() {
        if *addBlockData == "" {
            addBlockCmd.Usage()
            os.Exit(1)
        }
        cli.bc.AddBlock(*addBlockData)
    }

    if printChainCmd.Parsed() {
        cli.printChain()
    }
}

有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:黄靠谱

查看原文:Golang之命令行操作(第三章)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

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