IM/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2026-05-01 07:35:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

script

Browse Source
This commit is contained in:
wangchuxiao
2023-03-09 12:34:08 +08:00
parent 889f3031bf
commit fae6fdc3ed
9 changed files with 21 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/common/cmd/rpc.go
+6 -13
View File
@@ -2,34 +2,27 @@ package cmd
import (
"OpenIM/internal/startrpc"
"OpenIM/pkg/common/config"
"OpenIM/pkg/discoveryregistry"
"fmt"
"github.com/spf13/cobra"
"google.golang.org/grpc"
)
type RpcCmd struct {
*RootCmd
rpcRegisterName string
}
func NewRpcCmd(rpcRegisterName string) *RpcCmd {
rpcCmd := &RpcCmd{NewRootCmd(), rpcRegisterName}
fmt.Println("line 18", *rpcCmd, rpcCmd, rpcRegisterName, config.Config.RpcRegisterName)
func NewRpcCmd() *RpcCmd {
rpcCmd := &RpcCmd{NewRootCmd()}
return rpcCmd
}
func (r *RpcCmd) AddRpc(rpcFn func(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) error) {
func (r *RpcCmd) addRpc(rpcRegisterName string, rpcFn func(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) error) {
r.Command.RunE = func(cmd *cobra.Command, args []string) error {
return startrpc.Start(r.getPortFlag(cmd), r.rpcRegisterName, r.getPrometheusPortFlag(cmd), rpcFn)
return startrpc.Start(r.getPortFlag(cmd), 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)
func (r *RpcCmd) Exec(rpcRegisterName string, rpcFn func(client discoveryregistry.SvcDiscoveryRegistry, server *grpc.Server) error) error {
r.addRpc(rpcRegisterName, rpcFn)
return r.Execute()
}