Golang Cobra的使用

在阅读Docker源码时发现，Docker命令行是使用cobra库创建的，为了更好的理解源码，先来熟悉一下Cobra。
Cobra既是一个用来创建强大的现代命令行应用的库，又是一个用来生成应用和命令文件的脚手架。
很多流行的Go项目都使用Cobra，例如Kubernetes, Hugo, rkt, etcd, Moby (former Docker), Docker (distribution), OpenShift, Delve, GopherJS, CockroachDB, Bleve, ProjectAtomic (enterprise), Giant Swarm’s gsctl, Nanobox/Nanopack, rclone, nehm, Pouch, Istio, Prototool, mattermost-server, Gardener, Linkerd等。

概念

命令(Commands)，参数(Args)和标识(Flags)是Cobra重要的三个概念。

• 命令(Commands)代表动作
• 参数(Args)代表事件
• 标示(Flags)是对动作的修饰
  好的命令行应该像自然语句一样流畅，让用户一眼就明白其作用。
  例如以下例子，docker是根命令，pull是命令。

docker pull hello-world

命令(Commands)

下列代码将docker作为根命令，pull作为其子命令。

package main
import (
	"fmt"
	"strings"
	"github.com/spf13/cobra"
)

func main() {
	var cmdPull = &cobra.Command{
		Use:   "pull [OPTIONS] NAME[:TAG|@DIGEST]",
		Short: "Pull an image or a repository from a registry",
		Run: func(cmd *cobra.Command, args []string) {
			fmt.Println("Pull: " + strings.Join(args, " "))
		},
	}

	var rootCmd = &cobra.Command{Use: "docker"}
	rootCmd.AddCommand(cmdPull)
	rootCmd.Execute()
}

运行结果

当然，这里只是展示cobra的基本使用方法，docker命令的具体实现见源码https://github.com/docker/cli/blob/v19.03.5/cmd/docker/docker.go

参数(Args)

通过Command的Args属性对参数进行校验。

标示(Flags)

标示分为全局标示（Persistent Flags）和局部标示（Local Flags）

全局标示（Persistent Flags）

全局标示（Persistent Flags）会作用于其指定的命令与指定命令所有的子命令。
例如在上例中加入一行代码

rootCmd.PersistentFlags().BoolP("help", "h", false, "Print usage")

局部标示（Local Flags）

局部标示（Local Flags）仅作用于其指定命令。
例如docker image inspect命令的源码中定义了局部标示-fhttps://github.com/docker/cli/blob/v19.03.5/cli/command/image/inspect.go

flags := cmd.Flags()
	flags.StringVarP(&opts.format, "format", "f", "", "Format the output using the given Go template")

注：Docker自定义了命令行模版，所以Flags在docker命令中显示的是Options
https://github.com/docker/cli/blob/v19.03.5/cli/cobra.go#L304

Demo

参照Docker源码，实现类似docker ps的功能：返回运行中的容器ID

package main

import (
	"fmt"
	"context"
	"github.com/docker/docker/api/types"
	"github.com/docker/docker/client" // v1.13.1
	"github.com/spf13/cobra"
)


func runPs(dockerCli *client.Client) {
	containers, err := dockerCli.ContainerList(context.Background(), types.ContainerListOptions{})
	if err != nil {
		panic(err)
	}

	fmt.Println("运行中的容器ID:")
	for _, container := range containers {
		fmt.Printf("%s \n", container.ID[:10])
	}

}


func main() {
	dockerCli, _ := client.NewEnvClient()

	// 定义ps命令
	var cmdPs = &cobra.Command{
		Use:              "ps",
		Short:            "List containers",
		Run: func(cmd *cobra.Command, args []string)  {
			runPs(dockerCli)
		},
	}
	// 定义根命令
	var rootCmd = &cobra.Command{Use: "play_docker"}
	// 加入ps命令
	rootCmd.AddCommand(cmdPs)
	// 初始化cobra
	rootCmd.Execute()
}

执行我们的山寨命令行./play_docker ps看到返回的结果和docker ps是一致的