From 320c978e9483d6e0c693ec3dfda878078eecf395 Mon Sep 17 00:00:00 2001 From: wangchuxiao Date: Tue, 7 Mar 2023 20:07:13 +0800 Subject: [PATCH] test cobra --- cmd/api/main.go | 2 +- pkg/common/cmd/root.go | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/cmd/api/main.go b/cmd/api/main.go index c60339e76..9cd0572e9 100644 --- a/cmd/api/main.go +++ b/cmd/api/main.go @@ -18,7 +18,7 @@ func main() { rootCmd := cmd.NewRootCmd() rootCmd.AddPortFlag() rootCmd.AddRunE(run) - if err := rootCmd.Command.Execute(); err != nil { + if err := rootCmd.Execute(); err != nil { fmt.Println(err) os.Exit(1) } diff --git a/pkg/common/cmd/root.go b/pkg/common/cmd/root.go index c52df6a28..5e2708fce 100644 --- a/pkg/common/cmd/root.go +++ b/pkg/common/cmd/root.go @@ -32,6 +32,7 @@ func NewRootCmd() RootCmd { return rootCmd.getConfFromCmdAndInit(cmd) } rootCmd.init() + rootCmd return rootCmd } @@ -73,3 +74,7 @@ func (r RootCmd) getConfFromCmdAndInit(cmdLines *cobra.Command) error { configFolderPath, _ := cmdLines.Flags().GetString(constant.FlagConf) return config.InitConfig(configFolderPath) } + +func (r RootCmd) Execute() error { + return r.Command.Execute() +}