blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-06-02 05:09:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

test cobra

Browse Source
This commit is contained in:
wangchuxiao 2023-03-07 20:41:09 +08:00
parent f0943276cb
commit 30e9b44054
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
pkg/common/cmd/root.go
View File

@ -15,21 +15,21 @@ type RootCmd struct {
prometheusPortFlag bool
}
func NewRootCmd() *RootCmd {
func NewRootCmd() (rootCmd *RootCmd) {
rootCmd = &RootCmd{}
c := cobra.Command{
Use: "start",
Short: "Start the server",
Long: `Start the server`,
}
rootCmd := &RootCmd{}
c.PersistentPreRunE = func(cmd *cobra.Command, args []string) error {
if rootCmd.portFlag {
rootCmd.port = rootCmd.getPortFlag(cmd)
}
if rootCmd.prometheusPortFlag {
rootCmd.prometheusPort = rootCmd.getPrometheusPortFlag(cmd)
}
return rootCmd.getConfFromCmdAndInit(cmd)
PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
if rootCmd.portFlag {
rootCmd.port = rootCmd.getPortFlag(cmd)
}
if rootCmd.prometheusPortFlag {
rootCmd.prometheusPort = rootCmd.getPrometheusPortFlag(cmd)
}
return rootCmd.getConfFromCmdAndInit(cmd)
},
}
rootCmd.Command = c
rootCmd.init()