k8s学习之cobra命令库学习

发布于:2024-07-07 ⋅ 阅读:(35) ⋅ 点赞:(0)

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 。

5 总结

本篇文章主要是一步步的吧cobra这个库如何使用等进行了详细的解释,我们在使用这个库的时候,比较能够理解他工作的原理,然后再根据具体的k8s如何使用以及具体字段的使用来充分发挥cmd的能力和价值


网站公告

今日签到

点亮在社区的每一天
去签到