最开始的解析命令行参数是使用的标准库里面的flag包,后来想增加新的参数的时候比较复杂和困难,因此使用cobra更加简单一些
比如执行go-fly server port 8081是运行项目
执行go-fly install是导入数据库
目录结构:
增加cmd目录,作为cmd包,代码中直接定义全局变量和可导出的函数 , root.go里面是这样的
package cmd import ( "errors" fmtgithub.com/spf13/cobraos" ) var rootCmd = &cobra.Command{ Use: go-fly,Short: 快速的GO语言WEB在线客服 https://gofly.sopans.com`, Args:args,Run: func(cmd *cobra.Command,args []string) { },} func args(cmd *cobra.Command,1)">) error{ if len(args)<1{ return errors.New(至少需要一个参数!) } return nil } func Execute() { if err := rootCmd.Execute(); err != nil { fmt.Println(err) os.Exit() } } func init() { rootCmd.AddCommand(versionCmd) rootCmd.AddCommand(serverCmd) rootCmd.AddCommand(installCmd) }
package main import ( github.com/taoshihan1991/imaptool/cmd ) func main() { cmd.Execute() }
AddCommand就是在增加参数,比如versionCmd这个实例,定义在了version.go里面,执行go-fly version的时候打印版本号
package cmd import ( github.com/taoshihan1991/imaptool/config ) var versionCmd = &versionexample:go-fly version) { fmt.Println(go-fly "+config.Version) },}
直接访问效果: