test cobra

pkg/common/cmd/root.go

@@ -1,13 +1,31 @@
 package cmd
 
 import (
+	"OpenIM/pkg/common/config"
+	"OpenIM/pkg/common/constant"
 	"github.com/spf13/cobra"
 )
 
-func NewRootCmd() *cobra.Command {
-	return &cobra.Command{
+type RootCmd cobra.Command
+
+func NewRootCmd() RootCmd {
+	c := cobra.Command{
 		Use:   "start",
 		Short: "Start the server",
 		Long:  `Start the server`,
+		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+			return getConfFromCmdAndInit(cmd)
+		},
 	}
+	c.Flags()
+	return RootCmd(c)
+}
+
+func (r RootCmd) Init() {
+	cobra.Command(r).Flags().StringP(constant.FlagConf, "c", "", "Path to config file folder")
+}
+
+func getConfFromCmdAndInit(cmdLines *cobra.Command) error {
+	configFolderPath, _ := cmdLines.Flags().GetString(constant.FlagConf)
+	return config.InitConfig(configFolderPath)
 }