test cobra

This commit is contained in:
wangchuxiao 2023-03-07 20:07:13 +08:00
parent f5309ee3b8
commit 320c978e94
2 changed files with 6 additions and 1 deletions

View File

@ -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)
}

View File

@ -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()
}