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

test cobra

Browse Source
This commit is contained in:
wangchuxiao 2023-03-07 20:12:36 +08:00
parent 380bf1cc8c
commit c1746fb32e
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -36,45 +36,45 @@ func NewRootCmd() RootCmd {
return rootCmd return rootCmd
} }
func (r RootCmd) AddRunE(f func(cmd RootCmd) error) { func (r *RootCmd) AddRunE(f func(cmd RootCmd) error) {
r.Command.RunE = func(cmd *cobra.Command, args []string) error { r.Command.RunE = func(cmd *cobra.Command, args []string) error {
return f(r) return f(*r)
} }
} }
func (r RootCmd) init() { func (r *RootCmd) init() {
r.Command.Flags().StringP(constant.FlagConf, "c", "", "Path to config file folder") r.Command.Flags().StringP(constant.FlagConf, "c", "", "Path to config file folder")
} }
func (r RootCmd) AddPortFlag() { func (r *RootCmd) AddPortFlag() {
r.Command.Flags().StringP(constant.FlagPort, "p", "", "server listen port") r.Command.Flags().StringP(constant.FlagPort, "p", "", "server listen port")
r.portFlag = true r.portFlag = true
} }
func (r RootCmd) getPortFlag(cmd *cobra.Command) int { func (r *RootCmd) getPortFlag(cmd *cobra.Command) int {
port, _ := cmd.Flags().GetInt(constant.FlagPort) port, _ := cmd.Flags().GetInt(constant.FlagPort)
return port return port
} }
func (r RootCmd) GetPortFlag() int { func (r *RootCmd) GetPortFlag() int {
return r.port return r.port
} }
func (r RootCmd) AddPrometheusPortFlag() { func (r *RootCmd) AddPrometheusPortFlag() {
r.Command.Flags().StringP(constant.PrometheusPort, "pp", "", "server listen port") r.Command.Flags().StringP(constant.PrometheusPort, "pp", "", "server listen port")
r.prometheusPortFlag = true r.prometheusPortFlag = true
} }
func (r RootCmd) GetPrometheusPortFlag(cmd *cobra.Command) int { func (r *RootCmd) GetPrometheusPortFlag(cmd *cobra.Command) int {
port, _ := cmd.Flags().GetInt(constant.PrometheusPort) port, _ := cmd.Flags().GetInt(constant.PrometheusPort)
return port return port
} }
func (r RootCmd) getConfFromCmdAndInit(cmdLines *cobra.Command) error { func (r *RootCmd) getConfFromCmdAndInit(cmdLines *cobra.Command) error {
configFolderPath, _ := cmdLines.Flags().GetString(constant.FlagConf) configFolderPath, _ := cmdLines.Flags().GetString(constant.FlagConf)
return config.InitConfig(configFolderPath) return config.InitConfig(configFolderPath)
} }
func (r RootCmd) Execute() error { func (r *RootCmd) Execute() error {
return r.Command.Execute() return r.Command.Execute()
} }