diff --git a/pkg/common/cmd/rpc.go b/pkg/common/cmd/rpc.go index f05c3f43c..7c5ff5ec2 100644 --- a/pkg/common/cmd/rpc.go +++ b/pkg/common/cmd/rpc.go @@ -2,7 +2,6 @@ package cmd import ( "OpenIM/internal/startrpc" - "OpenIM/pkg/common/config" "OpenIM/pkg/discoveryregistry" "fmt" "github.com/spf13/cobra" @@ -20,12 +19,14 @@ func NewRpcCmd(rpcRegisterName string) *RpcCmd { func (r *RpcCmd) AddRpc(rpcFn func(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) error) { r.Command.RunE = func(cmd *cobra.Command, args []string) error { - fmt.Println(r.getPortFlag(cmd), r.rpcRegisterName, r.getPrometheusPortFlag(cmd), config.Config.RpcRegisterName) return startrpc.Start(r.getPortFlag(cmd), r.rpcRegisterName, r.getPrometheusPortFlag(cmd), rpcFn) } } func (r *RpcCmd) Exec(rpcFn func(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) error) error { + fmt.Println(r) + r.AddRpc(rpcFn) + fmt.Println(r) return r.Execute() }