k8s学习之cobra命令库学习

news/2024/10/9 15:15:07/

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.gogo.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的命令。

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

./greetA 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 shellgreet       A brief description of your commandhelp        Help about any commandmockMsg     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的能力和价值


http://www.ppmy.cn/news/1474369.html

相关文章

el-dropdown的command方法添加自定义参数

代码 <div v-for"(item, index) in queryParams.changeParams" :key"index"><el-form-item prop"dataConditionSearch"><el-dropdown command"handleCommand" style"margin-right: 3px;"><span class…

宪法学学习笔记(个人向) Part.3

宪法学学习笔记(个人向) Part 3 3. 国家基本制度 3.1 国家性质 3.1.1 国家性质概述 国家性质的概念 国家性质也称国体&#xff0c;或国家的阶级本质&#xff0c;是指各个阶级在国家中的地位&#xff08;哪个阶层是统治阶层&#xff0c;哪个阶层是被统治阶层&#xff0c;哪个…

Java中的类与对象

一、类 定义&#xff1a;对具有相同特征的这一类事物所进行的抽象描述 比如&#xff1a;人类 二、对象&#xff08;实例&#xff09; 定义&#xff1a;实际存在的该类事物的具体个体 比如&#xff1a;张三 三、定义类 编程是对现实生活中事物的体现&#xff0c;是对现实…

关于数组的常见算法

一、案例一 案例说明 案例&#xff1a;定义一个int型的一维数组&#xff0c;包含10个元素&#xff0c;分别赋一些随机整数&#xff0c;然后求出所有元素的最大值&#xff0c;最小值&#xff0c;总和&#xff0c;平均值&#xff0c;并输出出来 要求&#xff1a;所有随机数都是两…

element el-table实现表格动态增加/删除/编辑表格行,带校验规则

本篇文章记录el-table增加一行可编辑的数据列&#xff0c;进行增删改。 1.增加空白行 直接在页面mounted时对form里面的table列表增加一行数据&#xff0c;直接使用push() 方法增加一列数据这个时候也可以设置一些默认值。比如案例里面的 产品件数 。 mounted() {this.$nextTi…

通勤数据:Comma2k19 数据集

A Commute in Data: The comma2k19 Dataset 通勤数据&#xff1a;Comma2k19 数据集 https://arxiv.org/pdf/1812.05752v1 Abstract— comma.ai presents comma2k19, a dataset of over 33 hours of commute in California’s 280 highway. This means 2019 segments, 1 minut…

性能测试的流程(企业真实流程详解)(二)

性能测试的流程 1.需求分析以及需求确定(指标值&#xff0c;场景&#xff0c;环境&#xff0c;人员) 一般提出需求的人员有&#xff1a;客户&#xff0c;产品经理&#xff0c;项目组领导等 2.性能测试计划和方案制定 基准测试: 负觋测试: 压力测试: 稳定性测试: 其他:配置测试…

Codeforces Round #956 (Div. 2) and ByteRace 2024

A. Array Divisibility Given n&#x1d45b;, find an array of positive nonzero integers, with each element less than or equal to 105 that is beautiful subject to all 1≤&#x1d458;≤&#x1d45b;. It can be shown that an answer always exists. #include<…