k8s学习之cobra命令库学习

关注

原创 2024-07-07 20:46:09 1.6k 阅读 AI 写同款· GEO 优化 ›

从k8s当中学习go cli脚手架开发利器-cobra 在Run()在设定了一系列的参数信息后，创建了cmd对象，并执行cmd对象的Execute()，这里的cmd对象就是一个cobra命令对象，而Execute是cobra提供执行命令的方法，cobra内部使用pflag库，通过设置 pflag 属性，可以对 cobra 的运行产生作用。cobra对象如何生成的，是我们需要关心的，**NewKubeadmCommand(os.Stdin, os.Stdout, os.Stderr)**的实现在cmd/kubeadm/app/cmd/cmd.go。阅读详情

1.前言

打开k8s代码的时候，我发现基本上那几个核心服务都是使用cobra库作为命令行处理的能力。因此，为了对代码之后的代码学习的有比较深入的理解，因此先基于这个库写个demo，加深对这个库的一些理解吧

2.cobra库的基本简介

Github：GitHub - spf13/cobra: A Commander for modern Go CLI interactions 在这里了，Cobra是一个用Go语言实现的命令行工具。并且现在正在被很多项目使用，例如：Kubernetes、Hugo和Github CLI等。通过使用Cobra，我们可以快速的创建命令行工具，特别适合写 测试脚本，各种服务的 Admin CLI等。

因为命令行参数，golang本身自带了"flag" 库，使用flag库呢

flag.Parse()  

args := flag.Args()

解析完之后，需要每个命令去判断，十分的麻烦。

但是我们在使用到时候，需要针对命令行进行各种解析判断，但是有了这个库，我们就可以减少这段冗余的代码了

3 cobra的使用demo

话不多说，我们直接给出怎么用这个项目。我的环境是在macos上。

3.1 首先下载依赖库

命令行执行

go get -u github.com/spf13/cobra@latest

3.2 命令行CLI工具

$ go install github.com/spf13/cobra-cli@latest

因为我本身我的golang代码路径已经建好了bin目录，因此，执行完该命令后,如下

我们看到了cobra-cli的可执行文件。

这边是我们可以在命令用这个东西，进行初始化了。集体步骤如下

xxx@MBP src % mkdir greet
xxx@MBP src % cd greet
xxx@MBP test % ls
xxx@MBP greet % cobra-cli init
Error: Please run `go mod init <MODNAME>` before `cobra-cli init`
xxx@MBP greet % go mod init greet
go: creating new go.mod: module greet
xxx@MBP greet % ls
go.mod
xxx@MBP greet % cobra-cli init
Your Cobra application is ready at
/Users/XXX/workspace/golang/src/greet
xxx@MBP greet % ls
LICENSE cmd go.mod  go.sum  main.go
xxx@MBP greet %

第一步先建好一个test目录，然后执行go mod init test,然后再执行cobra-cli init的命令行，就会生成了脚手架文件了。

目录中会产生一下文件：

main.go cmd LICENSE go.mod go.sum
├── LICENSE
├── cmd
│   └── root.go
└── main.go
    go.mod 
    go.sum

3.3 编写代码文件

刚刚生成的文件，因为没有加具体的命令选项，可能用处不大，因此我们继续执行，添加一个命令字

cobra-cli add [command]

这样，他就会又生成了一个文件。

比如执行

cobra-cli add greet

// Package cmd /*
package cmd

import (
  "fmt"

  "github.com/spf13/cobra"
)

// greetCmd represents the greet command
var greetCmd = &cobra.Command{
  Use:   "greet",
  Short: "A brief description of your command",
  Long: `A longer description that spans multiple lines and likely contains examples
and usage of using your command. For example:

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.`,
  Run: func(cmd *cobra.Command, args []string) {
    // 这段代码是我加的。
    if len(args) < 1 {
      cmd.Help()
      return
    }
    name := args[0]
    fmt.Println("greet called:", name)
  },
}

func init() {
  rootCmd.AddCommand(greetCmd)

  // Here you will define your flags and configuration settings.

  // Cobra supports Persistent Flags which will work for this command
  // and all subcommands, e.g.:
  // greetCmd.PersistentFlags().String("foo", "", "A help for foo")

  // Cobra supports local flags which will only run when this command
  // is called directly, e.g.:
  greetCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}

因此，我们可以看到，新增了greet的命令。

然后我们编译一下，生成目标文件，后我们执行以下

./greet

A longer description that spans multiple lines and likely contains
examples and usage of using your application. For example:

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:
  greet [command]

Available Commands:
  completion  Generate the autocompletion script for the specified shell
  greet       A brief description of your command
  help        Help about any command
  mockMsg     A brief description of your command

Flags:
  -h, --help     help for greet
  -t, --toggle   Help message for toggle

Use "greet [command] --help" for more information about a command.

以上我们通过返回的内容，可以看出多了一个gree的子命令。

并且可以加flag，比如-h

上边的Run 就可以放我们想要处理的业务逻辑了。

如果我们想增加一些flag 值，只需要能够在run里边添加如下代码即可

g, _ := cmd.Flags().GetInt32("goroutine")
 p, _ := cmd.Flags().GetInt32("packet")
 fmt.Println("mockmsg called,flags:g=", g, ",p=", p, ",args:", args)

比如我们实现如下：

Run: func(cmd *cobra.Command, args []string) {
    if len(args) < 1 {
      cmd.Help()
      return
    }
    name := args[0]
    fmt.Println("greet called:", name)

    g, _ := cmd.Flags().GetInt32("goroutine")
    p, _ := cmd.Flags().GetInt32("packet")
    fmt.Println("mockmsg called,flags:g=", g, ",p=", p, ",args:", args)
  },

这样，问可以这样使用了

./greet greet xx -p 10 -g 100
greet called: xx
greetCmd called,flags:g= 100 ,p= 10 ,args: [xx]

4 K8s中如何使用

我主要是看了kubctl模块的代码，我发现这个模块对cmd这块又进行了更丰富的封装，用以支持更加复杂的场景。

其中我们从kubectl的main开始

一个入口方法

// NewDefaultKubectlCommand creates the `kubectl` command with default arguments
func NewDefaultKubectlCommand() *cobra.Command {
  ioStreams := genericiooptions.IOStreams{In: os.Stdin, Out: os.Stdout, ErrOut: os.Stderr}
  return NewDefaultKubectlCommandWithArgs(KubectlOptions{
    PluginHandler: NewDefaultPluginHandler(plugin.ValidPluginFilenamePrefixes),
    Arguments:     os.Args,
    ConfigFlags:   defaultConfigFlags().WithWarningPrinter(ioStreams),
    IOStreams:     ioStreams,
  })
}

因为kubectl是可以对文件，以及flag进行处理使用的，因此，需要针对这种情况各种设置参数，我们发现已经充分利用了cmd结构体的内各个字段

其中这两个方法是核心方法，基本上是初始化并赋值给command的结构了

// NewDefaultKubectlCommandWithArgs creates the `kubectl` command with arguments
func NewDefaultKubectlCommandWithArgs(o KubectlOptions) *cobra.Command {
  cmd := NewKubectlCommand(o)

其中在NewKubectlCommand 就是用来new cmd实例的。

func NewKubectlCommand(o KubectlOptions) *cobra.Command {
  warningHandler := rest.NewWarningWriter(o.IOStreams.ErrOut, rest.WarningWriterOptions{Deduplicate: true, Color: term.AllowsColorOutput(o.IOStreams.ErrOut)})
  warningsAsErrors := false
  // Parent command to which all subcommands are added.
  cmds := &cobra.Command{
    Use:   "kubectl",

这个就是我么看到的kubectl的命令的使用地方了，因为本文主要是讲command的使用，因此k8s的其他get命令等先不赘述，下一篇在讲

下面是具体贴一下Command具体的相关的字段了意思了

以下是对 `Command` 结构体中每个字段的详细解释：

1. `Use`：这是一个字符串，用于描述命令的基本使用方式。它规定了命令所需的参数和可选参数的格式，为用户提供了简洁明了的使用指导。
    - 例如：`"add [-F file | -D dir]... [-f format] profile"` 表示 `add` 命令可以有可选的 `-F` 和 `-D` 参数，其中 `-F` 后跟一个文件，`-D` 后跟一个目录，这两个参数是互斥的，并且可以多次出现，还有一个可选的 `-f` 参数后跟格式信息，最后需要一个 `profile` 参数。

2. `Aliases`：一个字符串切片，存储了该命令的别名。这使得用户可以通过不同的名称来调用同一个命令，增加了命令使用的灵活性。
    - 比如：`["add_item", "insert"]` 是 `add` 命令的别名。

3. `SuggestFor`：也是一个字符串切片，其中包含了此命令可能被建议替代的其他命令名称。这有助于在用户输入类似但不完全准确的命令时提供相关的建议。

4. `Short`：一个简短的字符串，用于在帮助输出中提供命令的简短描述，让用户快速了解命令的主要功能。
    - 例如："添加新的项目"

5. `GroupID`：指定该子命令在其父命令的“帮助”输出中所属的组标识，便于对命令进行分组展示和管理。

6. `Long`：详细的长字符串，在“帮助 <此命令>”输出中提供更全面和深入的命令描述，包括更多的功能细节、使用示例、注意事项等。

7. `Example`：包含命令使用的示例字符串，通过实际的例子帮助用户更好地理解如何正确使用该命令。

8. `ValidArgs`：一个字符串切片，列出了在外壳自动补全中所有有效的非标志参数。

9. `ValidArgsFunction`：一个函数，动态地提供有效的非标志参数用于外壳自动补全，是一种更灵活的参数提供方式。

10. `Args`：定义了预期的参数的相关规则和处理方式。

11. `ArgAliases`：一个字符串切片，列出了有效参数的别名，这些别名不会在外壳自动补全中被提示，但手动输入时会被接受。

12. `BashCompletionFunction`：用于传统 Bash 自动补全生成器的自定义 Bash 函数，为特定的 Bash 环境提供定制的自动补全功能。

13. `Deprecated`：如果命令已被弃用，存储了使用该命令时将显示的提示信息，告知用户该命令不应再被使用。

14. `Annotations`：一个键值对映射，允许应用程序为命令添加自定义的标识、分组或特殊选项等元数据。

15. `Version`：存储命令的版本信息，用于版本控制和显示。

16. 各种 `Run` 函数：
    - `PersistentPreRun` 和 `PersistentPreRunE`：在命令执行前被调用，且子命令会继承并执行，用于进行一些持久的预处理操作。
    - `PreRun` 和 `PreRunE`：在命令执行前被调用，但子命令不会继承，用于特定于当前命令的预处理。
    - `Run` 和 `RunE`：实际的工作函数，实现命令的主要逻辑。
    - `PostRun` 和 `PostRunE`：在 `Run` 函数执行后被调用，用于进行后续的处理操作。

17. `commandgroups`：一个指针切片，指向子命令所属的组对象。

18. `args`：实际从标志解析得到的参数切片。

19. `flagErrorBuf`：一个字节缓冲区，用于存储来自 `pflag` 的错误消息。

20. `flags`：一个 `flag.FlagSet` 对象，包含了所有的标志。

21. `pflags`：存储持久的标志。

22. `lflags`：本地标志的缓存，用于优化 `LocalFlags` 函数调用。

23. `iflags`：继承的标志的缓存，用于优化相关函数调用。

24. `parentsPflags`：父命令的所有持久标志。

25. `globNormFunc`：一个全局的标准化函数，用于处理标志名称的标准化。

26. `usageFunc`：用户定义的使用函数，用于自定义命令的使用说明。

27. `usageTemplate`：用户定义的使用模板。

28. `flagErrorFunc`：用户定义的标志错误处理函数。

29. `helpTemplate`：用户定义的帮助模板。

30. `helpFunc`：用户定义的帮助函数。

31. `helpCommand`：具有“帮助”用途的命令，如果用户未定义，则使用默认的帮助命令。

32. `helpCommandGroupID`：帮助命令所属的组标识。

33. `completionCommandGroupID`：自动完成命令所属的组标识。

34. `versionTemplate`：用户定义的版本模板。

35. `errPrefix`：用户定义的错误消息前缀。

36. `inReader`：用户定义的输入读取器，替代标准输入。

37. `outWriter`：用户定义的输出写入器，替代标准输出。

38. `errWriter`：用户定义的错误输出写入器，替代标准错误输出。

39. `FParseErrWhitelist`：要忽略的标志解析错误的列表。

40. `CompletionOptions`：用于控制外壳自动完成的选项。

41. `commandsAreSorted`：一个布尔值，指示命令切片是否已排序。

42. `commandCalledAs`：一个结构体，记录命令被调用时的名称和是否被调用的状态。

43. `ctx`：上下文对象，用于传递和管理命令执行的上下文信息。

44. `commands`：一个指针切片，包含此命令支持的子命令。

45. `parent`：指向该命令的父命令的指针。

46. `Max lengths` 相关字段：记录命令相关字符串的最大长度，用于格式化和展示。

47. `TraverseChildren`：一个布尔值，决定是否在执行子命令之前解析所有父命令的标志。

48. `Hidden`：如果为真，该命令将在可用命令列表中隐藏，不向用户显示。

49. `SilenceErrors`：一个布尔值，用于控制是否静默处理错误。

50. `SilenceUsage`：一个布尔值，用于控制在发生错误时是否静默使用信息。

51. `DisableFlagParsing`：如果为真，将禁用标志解析，所有标志将作为参数传递给命令。

52. `DisableAutoGenTag`：如果为真，在生成文档时将禁用自动生成的标签。

53. `DisableFlagsInUseLine`：如果为真，将在使用行中禁用标志的添加。

54. `DisableSuggestions`：如果为真，将禁用基于编辑距离的建议功能。

55. `SuggestionsMinimumDistance`：定义显示建议的最小编辑距离，必须大于 0 。