一、main.go
// main.go
package main
import (
"fmt"
"projectdemo/cmd"
)
func main() {
if err := cmd.RootCmd.Execute(); err != nil {
fmt.Println(err)
}
}
二、cmd/root.go
// cmd/root.go
package cmd
import (
"fmt"
"github.com/spf13/cobra"
)
var (
version bool
info string
)
// RootCmd represents the base command when called without any subcommands
var RootCmd = &cobra.Command{
Use: "demo",
Short: "demo短描述",
Long: `这是demo的长描述,可以很长.....`,
RunE: func(cmd *cobra.Command, args []string) error {
if version {
fmt.Println("demo v1.0.0")
}
if info != "" {
fmt.Println(info)
}
return nil
},
}
func init() {
//选项分为两种
// 第一种永久选项 global flag
RootCmd.PersistentFlags().BoolVarP(&version, "version", "v", false, "输出版本信息")
// 第二种 本地选项,只能在定义它的命令中使用
RootCmd.Flags().StringVarP(&info, "info", "i", "", "原样输出您的输入")
}
三、cmd/start.go
// cmd/start.go
package cmd
import (
"fmt"
"github.com/spf13/cobra"
)
var (
// pusher service config option
confFile string
)
// StartCmd 程序的启动时, 组装在这里进行
var StartCmd = &cobra.Command{
Use: "start",
Short: "启动服务",
Long: `将启动grpc和http对外进行服务`,
RunE: func(cmd *cobra.Command, args []string) error {
return fmt.Errorf("未完善服务代码")
},
}
func init() {
//永久选项 global flag
StartCmd.PersistentFlags().StringVarP(&confFile, "config", "f",
"etc/pro.toml", "配置文件路径")
//把start 注册为 root的子节点
RootCmd.AddCommand(StartCmd)
}