blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +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
}
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 {
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")
}
func (r RootCmd) AddPortFlag() {
func (r *RootCmd) AddPortFlag() {
r.Command.Flags().StringP(constant.FlagPort, "p", "", "server listen port")
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)
return port
}
func (r RootCmd) GetPortFlag() int {
func (r *RootCmd) GetPortFlag() int {
return r.port
}
func (r RootCmd) AddPrometheusPortFlag() {
func (r *RootCmd) AddPrometheusPortFlag() {
r.Command.Flags().StringP(constant.PrometheusPort, "pp", "", "server listen port")
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)
return port
}
func (r RootCmd) getConfFromCmdAndInit(cmdLines *cobra.Command) error {
func (r *RootCmd) getConfFromCmdAndInit(cmdLines *cobra.Command) error {
configFolderPath, _ := cmdLines.Flags().GetString(constant.FlagConf)
return config.InitConfig(configFolderPath)
}
func (r RootCmd) Execute() error {
func (r *RootCmd) Execute() error {
return r.Command.Execute()
}