blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-26 19:46: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 package cmd
type MsgUtilsCmd struct { type MsgUtilsCmd struct {
RootCmd *RootCmd
userID string userID string
userIDFlag bool userIDFlag bool

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

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