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

test cobra

Browse Source
This commit is contained in:
wangchuxiao 2023-03-07 20:39:07 +08:00
parent 5dfda23ded
commit f0943276cb
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/common/cmd/msg.go
View File

@ -1,7 +1,7 @@
package cmd
type MsgUtilsCmd struct {
RootCmd
*RootCmd
userID string
userIDFlag bool

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

@ -15,13 +15,13 @@ type RootCmd struct {
prometheusPortFlag bool
}
func NewRootCmd() RootCmd {
func NewRootCmd() *RootCmd {
c := cobra.Command{
Use: "start",
Short: "Start the server",
Long: `Start the server`,
}
rootCmd := RootCmd{}
rootCmd := &RootCmd{}
c.PersistentPreRunE = func(cmd *cobra.Command, args []string) error {
if rootCmd.portFlag {
rootCmd.port = rootCmd.getPortFlag(cmd)
@ -31,8 +31,8 @@ func NewRootCmd() RootCmd {
}
return rootCmd.getConfFromCmdAndInit(cmd)
}
rootCmd.init()
rootCmd.Command = c
rootCmd.init()
return rootCmd
}
@ -47,7 +47,7 @@ func (r *RootCmd) init() {
}
func (r *RootCmd) AddPortFlag() {
//r.Command.Flags().IntP(constant.FlagPort, "p", 0, "server listen port")
r.Command.Flags().IntP(constant.FlagPort, "p", 0, "server listen port")
r.portFlag = true
}