go Cobra命令行工具入门

作者：@Go和分布式IM
本文为作者原创，转载请注明出处：https://www.cnblogs.com/wishFreedom/p/16407894.html

简介

Star：26.5K
 
Cobra是一个用Go语言实现的命令行工具。并且现在正在被很多项目使用，例如：Kubernetes、Hugo和Github CLI等。通过使用Cobra，我们可以快速的创建命令行工具，特别适合写测试脚本，各种服务的Admin CLI等。
 
比如 Mattermost 项目，就写了很多 Admin CLI：
 

为什么需要cobra

我们看一个简单的demo

使用前

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package main
  
import (
    "flag"
    "fmt"
)
  
func main() {
    flag.Parse()
  
    args := flag.Args()
    if len(args) <= 0 {
        fmt.Println("Usage:  admin-cli [command]")
        return
    }
  
    switch args[0] {
    case "help":
        // ...
    case "export":
        //...
        if len(args) == 3 { // 导出到文件
            // todo
        } else if len(args) == 2 { // 导出...
            // todo
        }
    default:
        //...
    }
}

使用后

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package main
  
import (
    "fmt"
    "github.com/spf13/cobra"
    "os"
)
  
// rootCmd represents the base command when called without any subcommands
var rootCmd = &cobra.Command{
    Use:   "api",
    Short: "A brief description of your application",
    Long:  `A longer description `,
}
  
// 命令一
var mockMsgCmd = &cobra.Command{
    Use:   "mockMsg",
    Short: "批量发送测试文本消息",
    Long:  ``,
    Run: func(cmd *cobra.Command, args []string) {
        fmt.Println("mockMsg called")
    },
}
  
// 命令二
var exportCmd = &cobra.Command{
    Use:   "export",
    Short: "导出数据",
    Long:  ``,
    Run: func(cmd *cobra.Command, args []string) {
        fmt.Println("export called")
    },
}
  
func Execute() {
    err := rootCmd.Execute()
    if err != nil {
        os.Exit(1)
    }
}
  
func init() {
    rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
  
    rootCmd.AddCommand(mockMsgCmd)
    rootCmd.AddCommand(exportCmd)
  
    exportCmd.Flags().StringP("out", "k", "./backup", "导出路径")
}
  
func main() {
    Execute()
}

　　

运行：
?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
$ go run main.go
A longer description
  
Usage:
  api [command]
  
Available Commands:
  completion  Generate the autocompletion script for the specified shell
  export      导出数据
  help        Help about any command
  mockMsg     批量发送测试文本消息
  
Flags:
  -h, --help     help for api
  -t, --toggle   Help message for toggle
  
Use "api [command] --help" for more information about a command.

　　

发现了吗？你不用再处理各种参数组合了，从此释放了出来，只需要写自己的业务逻辑即可！

基本概念

Cobra由三部分组成：
  • 命令(Commands )：代表行为。命令是程序的中心点，程序的每个功能都应该可以通过命令进行交互，一个命令可以有任意个子命令。
  • 参数(Args)：命令的参数
  • 标志(Flags)：修饰命令。它修饰命令该如何完成。
官方推荐命令格式为：
?
1
$ ./appName command args --Flag 　
 
如 hugo server --port=1313 ：
  • appName: hugo 
  • command: server 
  • flag: port

安装

Go pkg

添加依赖
?
1
$ go get -u github.com/spf13/cobra@latest

　　

导入即可：
?
1
import "github.com/spf13/cobra"　　

命令行工具

建议安装命令行工具 `cobra-cli` ，以方便快速创建cobra项目，增加command等。
 
?
1
2
# 命令行工具
$ go install github.com/spf13/cobra-cli@latest

　　

安装完成之后，执行 `cobra-cli --help` （请确保GOBIN已配置），输出下列信息则代表成功：
?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
$ cobra-cli --help
Cobra is a CLI library for Go that empowers applications.
This application is a tool to generate the needed files
to quickly create a Cobra application.
  
Usage:
  cobra-cli [command]
  
Available Commands:
  add         Add a command to a Cobra Application
  completion  Generate the autocompletion script for the specified shell
  help        Help about any command
  init        Initialize a Cobra Application
  
Flags:
  -a, --author string    author name for copyright attribution (default "YOUR NAME")
      --config string    config file (default is $HOME/.cobra.yaml)
  -h, --help             help for cobra-cli
  -l, --license string   name of license for the project
      --viper            use Viper for configuration
  
Use "cobra-cli [command] --help" for more information about a command.

　　

入门实践

新建cobra命令行程序

安装了cobra-cli工具之后，执行 init 初始化创建项目：
?
1
$ cobra-cli init

　　

此时，在当前目录自动生成如下文件：
?
1
2
3
4
├── LICENSE
├── cmd
│   └── root.go
└── main.go

　　

main.go：
?
1
2
3
4
5
6
7
package main
  
import "tools/api/cmd"
  
func main() {
   cmd.Execute()
}

 

?
1
root.go（有删减）：
?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
package cmd
  
import (
   "fmt"
  
   "github.com/spf13/cobra"
)
  
// rootCmd represents the base command when called without any subcommands
var rootCmd = &cobra.Command{
   Use:   "api",
   Short: "A brief description of your application",
   Long:  `A longer description `,
   //Run: func(cmd *cobra.Command, args []string) {
   // fmt.Println("api called")
   //},
}
  
// Execute adds all child commands to the root command and sets flags appropriately.
// This is called by main.main(). It only needs to happen once to the rootCmd.
func Execute() {
   err := rootCmd.Execute()
   if err != nil {
      os.Exit(1)
   }
}
  
func init() {
   // 全局flag
   // rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.api.yaml)")
  
   // local flag，暂不知道用处
   rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}

　

此时运行，不用指定参数，会执行rootCmd，打印使用说明：
?
1
2
$ go build
$ ./api

　　

输出：
?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
A longer description
  
Usage:
  api [command]
  
Available Commands:
  completion  Generate the autocompletion script for the specified shell
  help        Help about any command
  
Flags:
  -h, --help     help for api
  -t, --toggle   Help message for toggle
  
Use "api [command] --help" for more information about a command.

　　

命令构成

分析上面的默认输出：
  • Available Commands：代表可以执行的命令。比如./api connect
  • Flags：是参数。比如./api connect --ip=127.0.0.1:6379，--ip就是flag，127.0.0.1:6379就是flag的值。

新增命令

我们来新增一个命令试试，这也是命令行程序的魅力，通过不同的参数执行不同的动作。
语法：
?
1
$ cobra-cli add [command]

　　

比如：
?
1
2
$ cobra-cli add mock-msg
mockMsg created at /Users/xxx/repo/tools/api

　　

此时，在cmd下会多一个文件（mock_msg.go），内容如下：
?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
package cmd
  
import (
   "fmt"
  
   "github.com/spf13/cobra"
)
  
var mockMsgCmd = &cobra.Command{
   Use:   "mockMsg",
   Short: "A brief description of your command",
   Long: `mock msg command`,
   Run: func(cmd *cobra.Command, args []string) {
      fmt.Println("mockMsg called")
   },
}
  
func init() {
   rootCmd.AddCommand(mockMsgCmd)
}

　　

再执行rootCmd：
?
1
2
$ go build
$ ./api

　　

会发现，多了一个命令：
?
1
2
3
4
5
6
// ...
Available Commands:
  completion  Generate the autocompletion script for the specified shell
  help        Help about any command
  mockMsg     A brief description of your command
// ...

　　

执行mocMsg命令：
?
1
2
3
$ ./api mockMsg
  
mockMsg called

　　

此时，就可以在生成的 mock_msg.go: Run() 函数中，放你自己的业务逻辑代码了。

如何显示自己的命令用法

上面新增了一个命令mockMsg，通过 ./api help 打印了命令和help，但是 `Use` 里面指定的内容打印到哪里去了呢？
 
这个时候，需要针对Command在指定help，此时就能打印这个命令的具体用法了。
?
1
2
3
4
5
6
7
8
9
10
./api mockMsg help
批量生产mq消息
  
Usage:
  benchmark mockmsg [flags]
  
Flags:
  -g, --goroutine int32   并发routine数量 (default 1)
  -h, --help              help for mockmsg
  -p, --packet int32      每个routine一秒写入mq的数量 (default 20)
?
1
<br>-g和-p是新增的2个flag：
?
1
2
3
4
5
6
func init() {
   mockmsgCmd.Flags().Int32P("goroutine", "g", 1, "并发routine数量")
   mockmsgCmd.Flags().Int32P("packet", "p", 20, "每个routine一秒写入mq的数量")
  
   rootCmd.AddCommand(mockmsgCmd)
}

　　

获取这2个值：
?
1
2
3
4
5
6
7
8
9
10
11
// mockmsgCmd represents the mockmsg command
var mockmsgCmd = &cobra.Command{
   Use:   "mockmsg",
   Short: "批量生产mq消息",
   Run: func(cmd *cobra.Command, args []string) {
      // 这里要写全名
      g, _ := cmd.Flags().GetInt32("goroutine")
      p, _ := cmd.Flags().GetInt32("packet")
      fmt.Println("mockmsg called,flags:g=", g, ",p=", p, ",args:", args)
   },
}

　　

执行：
?
1
2
$ go run main.go mockmsg -p 322 -g 5 args1 args2
mockmsg called,flags:g= 5 ,p= 322 ,args: [args1 args2]

总结

我们通过一个例子，介绍了使用cobra带来的好处。通过一个完整的入门实践，演示了创建项目、添加命令和使用的一些示例，希望对你有所帮助！
 
参考：
posted @   Go和分布式IM  阅读(1130)  评论(0)  编辑  收藏  举报
相关博文：
· 优雅退出在Golang中的实现
· Go死锁——当Channel遇上Mutex时
· Cobra 命令行工具
· Cobra 库上手—自建命令行工具
· Go命令行工具cobra
阅读排行：
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码，我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了，比商业数据库还牛
· 白话解读 Dapr 1.15：你的「微服务管家」又秀新绝活了
· 上周热点回顾（2.24-3.2）
点击右上角即可分享
微信分享提示