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

rpc restore

Browse Source
This commit is contained in:
Gordon
2021-12-17 15:55:43 +08:00
parent 42726cf2b6
commit d6614decd1
7 changed files with 31 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/rpc/auth/rpcAuth.go
+4 -6
View File
@@ -5,12 +5,11 @@ import (
"Open_IM/pkg/common/log"
"Open_IM/pkg/grpc-etcdv3/getcdv3"
pbAuth "Open_IM/pkg/proto/auth"
"Open_IM/pkg/utils"
"google.golang.org/grpc"
"net"
"strconv"
"strings"
"github.com/spf13/viper"
"google.golang.org/grpc"
)
type rpcAuth struct {
@@ -33,7 +32,7 @@ func NewRpcAuthServer(port int) *rpcAuth {
func (rpc *rpcAuth) Run() {
log.Info("", "", "rpc get_token init...")
address := ":" + strconv.Itoa(rpc.rpcPort)
address := utils.ServerIP + ":" + strconv.Itoa(rpc.rpcPort)
listener, err := net.Listen("tcp", address)
if err != nil {
log.Error("", "", "listen network failed, err = %s, address = %s", err.Error(), address)
@@ -48,8 +47,7 @@ func (rpc *rpcAuth) Run() {
//service registers with etcd
pbAuth.RegisterAuthServer(srv, rpc)
host := viper.GetString("endpoints.rpc_auth")
err = getcdv3.RegisterEtcd(rpc.etcdSchema, strings.Join(rpc.etcdAddr, ","), host, rpc.rpcPort, rpc.rpcRegisterName, 10)
err = getcdv3.RegisterEtcd(rpc.etcdSchema, strings.Join(rpc.etcdAddr, ","), utils.ServerIP, rpc.rpcPort, rpc.rpcRegisterName, 10)
if err != nil {
log.Error("", "", "register rpc get_token to etcd failed, err = %s", err.Error())
return