kubectl源码分析之completion

发布一个k8s部署视频：https://edu.csdn.net/course/detail/26967

课程内容：各种k8s部署方式。包括minikube部署，kubeadm部署，kubeasz部署，rancher部署，k3s部署。包括开发测试环境部署k8s，和生产环境部署k8s。

腾讯课堂连接地址https://ke.qq.com/course/478827?taid=4373109931462251&tuin=ba64518

第二个视频发布 https://edu.csdn.net/course/detail/27109

腾讯课堂连接地址https://ke.qq.com/course/484107?tuin=ba64518

介绍主要的k8s资源的使用配置和命令。包括configmap，pod，service，replicaset，namespace，deployment，daemonset，ingress，pv，pvc，sc，role，rolebinding，clusterrole，clusterrolebinding，secret，serviceaccount，statefulset，job，cronjob，podDisruptionbudget，podSecurityPolicy,networkPolicy,resourceQuota，limitrange，endpoint，event，conponentstatus，node，apiservice，controllerRevision等。

第三个视频发布：https://edu.csdn.net/course/detail/27574

详细介绍helm命令，学习helm chart语法，编写helm chart。深入分析各项目源码，学习编写helm插件

第四个课程发布：https://edu.csdn.net/course/detail/28488

本课程将详细介绍k8s所有命令，以及命令的go源码分析，学习知其然，知其所以然
————————————————

var (//支持的shell及函数
	completionShells = map[string]func(out io.Writer, boilerPlate string, cmd *cobra.Command) error{
		"bash": runCompletionBash,
		"zsh":  runCompletionZsh,
	}
)

//创建completion命令
func NewCmdCompletion(out io.Writer, boilerPlate string) *cobra.Command {
	shells := []string{}
	for s := range completionShells {
		shells = append(shells, s)
	}

	cmd := &cobra.Command{//创建cobra命令
		Use:                   "completion SHELL",
		DisableFlagsInUseLine: true,
		Short:                 i18n.T("Output shell completion code for the specified shell (bash or zsh)"),
		Long:                  completionLong,
		Example:               completionExample,
		Run: func(cmd *cobra.Command, args []string) {
			err := RunCompletion(out, boilerPlate, cmd, args)//运行
			cmdutil.CheckErr(err)
		},
		ValidArgs: shells,//有效参数
	}

	return cmd
}

//运行
func RunCompletion(out io.Writer, boilerPlate string, cmd *cobra.Command, args []string) error {
	if len(args) == 0 {//参数不能是0个
		return cmdutil.UsageErrorf(cmd, "Shell not specified.")
	}
	if len(args) > 1 {//参数不能大于1个
		return cmdutil.UsageErrorf(cmd, "Too many arguments. Expected only the shell type.")
	}
	run, found := completionShells[args[0]]//获取运行函数
	if !found {
		return cmdutil.UsageErrorf(cmd, "Unsupported shell type %q.", args[0])
	}

	return run(out, boilerPlate, cmd.Parent())//运行
}

//运行bash completion
func runCompletionBash(out io.Writer, boilerPlate string, kubectl *cobra.Command) error {
	if len(boilerPlate) == 0 {//设置license
		boilerPlate = defaultBoilerPlate
	}
	if _, err := out.Write([]byte(boilerPlate)); err != nil {//输出license
		return err
	}

	return kubectl.GenBashCompletion(out)//产生命令补全
}

//产生命令补全
func (c *Command) GenBashCompletion(w io.Writer) error {
	buf := new(bytes.Buffer)//创建buffer
	writePreamble(buf, c.Name())//写序言
	if len(c.BashCompletionFunction) > 0 {//如果有BashCompletionFunction函数，则写函数
		buf.WriteString(c.BashCompletionFunction + "\n")
	}
	gen(buf, c)//写具体命令
	writePostscript(buf, c.Name())//写post脚本

	_, err := buf.WriteTo(w)//buffer写到writer里
	return err
}

//递归调用写命令completion
func gen(buf *bytes.Buffer, cmd *Command) {
	for _, c := range cmd.Commands() {//遍历命令
		if !c.IsAvailableCommand() || c == cmd.helpCommand {
			continue
		}
		gen(buf, c)//递归调用
	}
	commandName := cmd.CommandPath()//获取命令名称
	commandName = strings.Replace(commandName, " ", "_", -1)
	commandName = strings.Replace(commandName, ":", "__", -1)

	if cmd.Root() == cmd {//命令是root命令
		buf.WriteString(fmt.Sprintf("_%s_root_command()\n{\n", commandName))
	} else {
		buf.WriteString(fmt.Sprintf("_%s()\n{\n", commandName))
	}

	buf.WriteString(fmt.Sprintf("    last_command=%q\n", commandName))
	buf.WriteString("\n")
	buf.WriteString("    command_aliases=()\n")
	buf.WriteString("\n")

	writeCommands(buf, cmd)//写命令
	writeFlags(buf, cmd)//写flags
	writeRequiredFlag(buf, cmd)//写reuqiredflag
	writeRequiredNouns(buf, cmd)//写requiredNouns
	writeArgAliases(buf, cmd)//写别名
	buf.WriteString("}\n\n")
}

来源：oschina

链接：https://my.oschina.net/u/4260177/blog/4326349

标签

Helm