ARTICLE DETAIL

资讯详情

深耕网站视觉设计与运营推广的一线实战洞察。

Cobra 如何自定义 --version 标志的输出?(Version 字段与 SetVersionTemplate)

Cobra 如何自定义 --version 标志的输出?(Version 字段与 SetVersionTemplate) Cobra 如何自定义 --version 标志的输出Version 字段与 SetVersionTemplate【免费下载链接】cobraA Commander for modern Go CLI interactions项目地址: https://gitcode.com/GitHub_Trending/co/cobra用 Cobra 写 Go CLI 程序时--version标志的输出格式经常要按团队规范调整比如加上产品名前缀、去掉多余的空格。Cobra 对这个行为提供了两个直接的开关在命令上设置Version字段来启用版本标志再用SetVersionTemplate方法替换打印输出所用的模板。本文以仓库内的用户指南和源码为准说明如何完成这条定制路径以及如何验证输出确实变成了你要的格式。前提Version 字段是启用 --version 的唯一开关用户指南的 Version Flag 一节说明了触发条件Cobra adds a top-level --version flag if the Version field is set on the root command.对应的行为定义在 command.go 中。Command结构体的Version字段注释写明只有当该值非空、且命令没有自己定义 version 标志时Cobra 才会自动添加一个布尔类型的version标志如果命令也没有定义-v短标志会一并加上。执行逻辑见 command.goexecute()里先通过InitDefaultVersionFlag()初始化默认版本标志随后只有c.Version ! 且命令行传入了 version 标志时才会取出模板函数并把结果写到 stdout。也就是说Version为空字符串时--version行为根本不会生效此时SetVersionTemplate也没有可作用的对象如果你的代码已经手动声明了一个非布尔类型的 version 标志程序会打印version flag declared as non-bool. Please correct your code这是文档源码中明确给出的报错文本可用于判断标志类型声明错了。第一步设置 Version 字段确认默认输出安装库的命令见 READMEgo get -u github.com/spf13/cobralatest最小可运行示例如下Use与Version的取值取自仓库测试用例 command_test.go 中的写法package main import ( fmt github.com/spf13/cobra ) func main() { rootCmd : cobra.Command{ Use: root, Version: 1.0.0, Run: func(cmd *cobra.Command, args []string) { fmt.Println(root command run) }, } rootCmd.Execute() }运行go run main.go --version此时不设置任何模板走的是默认模板。command.go 中定义了默认模板const defaultVersionTemplate {{with .DisplayName}}{{printf %s .}}{{end}}{{printf version %s .Version}} 测试用例TestShorthandVersionFlagExecuted验证了这种情况下--version以及短标志-v的输出为root version 1.0.0——这是仓库测试中给出的示例结果可作为默认格式的核对基准。如果你不需要改格式这一步结束即可要定制输出则继续。第二步用 SetVersionTemplate 替换输出模板SetVersionTemplate的声明见 command.go// SetVersionTemplate sets version template to be used. Application can use it to set custom template. func (c *Command) SetVersionTemplate(s string) { ... }模板参数就是命令对象本身模板中可以访问{{.Version}}、{{.DisplayName}}等字段。把上一步的示例补上一行即可rootCmd.SetVersionTemplate(customized version: {{.Version}})这一行写法直接来自测试用例TestVersionTemplatecommand_test.go。该测试用--version执行命令后断言输出包含customized version: 1.0.0TestShorthandVersionTemplate进一步确认短标志-v走的是同一套模板。验证方式go run main.go --version go run main.go -v两条命令都应输出customized version: 1.0.0文档测试中的示例输出。如果输出仍是root version 1.0.0说明SetVersionTemplate没有被执行到——检查它是否写在Execute()之前以及是否设置在了承载 version 标志的那个命令上。边界与限制以下限制来自源码注释和用户指南直接影响这条定制路径模板引擎的代价用户指南 明确提示SetVersionTemplate指定的模板使用 Go 的text/template求值这会增大编译后可执行文件的体积。如果版本输出只是简单换格式需要权衡是否值得引入模板。子命令继承VersionTemplate()与getVersionTemplateFunc()command.go的实现显示子命令没有自己的模板时会沿父命令向上查找模板函数根命令上设置的模板对整个命令树生效。只覆盖没有自定义 version 标志的命令InitDefaultVersionFlagcommand.go在命令已经存在名为 version 的标志时不做任何事此时模板不会被 Cobra 的默认版本流程使用。Version 字段必须非空Version为空时默认版本标志不会被初始化模板也无从求值。参考用户指南Version Flag 一节——触发条件与SetVersionTemplate的官方说明command.go——Version字段、默认模板与执行流程command_test.go——--version、-v与自定义模板的测试断言可作为输出核对基准【免费下载链接】cobraA Commander for modern Go CLI interactions项目地址: https://gitcode.com/GitHub_Trending/co/cobra创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表