使用Cobra与Client-go实现Kubernetes自定义插件开发

在我们使用kubectl查看Kubernetes资源的时候，想直接查看对应资源的容器名称和镜像名称，目前kubectl还不支持该选型，需要我们describe然后来查看，对于集群自己比较多，不是很方便，因此萌生了自己开发kubectl 插件来实现该功能。

Cobra

Cobra是一个命令行程序库，是一个用来编写命令行的神器，提供了一个脚手架，用于快速生成基于Cobra应用程序框架。我们可以利用Cobra快速的去开发出我们想要的命令行工具，非常的方便快捷。

详细可参考：https://juejin.cn/post/6983299467537547294

Client-go

在Kubernetes运维中，我们可以使用kubectl、客户端库或者REST请求来访问Kubernetes API。而实际上，无论是kubectl还是客户端库，都是封装了REST请求的工具。client-go作为一个客户端库，能够调用Kubernetes API，实现对Kubernetes集群中资源对象（包括Deployment、Service、Ingress、ReplicaSet、Pod、Namespace、Node等）的增删改查等操作。

详细可参考：https://juejin.cn/post/6962869412785487909

Kubernetes插件Krew

Krew是类似于系统的apt、dnf或者brew的kubectl插件包管理工具，利用其可以轻松的完成kubectl插件的全面周期管理，包括搜索、下载、卸载等。

kubectl其工具已经比较完善，但是对于一些个性化的命令，其宗旨是希望开发者能以独立而紧张形式发布自定义的kubectl子命令，插件的开发语言不限，需要将最终的脚步或二进制可执行程序以kubectl-的前缀命名，然后放到PATH中即可，可以使用kubectl plugin list查看目前已经安装的插件。

详细可参考：https://juejin.cn/post/6969183421381738533

插件命名为：kubeimg
目前仅简单实现一个image命令，用于查看不同资源对象（Deployments/DaemonSets/StatefulSets/Jobs/Cronjobs）的名称，和对应容器名称，镜像名称。
支持json格式输出。
最后将其作为krew插件使用。
可以直接根据名称空间来进行查看对应资源。

项目初始化

安装Cobra：

go get -v github.com/spf13/cobra/cobra

初始化项目：

$ ~/workspace/goworkspace/src/github.com/kaliarch/kubeimg  /Users/xuel/workspace/goworkspace/bin/cobra init --pkg-name kubeimg

Your Cobra application is ready at

/Users/xuel/workspace/goworkspace/src/github.com/kaliarch/kubeimg

$ ~/workspace/goworkspace/src/github.com/kaliarch/kubeimg  ls

LICENSE cmd     main.go

$ ~/workspace/goworkspace/src/github.com/kaliarch/kubeimg  tree

├── LICENSE

├── cmd

│   └── root.go

└── main.go

1 directory, 3 files

创建go mod，下载相关包：

go mod init kubeimg

增加子命令

增加一个子命令image：

$ /Users/xuel/workspace/goworkspace/bin/cobra add image

image created at /Users/xuel/workspace/goworkspace/src/github.com/kaliarch/kubeimg

// 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() {

cobra.CheckErr(rootCmd.Execute())

}

func init() {

KubernetesConfigFlags = genericclioptions.NewConfigFlags(true)

imageCmd.Flags().BoolP("deployments", "d", false, "show deployments image")

imageCmd.Flags().BoolP("daemonsets", "e", false, "show daemonsets image")

imageCmd.Flags().BoolP("statefulsets", "f", false, "show statefulsets image")

imageCmd.Flags().BoolP("jobs", "o", false, "show jobs image")

imageCmd.Flags().BoolP("cronjobs", "b", false, "show cronjobs image")

imageCmd.Flags().BoolP("json", "j", false, "show json format")

KubernetesConfigFlags.AddFlags(rootCmd.PersistentFlags())

}

实现image命令

var imageCmd = &cobra.Command{

Use:   "image",

Short: "show resource image",

Long:  `show k8s resource image`,

RunE:  image,

}

func init() {

rootCmd.AddCommand(imageCmd)

}

初始化Clientset

// ClientSet k8s clientset

func ClientSet(configFlags *genericclioptions.ConfigFlags) *kubernetes.Clientset {

config, err := configFlags.ToRESTConfig()

if err != nil {

    panic("kube config load error")

}

clientSet, err := kubernetes.NewForConfig(config)

if err != nil {

panic("gen kube config error")

}

return clientSet

}

实现查看资源对象

利用反射实现根据不同资源类型查看具体对应资源镜像及镜像名称功能。

func image(cmd *cobra.Command, args []string) error {

clientSet := kube.ClientSet(KubernetesConfigFlags)

ns, _ := rootCmd.Flags().GetString("namespace")

// 生命一个全局资源列表

var rList []interface{}

if flag, _ := cmd.Flags().GetBool("deployments"); flag {

    deployList, err := clientSet.AppsV1().Deployments(ns).List(context.Background(), v1.ListOptions{})

    if err != nil {

        fmt.Printf("list deployments error: %s", err.Error())

    }

    rList = append(rList, deployList)

}

...

deployMapList := make([]map[string]string, 0)

for i := 0; i < len(rList); i++ {

    switch t := rList[i].(type) {

    case *kv1.DeploymentList:

        for k := 0; k < len(t.Items); k++ {

            for j := 0; j < len(t.Items[k].Spec.Template.Spec.Containers); j++ {

                deployMap := make(map[string]string)

                deployMap["NAMESPACE"] = ns

                deployMap["TYPE"] = "deployment"

                deployMap["RESOURCE_NAME"] = t.Items[k].GetName()

                deployMap["CONTAINER_NAME"] = t.Items[k].Spec.Template.Spec.Containers[j].Name

                deployMap["IMAGE"] = t.Items[k].Spec.Template.Spec.Containers[j].Image

                deployMapList = append(deployMapList, deployMap)

            }

        }

利用tabel来对结果进行输出：

func GenTable(mapList []map[string]string) *table.Table {

t, err := gotable.Create(title...)

if err != nil {

    fmt.Printf("create table error: %s", err.Error())

    return nil

}

t.AddRows(mapList)

return t

}

最终项目结构：

对完成的插件进行测试，编译go build生成kubeimg二进制可执行文件。
查看所有帮助，其中可以看到cobra帮我们自动生成了help和completion两个命令，可以快速实现table补全，支持bash/fish/zsh/powershell：

./kubeimg --help

查看image命令flags：

./kubeimg image --help

查看Deployment资源

不知道ing名称名称空间，默认查看所有，名称空间下的资源：

./kubeimg image -d

查看某个名称空间下资源

./kubeimg image -d -n kube-system

查看所有资源

可以看到imlc-operator-controller-manager一个Pod中有两个container：

./kubeimg image -b -e -d -o -f

json格式输出

./kubeimg image -o -j

作为krew插件使用

需要将最终的脚步或二进制可执行程序以kubectl-的前缀命名，然后放到PATH中即可，可以使用kubectl plugin list查看目前已经安装的插件。

$ kubectl plugin list

The following compatible plugins are available:=

/usr/local/bin/kubectl-debug

- warning: kubectl-debug overwrites existing command: "kubectl debug"

/usr/local/bin/kubectl-v1.10.11

/usr/local/bin/kubectl-v1.20.0

/Users/xuel/.krew/bin/kubectl-df_pv

/Users/xuel/.krew/bin/kubectl-krew

# 将自己开发的插件重新命名为kubectl-img放到可执行路基下

$ cp kubeimg /Users/xuel/.krew/bin/kubectl-img

$ kubectl plugin list

The following compatible plugins are available:=

/usr/local/bin/kubectl-debug

- warning: kubectl-debug overwrites existing command: "kubectl debug"

/usr/local/bin/kubectl-v1.10.11

/usr/local/bin/kubectl-v1.20.0

/Users/xuel/.krew/bin/kubectl-df_pv

/Users/xuel/.krew/bin/kubectl-krew

/Users/xuel/.krew/bin/kubectl-img

$ cp kubeimg /Users/xuel/.krew/bin/kubectl-img

之后就可以想使用kubectl插件一样使用了。

目前实现的比较简单，以此来抛砖引玉的功能，后期可以进行更多功能或其他插件的开发，自己动手丰衣足食。

后期待再完善开源到GitHub，个人主页：https://github.com/redhatxl，以供大家学习交流。

原文链接：https://juejin.cn/post/6983324056502140964，作者：kaliarch

使用Cobra与Client-go实现Kubernetes自定义插件开发

使用Cobra与Client-go实现Kubernetes自定义插件开发

Cobra

Client-go

Kubernetes插件Krew

项目初始化

增加子命令

实现image命令

初始化Clientset

实现查看资源对象

查看Deployment资源

查看某个名称空间下资源

查看所有资源

json格式输出

作为krew插件使用

Recommend

Load testing for engineering teams | k6

肥多多618狂欢节：直播、竞拍、拼团年中钜惠盛宴圆满结束

必要商城全面出击，C2M模式成为“智”造业重要推手

Facebook Clashes with the US Government Over Vaccine Misinformation

精准营销与降本增效成绩斐然丽人丽妆龙头优势愈加凸显

C2M模式成电商平台竞争新赛道，必要商城打开亿万市场前景

SaaS vs. On-Prem Vs. Hybrid Data Infrastructure Model

如何用Loki来分析Kubernetes事件

看上讯信息ADM如何玩转数据脱敏？

One Secret to Improve SEO in 2021: Guestographics

About Joyk